commit 9d69f22feac731e7a7b545188ff6bb8eb0f04355 Author: Diavolo Date: Mon May 31 09:58:24 2021 +0200 Init diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..4ce6fdd --- /dev/null +++ b/.gitignore @@ -0,0 +1,340 @@ +## Ignore Visual Studio temporary files, build results, and +## files generated by popular Visual Studio add-ons. +## +## Get latest from https://github.com/github/gitignore/blob/master/VisualStudio.gitignore + +# User-specific files +*.rsuser +*.suo +*.user +*.userosscache +*.sln.docstates + +# User-specific files (MonoDevelop/Xamarin Studio) +*.userprefs + +# Build results +[Dd]ebug/ +[Dd]ebugPublic/ +[Rr]elease/ +[Rr]eleases/ +x64/ +x86/ +[Aa][Rr][Mm]/ +[Aa][Rr][Mm]64/ +bld/ +[Bb]in/ +[Oo]bj/ +[Ll]og/ + +# Visual Studio 2015/2017 cache/options directory +.vs/ +# Uncomment if you have tasks that create the project's static files in wwwroot +#wwwroot/ + +# Visual Studio 2017 auto generated files +Generated\ Files/ + +# MSTest test Results +[Tt]est[Rr]esult*/ +[Bb]uild[Ll]og.* + +# NUNIT +*.VisualState.xml +TestResult.xml + +# Build Results of an ATL Project +[Dd]ebugPS/ +[Rr]eleasePS/ +dlldata.c + +# Benchmark Results +BenchmarkDotNet.Artifacts/ + +# .NET Core +project.lock.json +project.fragment.lock.json +artifacts/ + +# StyleCop +StyleCopReport.xml + +# Files built by Visual Studio +*_i.c +*_p.c +*_h.h +*.ilk +*.meta +*.obj +*.iobj +*.pch +*.pdb +*.ipdb +*.pgc +*.pgd +*.rsp +*.sbr +*.tlb +*.tli +*.tlh +*.tmp +*.tmp_proj +*_wpftmp.csproj +*.log +*.vspscc +*.vssscc +.builds +*.pidb +*.svclog +*.scc + +# Chutzpah Test files +_Chutzpah* + +# Visual C++ cache files +ipch/ +*.aps +*.ncb +*.opendb +*.opensdf +*.sdf +*.cachefile +*.VC.db +*.VC.VC.opendb + +# Visual Studio profiler +*.psess +*.vsp +*.vspx +*.sap + +# Visual Studio Trace Files +*.e2e + +# TFS 2012 Local Workspace +$tf/ + +# Guidance Automation Toolkit +*.gpState + +# ReSharper is a .NET coding add-in +_ReSharper*/ +*.[Rr]e[Ss]harper +*.DotSettings.user + +# JustCode is a .NET coding add-in +.JustCode + +# TeamCity is a build add-in +_TeamCity* + +# DotCover is a Code Coverage Tool +*.dotCover + +# AxoCover is a Code Coverage Tool +.axoCover/* +!.axoCover/settings.json + +# Visual Studio code coverage results +*.coverage +*.coveragexml + +# NCrunch +_NCrunch_* +.*crunch*.local.xml +nCrunchTemp_* + +# MightyMoose +*.mm.* +AutoTest.Net/ + +# Web workbench (sass) +.sass-cache/ + +# Installshield output folder +[Ee]xpress/ + +# DocProject is a documentation generator add-in +DocProject/buildhelp/ +DocProject/Help/*.HxT +DocProject/Help/*.HxC +DocProject/Help/*.hhc +DocProject/Help/*.hhk +DocProject/Help/*.hhp +DocProject/Help/Html2 +DocProject/Help/html + +# Click-Once directory +publish/ + +# Publish Web Output +*.[Pp]ublish.xml +*.azurePubxml +# Note: Comment the next line if you want to checkin your web deploy settings, +# but database connection strings (with potential passwords) will be unencrypted +*.pubxml +*.publishproj + +# Microsoft Azure Web App publish settings. Comment the next line if you want to +# checkin your Azure Web App publish settings, but sensitive information contained +# in these scripts will be unencrypted +PublishScripts/ + +# NuGet Packages +*.nupkg +# The packages folder can be ignored because of Package Restore +**/[Pp]ackages/* +# except build/, which is used as an MSBuild target. +!**/[Pp]ackages/build/ +# Uncomment if necessary however generally it will be regenerated when needed +#!**/[Pp]ackages/repositories.config +# NuGet v3's project.json files produces more ignorable files +*.nuget.props +*.nuget.targets + +# Microsoft Azure Build Output +csx/ +*.build.csdef + +# Microsoft Azure Emulator +ecf/ +rcf/ + +# Windows Store app package directories and files +AppPackages/ +BundleArtifacts/ +Package.StoreAssociation.xml +_pkginfo.txt +*.appx + +# Visual Studio cache files +# files ending in .cache can be ignored +*.[Cc]ache +# but keep track of directories ending in .cache +!?*.[Cc]ache/ + +# Others +ClientBin/ +~$* +*~ +*.dbmdl +*.dbproj.schemaview +*.jfm +*.pfx +*.publishsettings +orleans.codegen.cs + +# Including strong name files can present a security risk +# (https://github.com/github/gitignore/pull/2483#issue-259490424) +#*.snk + +# Since there are multiple workflows, uncomment next line to ignore bower_components +# (https://github.com/github/gitignore/pull/1529#issuecomment-104372622) +#bower_components/ + +# RIA/Silverlight projects +Generated_Code/ + +# Backup & report files from converting an old project file +# to a newer Visual Studio version. Backup files are not needed, +# because we have git ;-) +_UpgradeReport_Files/ +Backup*/ +UpgradeLog*.XML +UpgradeLog*.htm +ServiceFabricBackup/ +*.rptproj.bak + +# SQL Server files +*.mdf +*.ldf +*.ndf + +# Business Intelligence projects +*.rdl.data +*.bim.layout +*.bim_*.settings +*.rptproj.rsuser +*- Backup*.rdl + +# Microsoft Fakes +FakesAssemblies/ + +# GhostDoc plugin setting file +*.GhostDoc.xml + +# Node.js Tools for Visual Studio +.ntvs_analysis.dat +node_modules/ + +# Visual Studio 6 build log +*.plg + +# Visual Studio 6 workspace options file +*.opt + +# Visual Studio 6 auto-generated workspace file (contains which files were open etc.) +*.vbw + +# Visual Studio LightSwitch build output +**/*.HTMLClient/GeneratedArtifacts +**/*.DesktopClient/GeneratedArtifacts +**/*.DesktopClient/ModelManifest.xml +**/*.Server/GeneratedArtifacts +**/*.Server/ModelManifest.xml +_Pvt_Extensions + +# Paket dependency manager +.paket/paket.exe +paket-files/ + +# FAKE - F# Make +.fake/ + +# JetBrains Rider +.idea/ +*.sln.iml + +# CodeRush personal settings +.cr/personal + +# Python Tools for Visual Studio (PTVS) +__pycache__/ +*.pyc + +# Cake - Uncomment if you are using it +# tools/** +# !tools/packages.config + +# Tabs Studio +*.tss + +# Telerik's JustMock configuration file +*.jmconfig + +# BizTalk build output +*.btp.cs +*.btm.cs +*.odx.cs +*.xsd.cs + +# OpenCover UI analysis results +OpenCover/ + +# Azure Stream Analytics local run output +ASALocalRun/ + +# MSBuild Binary and Structured Log +*.binlog + +# NVidia Nsight GPU debugger configuration file +*.nvuser + +# MFractors (Xamarin productivity tool) working folder +.mfractor/ + +# Local History for Visual Studio +.localhistory/ + +# BeatPulse healthcheck temp database +healthchecksdb \ No newline at end of file diff --git a/common.c b/common.c new file mode 100644 index 0000000..8097d50 --- /dev/null +++ b/common.c @@ -0,0 +1,15 @@ +#include "common.h" + +// Prints the time +void printTimeStamp(FILE *fp) +{ + time_t ltime; + struct tm result; + char stime[32]; + bzero(stime, sizeof(stime)); + + ltime = time(NULL); + localtime_r(<ime, &result); + asctime_r(&result, stime); + fprintf(fp, "\nTimeStamp: %s\n", stime); +} diff --git a/common.h b/common.h new file mode 100644 index 0000000..1209693 --- /dev/null +++ b/common.h @@ -0,0 +1,16 @@ +#ifndef COMMON_H +#define COMMON_H + +#include +#include +#include +#include +#include +#include +#include + +void printTimeStamp(FILE *fp); + +#define BUFFER_SIZE 4096 + +#endif /* COMMON_H */ diff --git a/main.c b/main.c new file mode 100644 index 0000000..dcc15b6 --- /dev/null +++ b/main.c @@ -0,0 +1,35 @@ +#include "common.h" +#include +#include "monitor.h" + +volatile sig_atomic_t sigint_received = 0; + +void sigint_handler(int a) +{ + sigint_received = 1; +} + +int main(int argc, char *argv[]) +{ + if (argc < 2) + { + printf("Name of the program is missing\n"); + printf("Usage: \n"); + return 2; + } + + struct sigaction act; + act.sa_handler = sigint_handler; + sigaction(SIGINT, &act, NULL); + + setup(argv[1]); + + while (!sigint_received) + { + analyse(); + sleep(10); + } + + printf("Normal shutdown\n"); + return 0; +} diff --git a/monitor.c b/monitor.c new file mode 100644 index 0000000..e10523c --- /dev/null +++ b/monitor.c @@ -0,0 +1,129 @@ +#include "common.h" +#include +#include "monitor.h" +#include "utils.h" + +#include +#include + +#define MAX_MEMORY 1048576 +#define PROGRAM_NAME 255 +#define MAX_TRACK_SIZE 30 + +FILE *popen(const char *command, const char *mode); +int pclose(FILE *stream); +char *strerror(int errnum); + +void checkStatus(FILE *file, int *currRealMem, int *peakRealMem, + int *currVirtMem, int *peakVirtMem) +{ + + char result[BUFFER_SIZE]; + bzero(result, sizeof(result)); + + if (file == NULL) + { + perror("popen"); + exit(EXIT_FAILURE); + } + + while (fscanf(file, " %1023s", result) == 1) + { + if (strcmp(result, "VmRSS:") == 0) + { + fscanf(file, " %d", currRealMem); + } + + if (strcmp(result, "VmHWM:") == 0) + { + fscanf(file, " %d", peakRealMem); + } + + if (strcmp(result, "VmSize:") == 0) + { + fscanf(file, " %d", currVirtMem); + } + + if (strcmp(result, "VmPeak:") == 0) + { + fscanf(file, " %d", peakVirtMem); + } + } +} + +void isProcessWild(int currRealMem, int peakRealMem, + int currVirtMem, int peakVirtMem, int ID) +{ + // Logs to a file in case something is up + if (currRealMem > MAX_MEMORY) + { + FILE *fp; + fp = fopen("Diavolo.txt", "a"); + printTimeStamp(fp); + fprintf(fp, "WARNING: Exceeded %d KiB of Real Memory for process ID: %d\n", MAX_MEMORY, ID); + fprintf(fp, "Process ID: %d\ncurrRealMem:%d KiB\npeakRealMem:%d KiB\ncurrVirtMem:%d KiB\npeakVirtMem:%d KiB\n", ID, currRealMem, peakRealMem, currVirtMem, peakVirtMem); + fclose(fp); + // Sends a polite request to terminate + int result = kill(ID, SIGTERM); + if (result) + { + printf("Error while terminating process: %d\n", ID); + printf("Error message: %s\n", strerror(errno)); + } + } +} + +static processID[MAX_TRACK_SIZE] = {0}; + +void setup(const char *pid) +{ + char *command = va("pgrep %s", pid); + FILE *file = popen(command, "r"); + + if (file == NULL) + { + perror("popen"); + exit(EXIT_FAILURE); + } + + char result[BUFFER_SIZE]; + bzero(result, sizeof(result)); + int i = 0; + + while (fgets(result, BUFFER_SIZE, file)) + { + printf("Process ID: %s", result); + processID[i] = atoi(result); + i++; + + if (i > MAX_TRACK_SIZE - 1) + { + printf("Too many processes found\n"); + break; + } + } + + pclose(file); +} + +void analyse() +{ + int i; + FILE *status; + int currRealMem = 0, peakRealMem = 0, currVirtMem = 0, peakVirtMem = 0; + + for (i = 0; i < MAX_TRACK_SIZE && processID[i] != 0; i++) + { + char *command = va("cat /proc/%d/status", processID[i]); + status = popen(command, "r"); + if (file == NULL) + { + perror("popen"); + exit(EXIT_FAILURE); + } + + checkStatus(status, &currRealMem, &peakRealMem, &currVirtMem, &peakVirtMem); + isProcessWild(currRealMem, peakRealMem, currVirtMem, peakVirtMem, processID[i]); + pclose(status); + } +} diff --git a/monitor.h b/monitor.h new file mode 100644 index 0000000..a068eb7 --- /dev/null +++ b/monitor.h @@ -0,0 +1,7 @@ +#ifndef MONITOR_H +#define MONITOR_H + +void setup(const char *pid); +void analyse(); + +#endif /* MONITOR_H */ diff --git a/utils.c b/utils.c new file mode 100644 index 0000000..75abd1d --- /dev/null +++ b/utils.c @@ -0,0 +1,22 @@ +#include "common.h" +#include "utils.h" + +#define VA_BUFFER_COUNT 4 +#define VA_BUFFER_SIZE 4096 + +char *va(const char *fmt, ...) +{ + static char g_vaBuffer[VA_BUFFER_COUNT][VA_BUFFER_SIZE]; + static int g_vaNextBufferIndex = 0; + + va_list ap; + char *dest = &g_vaBuffer[g_vaNextBufferIndex][0]; + bzero(dest, VA_BUFFER_SIZE); + g_vaNextBufferIndex = (g_vaNextBufferIndex + 1) % VA_BUFFER_COUNT; + va_start(ap, fmt); + vsnprintf(dest, VA_BUFFER_SIZE, fmt, ap); + va_end(ap); + dest[VA_BUFFER_SIZE - 1] = '\0'; + + return dest; +} diff --git a/utils.h b/utils.h new file mode 100644 index 0000000..6ef62d4 --- /dev/null +++ b/utils.h @@ -0,0 +1,6 @@ +#ifndef UTILS_H +#define UTILS_H + +char *va(const char *fmt, ...); + +#endif /* UTILS_H */