Ticket #15585: FileTestWatchdog.cpp

#include <stdio.h>
#include <Application.h>
#include <Looper.h>
#include <MessageRunner.h>
#include <debugger.h>
#include <OS.h>
#include <syscalls.h>

enum {
    fileCnt = 50000
};

typedef bigtime_t Time;

enum {
    second = 1000000
};

inline bigtime_t Now()
{
    return system_time();
}


enum {
    resetMsg = 1,
    timerMsg,
};

enum {
    watchdogTimeout = second,
};

class WatchdogLooper: public BLooper
{
private:
    thread_id fMainThread;
    BMessageRunner *fTimer;

public:
    WatchdogLooper(thread_id mainThread): BLooper("watchdog"), fMainThread(mainThread)
    {
        BMessage tMsg(timerMsg);
        fTimer = new BMessageRunner(BMessenger(this), tMsg, watchdogTimeout, 1);
    }

    ~WatchdogLooper()
    {
        delete fTimer;
    }

    void MessageReceived(BMessage *msg)
    {
        switch (msg->what) {
        case resetMsg: {
            delete fTimer;
            BMessage tMsg(timerMsg);
            fTimer = new BMessageRunner(BMessenger(this), tMsg, watchdogTimeout, 1);
            break;
        }
        case timerMsg: {
            char str[256];
            sprintf(str, "FileTest: timeout, thread: %d", fMainThread);
            fprintf(stderr, "%s\n", str);
            _kern_kernel_debugger(str);
            break;
        }
        default:
            BLooper::MessageReceived(msg);
        }
    }

};


class TestApplication: public BApplication
{
private:
    WatchdogLooper *fWatchdog;
public:
    TestApplication(): BApplication("application/x-vnd.test.app") {
        fWatchdog = new WatchdogLooper(find_thread(NULL));
        fWatchdog->Run();
    }

    ~TestApplication()
    {
        delete fWatchdog;
    }

    void ArgvReceived(int32 argCnt, char **args)
    {
        BMessenger watchdog(fWatchdog);

        FILE *f;
        char name[256];
        Time t, t2, times[fileCnt];
        if (argCnt != 2) {
            PostMessage(B_QUIT_REQUESTED);
            return;
        }

        t = Now();
        for (int i = 0; i < fileCnt; i++) {
            sprintf(name, "%s/%d", args[1], i);
            f = fopen(name, "w");
            fprintf(f, "File %d/n", i);
            fclose(f); f = NULL;
            watchdog.SendMessage(resetMsg);
            t2 = Now();
            times[i] = t2 - t;
            t = t2;
        }
        for (int i = 0; i < fileCnt; i++) {
            printf("%.16E\n", double(times[i]) / double(second));
        }

        PostMessage(B_QUIT_REQUESTED);
    }

};

int main(int argCnt, char **args)
{
    TestApplication app;
    app.Run();
    return 0;
}