Search code examples
ccpu-usagepid

Get CPU usage of current process


I am trying to read the CPU usage of a current process based on PID. I am using the following code to fetch CPU usage:

#include <stdlib.h>
#include <sys/types.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <listdir.h>
struct pstat {
    long unsigned int utime_ticks;
    long int cutime_ticks;
    long unsigned int stime_ticks;
    long int cstime_ticks;
    long unsigned int vsize; // virtual memory size in bytes
    long unsigned int rss; //Resident  Set  Size in bytes

    long unsigned int cpu_total_time;
};

/*
 * read /proc data into the passed struct pstat
 * returns 0 on success, -1 on error
*/
int get_usage(const pid_t pid, struct pstat* result) {
    //convert  pid to string
    char pid_s[20];
    snprintf(pid_s, sizeof(pid_s), "%d", pid);
    char stat_filepath[30] = "/proc/"; strncat(stat_filepath, pid_s,
            sizeof(stat_filepath) - strlen(stat_filepath) -1);
    strncat(stat_filepath, "/stat", sizeof(stat_filepath) -
            strlen(stat_filepath) -1);

    FILE *fpstat = fopen(stat_filepath, "r");
    if (fpstat == NULL) {
        perror("FOPEN ERROR ");
        return -1;
    }

    FILE *fstat = fopen("/proc/stat", "r");
    if (fstat == NULL) {
        perror("FOPEN ERROR ");
        fclose(fstat);
        return -1;
    }

    //read values from /proc/pid/stat
    bzero(result, sizeof(struct pstat));
    long int rss;
    if (fscanf(fpstat, "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %lu"
                "%lu %ld %ld %*d %*d %*d %*d %*u %lu %ld",
                &result->utime_ticks, &result->stime_ticks,
                &result->cutime_ticks, &result->cstime_ticks, &result->vsize,
                &rss) == EOF) {
        fclose(fpstat);
        return -1;
    }
    fclose(fpstat);
    result->rss = rss * getpagesize();

    //read+calc cpu total time from /proc/stat
    long unsigned int cpu_time[10];
    bzero(cpu_time, sizeof(cpu_time));
    if (fscanf(fstat, "%*s %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu",
                &cpu_time[0], &cpu_time[1], &cpu_time[2], &cpu_time[3],
                &cpu_time[4], &cpu_time[5], &cpu_time[6], &cpu_time[7],
                &cpu_time[8], &cpu_time[9]) == EOF) {
        fclose(fstat);
        return -1;
    }

    fclose(fstat);

    for(int i=0; i < 10;i++)
        result->cpu_total_time += cpu_time[i];

    return 0;
}

Got the above code from this github project: https://github.com/fho/code_snippets/blob/master/c/getusage.c and is an answer of this question: https://stackoverflow.com/a/4410209/9951420

Now when I am trying to call the get_usage() in my function as follows:

int my_cpu_usage_func(int pid){
  struct pstat* result;
  double cpu = 0.0;
  int success = get_usage(pid, result);
  if(success == 0) printf("CPU usage read successfully\n");
  if(success == -1) printf("Couldn't read CPU usage\n");
  cpu = result->cpu_total_time;
  printf("CPU usage of %i %f", pid, cpu);
}

When I am compiling my_cpu_usage_func(int) I am first getting the following warning:

extmodule.c:131:23: note: initialize the variable 'result' to silence this warning struct pstat* result; ^ = NULL

And then when I am executing the same function I am getting the following error:

FOPEN ERROR : No such file or directory

Can anyone help in this? N.B. I am new to C programming, so any help will be awesome.


Solution

  • There are a few trivial errors in the program; I Annotated the fixes with // <<-- [And I added a main() function]


    #include <stdlib.h>
    #include <sys/types.h>
    #include <stdio.h>
    #include <string.h>
    #include <unistd.h>
    #pragma include <listdir.h> // <<-- non-existent header
    
    struct pstat {
        long unsigned int utime_ticks;
        long int cutime_ticks;
        long unsigned int stime_ticks;
        long int cstime_ticks;
        long unsigned int vsize; // virtual memory size in bytes
        long unsigned int rss; //Resident  Set  Size in bytes
    
        long unsigned int cpu_total_time;
    };
    
    /*
     * read /proc data into the passed struct pstat
     * returns 0 on success, -1 on error
    */
    int get_usage(const pid_t pid, struct pstat* result) {
        char stat_filepath[80] ;
        int ii;
            // use snprintf() , not str[n]cat() to compose strings
        ii = snprintf(stat_filepath, sizeof stat_filepath, "/proc/%d/stat", pid );
        if (ii >= sizeof stat_filepath) return -1;
    
        FILE *fpstat = fopen(stat_filepath, "r");
        if (fpstat == NULL) {
            perror("FOPEN ERROR "); // <<-- should be printed by caller
            return -1;
        }
    
        FILE *fstat = fopen("/proc/stat", "r");
        if (fstat == NULL) {
            perror("FOPEN ERROR "); // <<-- could be printed by caller
            fclose(fstat);
            return -1;
        }
    
        //read values from /proc/pid/stat
        memset(result, 0 , sizeof *result ); // bzero is a BSD-ism
        long int rss;
        if (fscanf(fpstat, "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %lu"
                    "%lu %ld %ld %*d %*d %*d %*d %*u %lu %ld",
                    &result->utime_ticks, &result->stime_ticks,
                    &result->cutime_ticks, &result->cstime_ticks, &result->vsize,
                    &rss) == EOF) {
            fclose(fpstat);
            return -1;
        }
        fclose(fpstat);
        result->rss = rss * getpagesize();
    
        //read+calc cpu total time from /proc/stat
        long unsigned int cpu_time[10];
        memset(cpu_time,0, sizeof cpu_time);
        if (fscanf(fstat, "%*s %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu",
                    &cpu_time[0], &cpu_time[1], &cpu_time[2], &cpu_time[3],
                    &cpu_time[4], &cpu_time[5], &cpu_time[6], &cpu_time[7],
                    &cpu_time[8], &cpu_time[9]) == EOF) {
            fclose(fstat);
            return -1;
        }
    
        fclose(fstat);
    
        for(ii=0; ii < 10;ii++)
            result->cpu_total_time += cpu_time[ii];
    
        return 0;
    }
    int my_cpu_usage_func(int pid){
      struct pstat result; // <<-- this is a structure, not a pointer
      double cpu = 0.0;
      int success ;
    
      success = get_usage(pid, &result); // <<-- but the function expects a pointer, so give it one
      if (success == -1) printf("Couldn't read CPU usage\n");
      else if(success == 0) {
            printf("CPU usage read successfully\n");
            cpu = result.cpu_total_time;
            printf("CPU usage of %d %f\n", pid, cpu);
            }
      return success;
    }
    
    int main(void)
    {
    my_cpu_usage_func(getpid());
    
    return 0;
    }