nightswatch.c

#include "header.h"

bool qPressed()
{

    fd_set fd;
    FD_SET(0, &fd);

    struct timeval tVal;
    tVal.tv_sec = 0;
    tVal.tv_usec = 0;

    int selRet = select(1, &fd, NULL, NULL, &tVal);
    if (selRet > 0)
    {
        char c = fgetc(stdin);
        if (c == EOF)
            return false;
        if (c == 'q')
        {
            fgetc(stdin); //for enter
            return true;
        }
    }
    else if (selRet < 0)
    {
        perror("select() for watching if q");
        return true;
    }
    else
    {
        return false;
    }

    return false;
}

int interr(int timeInt)
{
    long noOfThreads = sysconf(_SC_NPROCESSORS_ONLN);
    char buffer[1000];
    for (int i = 0; i < noOfThreads; i++)
        printf("CPU%d\t", i);
    printf("\n");
    char *itups[noOfThreads + 5];
    int cnt = 0;

    FILE *fd;
    while (true)
    {
        fd = fopen("/proc/interrupts", "r");
        if (fd == NULL)
        {
            perror("nightswatch interrupt: /proc/interrupts");
            return 1;
        }
        int i = 0;
        while (fgets(buffer, 1000, fd) != NULL && i++ < 3)
        {
            if (i != 3)
                continue;
            if (i == 3)
            {
                // puts(buffer);
                fflush(NULL);
                int j = 0;
                itups[0] = strtok(buffer, " \t\n");
                while (j < noOfThreads)
                {
                    // puts(itups[j]);
                    itups[++j] = strtok(NULL, " \t\n");
                }
            }
            for (int k = 1; k <= noOfThreads; k++)
            {
                printf("%s\t", itups[k]);
            }
            printf("\n");
        }

        if (qPressed())
            return 1;
        sleep(timeInt);
        if (qPressed())
            return 1;
    }
    return 1;
}

int newBorn(int timeInt)
{
    bool run = true;
    int i = 0;
    int pid;
    // printf("in newborn\n");
    while (run)
    {
        FILE *fd = fopen("/proc/loadavg", "r");
        if (fd == NULL)
        {
            perror("Nightswatch newborn read /proc/loadavg");
            return -1;
        }
        fscanf(fd, "%*s %*s %*s %*s %d", &pid);
        printf("%d\n", pid);
        if (qPressed())
            break;
        sleep(timeInt);
        if (qPressed())
            break;
        fclose(fd);
    }
    return 1;
}

int nightswatch(int numPar, char *par[])
{
    // printf("timeInt: %d\n", timeInt);
    if (numPar == 3 && strcmp(par[0], "-n") == 0)
    {
        int timeInt = atoi(par[1]);
        if (strcmp(par[2], "interrupt") == 0)
        {
            return interr(timeInt);
        }
        else if (strcmp(par[2], "newborn") == 0)
        {
            return newBorn(timeInt);
            // printf("ayee\n");
        }
        else
        {
            printf("wrong input\n");
            return -1;
        }
    }
    else if (numPar == 3)
    {
        printf("Wrong input\n");
        return -1;
    }
    else if (numPar == 1)
    {
        if (strcmp(par[0], "interrupt") == 0)
        {
            return interr(1);
        }
        else if (strcmp(par[0], "newborn") == 0)
        {
            return newBorn(1);
            // printf("ayee\n");
        }
        else
        {
            printf("wrong input\n");
            return -1;
        }
    }
    else {
         printf("Wrong input\n");
        return -1;
    }
}