Как разобрать/proc/cpuinfo на любом дистрибутиве linux в C

Question

Я запускаю Ubuntu, и я не понимаю, почему мне так сложно получить количество ядер в моей системе на C! Я пробовал синтаксический анализ /proc/cpuinfo с успехом в моей системе, работающей под управлением Ubuntu, но затем я попытался использовать еще одну систему, в которой работает arch linux, которая завершилась неудачно, потому что буфер был слишком мал, и я не могу понять, как заставить его работать на обоих мои системы.

#include <stdio.h> 
#include <stdlib.h> 

int main() { 
    FILE* fp; 
    char *buffer, *tmp; 
    size_t bytes_read, size; 

    if((fp = fopen("/proc/cpuinfo", "r")) == NULL) { 
    perror("open failed"); 
    exit(EXIT_FAILURE); 
    } 
    size = 1024; 
    if((buffer = malloc(size)) == NULL) { 
    perror("malloc failed"); 
    free(buffer); 
    exit(EXIT_FAILURE); 
    } 
    while(!feof(fp)) { 
    bytes_read = fread(buffer, 1, size, fp); 
    if(bytes_read == size) { 
     size += 128; 
     if((tmp = realloc(buffer, size)) == NULL) { 
     perror("realloc failed"); 
     free(buffer); 
     exit(EXIT_FAILURE); 
     }else { 
     buffer = tmp; 
     } 
    } 
    } 
    fclose(fp); 
    if(bytes_read == 0 || bytes_read == size) { 
    perror("read failed or buffer isn't big enough."); 
    exit(EXIT_FAILURE); 
    } 
    printf("%d bytes read out of %d\n", (int)bytes_read,(int) size); 
    buffer[bytes_read] = '\0'; 

    printf("%s", buffer); 
} 

Это выводит 572 bytes read out of 1152, но он перезаписывает буфер всякий раз, когда я снова использовать fread. И я не могу использовать sysconf(_SC_NPROCESSORS_ONLN);, потому что он не работает на Ubuntu.

Answer 1

Как с помощью popen(3) выполнить cat ^processor /proc/cpuinfo | wc -l, чтобы получить количество процессоров, а затем прочитать результат из трубы? Это довольно просто, и вам не придется поддерживать сложный код для чтения и анализа всего файла.

Вот пример:

#include <stdio.h> 

int ncpus(void) { 
    FILE *cmd = popen("grep '^processor' /proc/cpuinfo | wc -l", "r"); 

    if (cmd == NULL) 
     return -1; 

    unsigned nprocs; 
    size_t n; 
    char buff[8]; 

    if ((n = fread(buff, 1, sizeof(buff)-1, cmd)) <= 0) 
     return -1; 

    buff[n] = '\0'; 
    if (sscanf(buff, "%u", &nprocs) != 1) 
     return -1; 

    pclose(cmd); 

    return nprocs; 
} 

int main(void) { 
    int cpus = ncpus(); 
    if (cpus == -1) 
     fprintf(stderr, "Error retrieving number of CPUs\n"); 
    else 
     printf("Number of CPUs: %d\n", cpus); 
    return 0; 
} 

Вы можете улучшить обработку в ncpus() ошибку, чтобы быть немного удобно (прямо сейчас, вы на самом деле не знаю, что случилось, если, если возвращается -1) ,

UPDATE

Как упомянуто ниже в комментариях, nproc(1) может быть лучшим выбором здесь, по крайней мере, команда меньше. Он будет работать в любом случае, мы можем просто заменить grep + wc на nproc.