Compare commits

...

2 Commits

Author SHA1 Message Date
Nakidai 9bddfa7a68 Add some compatibility
- Use GetTickCount64 function on Windows to get uptime
- Try to get Windows version from NetServerGetInfo function
2023-12-16 15:03:10 +03:00
Nakidai 5cf683e05f Change constants from const to define 2023-12-16 14:16:56 +03:00
2 changed files with 25 additions and 16 deletions

View File

@ -1,8 +1,8 @@
#include "defs.h" #include "defs.h"
const int max_username_length = 128; #define MAX_USERNAME_LENGTH 128
const int max_hostname_length = 128; #define MAX_HOSTNAME_LENGTH 128
const int max_uptime_length = 40; #define MAX_UPTIME_LENGTH 40
struct art_entry arts[] = { struct art_entry arts[] = {
{ {

View File

@ -5,8 +5,9 @@
#include <string.h> #include <string.h>
#ifdef _WIN32 #ifdef _WIN32
#include <wdm.h> #include <sysinfoapi.h>
#include <winsock.h> #include <winsock.h>
#include <Lm.h>
#include <winbase.h> #include <winbase.h>
#else #else
#include <sys/utsname.h> #include <sys/utsname.h>
@ -27,8 +28,13 @@ void die(int code, char *fmt, ...)
void getuptime(char *buffer, int max_length) void getuptime(char *buffer, int max_length)
{ {
#ifdef _WIN32
#define uptime.tv_sec uptime_seconds
long long uptime_seconds = GetTickCount64() / 1000;
#else
struct timespec uptime; struct timespec uptime;
clock_gettime(CLOCK_BOOTTIME, &uptime); clock_gettime(CLOCK_BOOTTIME, &uptime);
#endif /* _WIN32 */
int days = uptime.tv_sec / 86400; int days = uptime.tv_sec / 86400;
int hours = uptime.tv_sec / 3600 % 24; int hours = uptime.tv_sec / 3600 % 24;
int minutes = uptime.tv_sec / 60 % 60; int minutes = uptime.tv_sec / 60 % 60;
@ -43,6 +49,9 @@ void getuptime(char *buffer, int max_length)
else else
snprintf(buffer, max_length, "Up %d days, %d hours, %d minutes", days, hours, minutes); snprintf(buffer, max_length, "Up %d days, %d hours, %d minutes", days, hours, minutes);
} }
#ifdef _WIN32
#undef uptime.tv_sec
#endif /* _WIN32 */
int main(int argc, char **argv) int main(int argc, char **argv)
{ {
@ -61,29 +70,29 @@ int main(int argc, char **argv)
art = arts[0].art; art = arts[0].art;
} }
char hostname[max_hostname_length]; char hostname[MAX_HOSTNAME_LENGTH];
char username[max_username_length]; char username[MAX_USERNAME_LENGTH];
char uptime[max_uptime_length]; char uptime[MAX_UPTIME_LENGTH];
#ifdef _WIN32 #ifdef _WIN32
RTL_OSVERSIONINFOW version_buffer = {.dwOSVersionInfoSize = sizeof(RTL_OSVERSIONINFOW)}; SERVER_INFO_101 host_info;
RtlGetVersion(&version_buffer); NetServerGetInfo(NULL, 101, &host_info);
GetUserNameA(username, max_username_length); GetUserNameA(username, MAX_USERNAME_LENGTH);
#else #else
struct utsname uname_buf; struct utsname uname_buf;
getlogin_r(username, max_username_length); getlogin_r(username, MAX_USERNAME_LENGTH);
uname(&uname_buf); uname(&uname_buf);
#endif #endif /* _WIN32 */
gethostname(hostname, max_hostname_length); gethostname(hostname, MAX_HOSTNAME_LENGTH);
getuptime(uptime, max_uptime_length); getuptime(uptime, MAX_UPTIME_LENGTH);
printf("%s %s@%s\n", art[0], username, hostname); printf("%s %s@%s\n", art[0], username, hostname);
printf("%s --\n", art[1]); printf("%s --\n", art[1]);
#ifdef _WIN32 #ifdef _WIN32
printf("%s NT %u.%u %s\n", art[2], version_buffer.dwMajorVersion, version_buffer.dwMinorVersion, version_buffer.szCSDVersion); printf("%s NT %d.%d\n", art[2], host_info.sv101_version_major & MAJOR_VERSION_MASK, host_info.sv101_version_minor);
#else #else
printf("%s %s %s\n", art[2], uname_buf.sysname, uname_buf.release); printf("%s %s %s\n", art[2], uname_buf.sysname, uname_buf.release);
#endif #endif /* _WIN32 */
printf("%s %s\n", art[3], uptime); printf("%s %s\n", art[3], uptime);
return 0; return 0;
} }