corinfo.h

/*
* Part of corinfo, a library for obtaining CPU, RAM, HDD and GPU information.
* https://github.com/ColumbusUtrigas/corinfo
*
* Copyright (c) 2019 ColumbusUtrigas.
* 
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
* 
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
* 
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/

#ifndef CORINFO_H
#define CORINFO_H

#if !defined(__linux) && !defined(_WIN32)
	#error Unsupported platform
#endif

#if !defined(__GNUC__) && !defined(_MSC_VER)
	#error Unsupported compiler
#endif

#ifdef __linux
	#include <sys/sysinfo.h>
	#include <sys/statvfs.h>
	#include <sys/types.h>
	#include <unistd.h>
#endif

#ifdef _WIN32
	#include <windows.h>
	#include <Powrprof.h>
	#pragma comment(lib, "Powrprof.lib")
#endif

#ifdef __GNUC__
	#include <cpuid.h>
	#define __corinfo_cpuid(cpu, cmd) __cpuid(cmd, cpu[0], cpu[1], cpu[2], cpu[3])
#endif

#ifdef _MSC_VER
	#define __corinfo_cpuid(cpu, cmd) __cpuid(cpu, cmd)
#endif

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

/*!
* @file corinfo.h
*/

struct corinfo
{
	struct
	{
		uint32_t Count;     //!< Number of CPU cores.
		uint32_t Frequency; //!< CPU frequency, in MHz.

		uint32_t Family;
		uint32_t Model;

		uint8_t Vendor[12];
		uint8_t Name[48];

		uint8_t MMX;
		uint8_t SSE;
		uint8_t SSE2;
		uint8_t SSE3;
		uint8_t SSE41;
		uint8_t SSE42;
		uint8_t AVX;
	} Cpu;

	struct
	{
		uint64_t Total; //!< Physical RAM size, in KB.
		uint64_t Free;  //!< Available physical RAM size, in KB.
		uint32_t Usage; //!< A number between 0 and 100 that specifies the approximate percentage of physical memory that is in use.
	} Ram;

	struct
	{
		uint64_t Total; //!< HDD size, in KB.
		uint64_t Free;  //!< Available HDD size, in KB.
		uint32_t Usage; //!< A number between 0 and 100 that specifies the approximate percentage of HDD memory that is in use.
	} Hdd;
};

/**
* @brief Main library function.
*
* ## Example
*
* ```c
* struct corinfo info;
* if (corinfo_GetInfo(&info) == -1) error();
* ```
*
* @param info Pointer to corinfo struct to store data.
* @return 0 in success, otherwise -1.
*/
int corinfo_GetInfo(struct corinfo* info);

int __corinfo_cpu_info(struct corinfo* info);
int __corinfo_ram_info(struct corinfo* info);
int __corinfo_hdd_info(struct corinfo* info);

#ifdef __linux
	int __corinfo_cpu_info(struct corinfo* info)
	{
		info->Cpu.Count = sysconf(_SC_NPROCESSORS_ONLN);

		FILE* f = fopen("/proc/cpuinfo", "r");
		if (f == NULL) return -1;

		int cpu[4];

		__corinfo_cpuid(cpu, 0x00000000);

		for (int i = 0; i < 4; i++)
		{
			info->Cpu.Vendor[0 + i] = (cpu[1] >> (i * 8)) & 0xFF;
			info->Cpu.Vendor[4 + i] = (cpu[3] >> (i * 8)) & 0xFF;
			info->Cpu.Vendor[8 + i] = (cpu[2] >> (i * 8)) & 0xFF;
		}

		__corinfo_cpuid(cpu, 0x80000002); memcpy(info->Cpu.Name +  0, cpu, 16);
		__corinfo_cpuid(cpu, 0x80000003); memcpy(info->Cpu.Name + 16, cpu, 16);
		__corinfo_cpuid(cpu, 0x80000004); memcpy(info->Cpu.Name + 32, cpu, 16);

		__corinfo_cpuid(cpu, 0x00000001);
		info->Cpu.Family = ((cpu[0] >>  8) & 0xF) +  ((cpu[0] >> 20) & 0xF);
		info->Cpu.Model  = ((cpu[0] >>  4) & 0xF) + (((cpu[0] >> 16) & 0xF) << 4);
		info->Cpu.MMX    =  (cpu[3] >> 23) & 0x1;
		info->Cpu.SSE    =  (cpu[3] >> 25) & 0x1;
		info->Cpu.SSE2   =  (cpu[3] >> 26) & 0x1;
		info->Cpu.SSE3   =  (cpu[2] >>  0) & 0x1;
		info->Cpu.SSE41  =  (cpu[2] >> 19) & 0x1;
		info->Cpu.SSE42  =  (cpu[2] >> 20) & 0x1;
		info->Cpu.AVX    =  (cpu[2] >> 28) & 0x1;

		char*   line = NULL;
		size_t  line_length = 0;
		ssize_t line_read = 0;

		while ((line_read = getline(&line, &line_length, f)) != -1)
		{
			if (memcmp(line, "cpu MHz", 7) == 0)
			{
				info->Cpu.Frequency = atoi(strchr(line, ':') + 1);
				break;
			}
		}

		if (line) free(line);

		fclose(f);

		return 0;
	}

	int __corinfo_ram_info(struct corinfo* info)
	{
		struct sysinfo sys;
		if (sysinfo(&sys) == -1) return -1;

		info->Ram.Total = sys.totalram / 1024;
		info->Ram.Free  = sys.freeram / 1024;

		if (info->Ram.Total != 0)
		{
			info->Ram.Usage = 100 - (info->Ram.Free / (float)info->Ram.Total) * 100;
		}

		return 0;
	}

	int __corinfo_hdd_info(struct corinfo* info)
	{
		struct statvfs hd;
		if (statvfs("./", &hd) == -1) return -1;

		info->Hdd.Total = hd.f_frsize * hd.f_blocks / 1024;
		info->Hdd.Free  = hd.f_frsize * hd.f_bavail / 1024;

		if (info->Hdd.Total != 0)
		{
			info->Hdd.Usage = 100 - (info->Hdd.Free / (double)info->Hdd.Total) * 100;
		}

		return 0;
	}
#endif



#ifdef _WIN32
	int __corinfo_cpu_info(struct corinfo* info)
	{
		SYSTEM_INFO sys;
		GetSystemInfo(&sys);

		info->Cpu.Count = sys.dwNumberOfProcessors;

		int cpu[4];

		__corinfo_cpuid(cpu, 0x00000000);

		for (int i = 0; i < 4; i++)
		{
			info->Cpu.Vendor[0 + i] = (cpu[1] >> (i * 8)) & 0xFF;
			info->Cpu.Vendor[4 + i] = (cpu[3] >> (i * 8)) & 0xFF;
			info->Cpu.Vendor[8 + i] = (cpu[2] >> (i * 8)) & 0xFF;
		}

		__corinfo_cpuid(cpu, 0x80000002); memcpy(info->Cpu.Name +  0, cpu, 16);
		__corinfo_cpuid(cpu, 0x80000003); memcpy(info->Cpu.Name + 16, cpu, 16);
		__corinfo_cpuid(cpu, 0x80000004); memcpy(info->Cpu.Name + 32, cpu, 16);

		__corinfo_cpuid(cpu, 0x00000001);
		info->Cpu.Family = ((cpu[0] >>  8) & 0xF) +  ((cpu[0] >> 20) & 0xF);
		info->Cpu.Model  = ((cpu[0] >>  4) & 0xF) + (((cpu[0] >> 16) & 0xF) << 4);
		info->Cpu.MMX    =  (cpu[3] >> 23) & 0x1;
		info->Cpu.SSE    =  (cpu[3] >> 25) & 0x1;
		info->Cpu.SSE2   =  (cpu[3] >> 26) & 0x1;
		info->Cpu.SSE3   =  (cpu[2] >>  0) & 0x1;
		info->Cpu.SSE41  =  (cpu[2] >> 19) & 0x1;
		info->Cpu.SSE42  =  (cpu[2] >> 20) & 0x1;
		info->Cpu.AVX    =  (cpu[2] >> 28) & 0x1;

		struct PPIStruct
		{
			ULONG  Number;
			ULONG  MaxMhz;
			ULONG  CurrentMhz;
			ULONG  MhzLimit;
			ULONG  MaxIdleState;
			ULONG  CurrentIdleState;
		} PROCESSOR_POWER_INFORMATION, *PPI;

		BYTE* buf = (BYTE*)malloc(sizeof(PROCESSOR_POWER_INFORMATION) * sys.dwNumberOfProcessors);
		if (buf == NULL) return -1;
		CallNtPowerInformation(ProcessorInformation, NULL, 0, buf, sizeof(PROCESSOR_POWER_INFORMATION) * sys.dwNumberOfProcessors);
		PPI = (PPIStruct*)buf;

		if (PPI != NULL) info->Cpu.Frequency = PPI->MaxMhz;
		if (buf) free(buf);

		return 0;
	}

	int __corinfo_ram_info(struct corinfo* info)
	{
		MEMORYSTATUSEX mem;
		mem.dwLength = sizeof(MEMORYSTATUSEX);
		GlobalMemoryStatusEx(&mem);

		info->Ram.Total = mem.ullTotalPhys / 1024;
		info->Ram.Free = mem.ullAvailPhys / 1024;

		if (info->Ram.Total != 0)
		{
			info->Ram.Usage = 100 - (info->Ram.Free / (float)info->Ram.Total) * 100;
		}

		return 0;
	}

	int __corinfo_hdd_info(struct corinfo* info)
	{
		ULARGE_INTEGER free_bytes_avail;
		ULARGE_INTEGER total_bytes;
		ULARGE_INTEGER total_free_bytes;

		if (!GetDiskFreeSpaceEx(NULL, &free_bytes_avail, &total_bytes, &total_free_bytes))
		{
			return -1;
		}

		info->Hdd.Total = total_bytes.QuadPart / 1024;
		info->Hdd.Free = free_bytes_avail.QuadPart / 1024;

		if (info->Hdd.Total != 0)
		{
			info->Hdd.Usage = 100 - (info->Hdd.Free / (double)info->Hdd.Total) * 100;
		}

		return 0;
	}
#endif


int corinfo_GetInfo(struct corinfo* info)
{
	if (info == NULL) return -1;
	memset(info, 0, sizeof(struct corinfo));

	if (__corinfo_cpu_info(info) == -1) return -1;
	if (__corinfo_ram_info(info) == -1) return -1;
	if (__corinfo_hdd_info(info) == -1) return -1;

	return 0;
}

#undef __corinfo_cpuid

#endif