144 lines
3.5 KiB
C
Executable File
144 lines
3.5 KiB
C
Executable File
#define MY_DLL_EXPORTS 1
|
|
|
|
#include "irc.h"
|
|
#include "events.h"
|
|
#include "module.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#ifdef _WIN32
|
|
#define BUFFER_SIZE 512
|
|
#include <windows.h>
|
|
DWORD startTick;
|
|
|
|
void parseUptime(const char *output, struct irc_conn *bot, const char *where) {
|
|
// Variable definitions moved to the top
|
|
const char *keyword = "Statistics since ";
|
|
const char *uptime_start;
|
|
int month, day, year, hour, minute;
|
|
SYSTEMTIME uptime_systemtime, current_time;
|
|
ULONGLONG uptime_ticks, current_ticks;
|
|
ULONGLONG uptime_seconds;
|
|
int days, hours, minutes;
|
|
|
|
|
|
uptime_start = strstr(output, keyword);
|
|
if (uptime_start == NULL) {
|
|
printf("Failed to parse uptime information.\n");
|
|
return;
|
|
}
|
|
|
|
// Skip the "Statistics since " keyword
|
|
uptime_start += strlen(keyword);
|
|
|
|
// Parse the date and time
|
|
if (sscanf(uptime_start, "%d/%d/%d %d:%d", &month, &day, &year, &hour, &minute) != 5) {
|
|
printf("Failed to parse date and time.\n");
|
|
return;
|
|
}
|
|
|
|
// Get the current time
|
|
GetLocalTime(¤t_time);
|
|
|
|
|
|
// Calculate the uptime
|
|
uptime_systemtime.wYear = year;
|
|
uptime_systemtime.wMonth = month;
|
|
uptime_systemtime.wDayOfWeek = 0;
|
|
uptime_systemtime.wDay = day;
|
|
uptime_systemtime.wHour = hour;
|
|
uptime_systemtime.wMinute = minute;
|
|
uptime_systemtime.wSecond = 0;
|
|
uptime_systemtime.wMilliseconds = 0;
|
|
|
|
|
|
SystemTimeToFileTime(&uptime_systemtime, (FILETIME *)&uptime_ticks);
|
|
SystemTimeToFileTime(¤t_time, (FILETIME *)¤t_ticks);
|
|
|
|
uptime_seconds = (current_ticks - uptime_ticks) / 10000000UL; // Convert to seconds
|
|
|
|
uptime_seconds -= 1320;
|
|
|
|
days = uptime_seconds / (24 * 3600);
|
|
hours = (uptime_seconds % (24 * 3600)) / 3600;
|
|
minutes = (uptime_seconds % 3600) / 60;
|
|
irc_privmsg(bot, where, "Uptime: %d days, %d hours, %d minutes\n", days, hours, minutes);
|
|
}
|
|
|
|
char *executeCommand(const char *command)
|
|
{
|
|
char buffer[BUFFER_SIZE];
|
|
char *old_result = NULL;
|
|
char *result = NULL;
|
|
FILE *fp;
|
|
|
|
fp = _popen(command, "r");
|
|
if (fp == NULL) {
|
|
printf("Failed to run command\n");
|
|
exit(1);
|
|
}
|
|
|
|
while (fgets(buffer, sizeof(buffer), fp) != NULL) {
|
|
if (result == NULL) {
|
|
result = _strdup(buffer);
|
|
} else {
|
|
old_result = result;
|
|
result = malloc(strlen(old_result) + strlen(buffer) + 1);
|
|
strcpy(result, old_result);
|
|
strcat(result, buffer);
|
|
free(old_result);
|
|
}
|
|
}
|
|
|
|
_pclose(fp);
|
|
return result;
|
|
}
|
|
#endif
|
|
|
|
MY_API void up(struct irc_conn *bot, char *user, char *host, char *chan, char *text)
|
|
{
|
|
#ifdef _WIN32
|
|
const char *command = "net statistics server";
|
|
char *output = executeCommand(command);
|
|
|
|
if (!strcmp(text, "!uptime"))
|
|
{
|
|
parseUptime(output, bot, chan);
|
|
free(output);
|
|
}
|
|
#else
|
|
char buf[100];
|
|
FILE* file;
|
|
|
|
printf("dbug up called: %s!%s %s\n", user, host, text);
|
|
|
|
if (!strcmp(text, "!uptime"))
|
|
{
|
|
file = popen("uptime", "r");
|
|
fgets(buf, 100, file);
|
|
pclose(file);
|
|
|
|
irc_privmsg(bot, chan, "%s", buf);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
MY_API void mod_init()
|
|
{
|
|
#ifdef _WIN32
|
|
startTick = GetTickCount();
|
|
#endif
|
|
|
|
register_module("uptime", "Aaron Blakely", "v0.1", "Uptime module");
|
|
printf("installing up handler\n");
|
|
add_handler(PRIVMSG_CHAN, up);
|
|
}
|
|
|
|
MY_API void mod_unload()
|
|
{
|
|
unregister_module("uptime");
|
|
printf("unloading up handler\n");
|
|
del_handler(PRIVMSG_CHAN, up);
|
|
}
|