-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1116 from sfregoso/system_resources
System resources
- Loading branch information
Showing
22 changed files
with
1,277 additions
and
4 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,41 @@ | ||
| // ====================================================================== | ||
| // \title Baremetal/SystemResources.cpp | ||
| // \author mstarch | ||
| // \brief hpp file for SystemResources component implementation class | ||
| // | ||
| // \copyright | ||
| // Copyright 2021, by the California Institute of Technology. | ||
| // ALL RIGHTS RESERVED. United States Government Sponsorship | ||
| // acknowledged. | ||
| // | ||
| // ====================================================================== | ||
|
|
||
| namespace Os { | ||
|
|
||
| SystemResources::SystemResourcesStatus SystemResources::getCpuCount(U32& cpuCount) { | ||
| // Assumes 1 CPU | ||
| cpuCount = 1; | ||
| return SYSTEM_RESOURCES_OK; | ||
| } | ||
|
|
||
| SystemResources::SystemResourcesStatus SystemResources::getCpuTicks(CpuTicks& cpu_ticks, U32 cpu_index) { | ||
| // Always 100 percent | ||
| cpu_ticks.used = 1; | ||
| cpu_ticks.total = 1; | ||
| return SYSTEM_RESOURCES_OK; | ||
| } | ||
|
|
||
|
|
||
| SystemResources::SystemResourcesStatus SystemResources::getMemUtil(MemUtil& memory_util) { | ||
| U8 stack_allocation = 0; | ||
| U8* heap_allocation = new U8; | ||
| if (heap_allocation != NULL) { | ||
| // Crude way to estimate memory usage: stack grows down, heap grows up. Stack - Heap = FREE bytes before collision | ||
| U64 free = static_cast<U64>(&stack_allocation - heap_allocation); | ||
| memory_util.total = free; | ||
| memory_util.util = 0; | ||
| delete heap_allocation; | ||
| } | ||
| return SYSTEM_RESOURCES_OK; | ||
| } | ||
| } // namespace Os |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,137 @@ | ||
| // ====================================================================== | ||
| // \title Linux/SystemResources.cpp | ||
| // \author sfregoso | ||
| // \brief hpp file for SystemResources component implementation class | ||
| // | ||
| // \copyright | ||
| // Copyright 2021, by the California Institute of Technology. | ||
| // ALL RIGHTS RESERVED. United States Government Sponsorship | ||
| // acknowledged. | ||
| // | ||
| // ====================================================================== | ||
| #include <stdio.h> /* fopen() */ | ||
| #include <stdlib.h> /* scanf */ | ||
| #include <sys/vfs.h> /* statfs() */ | ||
| #include <string.h> | ||
| #include <Os/SystemResources.hpp> | ||
| #include <Fw/Types/Assert.hpp> | ||
|
|
||
| namespace Os { | ||
| static const U32 LINUX_CPU_LINE_LIMIT = 1024; // Maximum lines to read before bailing | ||
|
|
||
| SystemResources::SystemResourcesStatus SystemResources::getCpuCount(U32 &cpuCount) { | ||
| char line[512] = {0}; | ||
| FILE *fp = NULL; | ||
| U32 cpu_count = 0; | ||
|
|
||
| if ((fp = fopen("/proc/stat", "r")) == NULL) { | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
|
|
||
| if (fgets(line, sizeof(line), fp) == NULL) { //1st line. Aggregate cpu line. Skip | ||
| fclose(fp); | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
|
|
||
| for (U32 i = 0; i < LINUX_CPU_LINE_LIMIT; i++) { | ||
| if (fgets(line, sizeof(line), fp) == NULL) { //cpu# line | ||
| break; | ||
| } | ||
|
|
||
| if (!(line[0] == 'c' && line[1] == 'p' && line[2] == 'u')) { | ||
| break; | ||
| } | ||
| cpu_count++; | ||
| } | ||
| fclose(fp); | ||
| cpuCount = cpu_count; | ||
|
|
||
| return SYSTEM_RESOURCES_OK; | ||
| } | ||
|
|
||
| SystemResources::SystemResourcesStatus SystemResources::getCpuTicks(CpuTicks &cpu_ticks, U32 cpu_index) { | ||
| char line[512] = {0}; | ||
| FILE *fp = NULL; | ||
| U32 cpu_data[4] = {0}; | ||
| U32 cpuCount = 0; | ||
| SystemResources::SystemResourcesStatus status = SYSTEM_RESOURCES_ERROR; | ||
| U64 cpuUsed = 0; | ||
| U64 cpuTotal = 0; | ||
|
|
||
| if ((status = getCpuCount(cpuCount)) != SYSTEM_RESOURCES_OK) { | ||
| return status; | ||
| } | ||
|
|
||
| if (cpu_index >= cpuCount) { | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
|
|
||
| if ((fp = fopen("/proc/stat", "r")) == NULL) { | ||
|
|
||
| return SYSTEM_RESOURCES_ERROR; | ||
|
|
||
| } | ||
|
|
||
| if (fgets(line, sizeof(line), fp) == NULL) { //1st line. Aggregate cpu line. | ||
| fclose(fp); | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
|
|
||
| for (U32 i = 0; i < cpu_index + 1; i++) { | ||
| if (fgets(line, sizeof(line), fp) == NULL) { //cpu# line | ||
| fclose(fp); | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
| if (i != cpu_index) { continue; } | ||
|
|
||
| if (!(line[0] == 'c' && line[1] == 'p' && line[2] == 'u')) { | ||
| fclose(fp); | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
|
|
||
| sscanf(line, "%*s %d %d %d %d", &cpu_data[0], | ||
| &cpu_data[1], | ||
| &cpu_data[2], | ||
| &cpu_data[3]); //cpu#: 4 numbers: usr, nice, sys, idle | ||
|
|
||
| cpuUsed = cpu_data[0] + cpu_data[1] + cpu_data[2]; | ||
| cpuTotal = cpu_data[0] + cpu_data[1] + cpu_data[2] + cpu_data[3]; | ||
|
|
||
| break; | ||
| } | ||
| fclose(fp); | ||
|
|
||
| cpu_ticks.used = cpuUsed; | ||
| cpu_ticks.total = cpuTotal; | ||
| return SYSTEM_RESOURCES_OK; | ||
| } | ||
|
|
||
| SystemResources::SystemResourcesStatus SystemResources::getMemUtil(MemUtil &memory_util) { | ||
| FILE *fp = NULL; | ||
| NATIVE_INT_TYPE total = 0; | ||
| NATIVE_INT_TYPE free = 0; | ||
| // Fallbacks | ||
| memory_util.total = 1; | ||
| memory_util.used = 1; | ||
|
|
||
| fp = fopen("/proc/meminfo", "r"); | ||
| if (fp == NULL) { | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
|
|
||
| if (fscanf(fp, "%*s %d %*s", &total) != 1 || /* 1st line is MemTotal */ | ||
| fscanf(fp, "%*s %d", &free) != 1) { /* 2nd line is MemFree */ | ||
| fclose(fp); | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
| fclose(fp); | ||
|
|
||
| // Check results | ||
| if (total < 0 or free < 0 or total < free) { | ||
| return SYSTEM_RESOURCES_ERROR; | ||
| } | ||
| memory_util.total = static_cast<U64>(total) * 1024; // KB to Bytes | ||
| memory_util.used = static_cast<U64>(total - free) * 1024; | ||
| return SYSTEM_RESOURCES_OK; | ||
| } | ||
| } |
Oops, something went wrong.