From a2bd4b2b5cee06e89084910171341e729fc215e2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=AB=8F=E8=A8=AA=E5=AD=90?= Date: Tue, 18 Jun 2024 01:43:55 +0900 Subject: =?UTF-8?q?FreeBSD:=20=E3=83=9B=E3=82=B9=E3=83=88=E5=90=8D?= =?UTF-8?q?=E3=81=A8=E3=83=A2=E3=83=87=E3=83=AB=E5=90=8D=E3=81=AE=E4=BF=AE?= =?UTF-8?q?=E6=AD=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/host.c | 63 +++++++++++++++++++++++++++++++++++++++++++------------------- src/user.c | 63 ++++++++++++++++++++++++++++++++++---------------------------- 2 files changed, 79 insertions(+), 47 deletions(-) (limited to 'src') diff --git a/src/host.c b/src/host.c index 5369de2..010af9e 100644 --- a/src/host.c +++ b/src/host.c @@ -1,23 +1,24 @@ #include "host.h" #include +#include #include #if defined(__linux__) #include int skip = 0; -#elif defined(__APPLE__) -#include #endif -void run_host_command(const char *command) { +const char *run_host_command(const char *command) { char buf[64]; + char *out = NULL; + size_t outsize = 0; FILE *p = popen(command, "r"); if (!p) { fprintf(stderr, "ホストコマンドを実効に失敗: %s", command); - return; + return NULL; } while (fgets(buf, sizeof(buf), p) != NULL) { @@ -40,28 +41,52 @@ void run_host_command(const char *command) { skip = 1; break; } - - if (!skip) printf("%s", buf); -#else - printf("%s", buf); #endif + + size_t len = strlen(buf); + char *nout = realloc(out, outsize + len + 1); + if (nout == NULL) { + perror("メモリの役割に失敗"); + free(out); + pclose(p); + return NULL; + } + + out = nout; + + memcpy(out + outsize, buf, len); + outsize += len; + out[outsize] = '\0'; } pclose(p); + + return out; } void display_host_model() { -#if defined(__OpenBSD__) || defined(__FreeBSD__) || \ - defined(__DragonFly__) - run_host_command("sysctl -n hw.vendor && echo \" \" && " +#if defined(__OpenBSD__) + printf("%s", run_host_command("sysctl -n hw.vendor && echo \" \" && " "sysctl -n hw.version && echo \" \" &&" - "sysctl -n hw.product"); + "sysctl -n hw.product")); +#elif defined(__FreeBSD__) + const char *family = run_host_command("kenv | grep smbios.system.family | " + "sed 's/\"//g' | sed 's/smbios.system.family=//'"); + if (strncmp(family, " ", strlen(family)) == 0) { + family = run_host_command("kenv | grep smbios.system.version | " + "sed 's/\"//g' | sed 's/smbios.system.version=//'"); + } + const char *product = run_host_command("kenv | grep smbios.system.product | " + "sed 's/\"//g' | sed 's/smbios.system.product=//'"); + const char *maker = run_host_command("kenv | grep smbios.system.maker | " + "sed 's/\"//g' | sed 's/smbios.system.maker=//'"); + printf("%s %s %s", maker, family, product); #elif defined(__NetBSD__) - run_host_command("sysctl -n machdep.dmi.system-vendor && " + printf("%s", run_host_command("sysctl -n machdep.dmi.system-vendor && " "echo \" \" && sysctl -n machdep.dmi.system-version && " - "echo \" \" && sysctl -n machdep.dmi.system-product"); + "echo \" \" && sysctl -n machdep.dmi.system-product")); #elif defined(__sun) - run_host_command("prtconf -b | awk -F':' '/banner-name/ {printf $2}'"); + printf("%s", run_host_command("prtconf -b | awk -F':' '/banner-name/ {printf $2}'")); #elif defined(__linux__) const char *cmd1 = NULL; const char *cmd2 = NULL; @@ -83,15 +108,15 @@ void display_host_model() { if (!cmd1) { printf("Unknown"); } else { - run_host_command(cmd1); + printf("%s", run_host_command(cmd1)); } if (cmd2) { - printf(" "); - run_host_command(cmd2); + const char *model = run_host_command(cmd2); + if (!skip) printf(" %s", model); } #elif defined(__APPLE__) - run_host_command("sysctl -n hw.model"); + printf("%s", run_host_command("sysctl -n hw.model")); FILE *p = popen("kextstat | grep -F -e \"FakeSMC\" -e \"VirtualSMC\"", "r"); if (!p) { diff --git a/src/user.c b/src/user.c index 9c01c82..a3805d5 100644 --- a/src/user.c +++ b/src/user.c @@ -3,55 +3,62 @@ #include #include -void display_user_name() { - char buf[128]; - FILE *p = popen("whoami", "r"); +void run_user_command(const char *command) { + char buf[64]; +#if defined(__NetBSD__) || defined(__FreeBSD__) + FILE *p = popen(command, "r"); if (!p) { - fprintf(stderr, "「whoami」コマンドを実効に失敗"); + snprintf(buf, sizeof(buf), "「%s」コマンドを見つけられません。", command); + perror(buf); return; } while (fgets(buf, sizeof(buf), p) != NULL) { - buf[strcspn(buf, "\n")] = '\0'; printf("%s", buf); } pclose(p); -} - -void display_user_host() { - char buf[64]; -#ifdef __NetBSD__ - FILE *p = popen("hostname", "r"); - if (!p) { - perror("hostnameコマンドを見つけられません。"); + // +#else + FILE *f = fopen(command, "r"); + if (!f) { + snprintf(buf, sizeof(buf), "「%s」ファイルを見つけられません。", command); + perror(buf); return; } - while (fgets(buf, sizeof(buf), p) != NULL) { - buf[strcspn(buf, "\n")] = '\0'; + while (fgets(buf, sizeof(buf), f) != NULL) { printf("%s", buf); } - pclose(p); -#else - const char *filename; -#ifdef __OpenBSD__ - filename = "/etc/myname"; -#else - filename = "/etc/hostname"; + fclose(f); #endif - FILE *f = fopen(filename, "r"); - if (!f) { - snprintf(buf, sizeof(buf), "「%s」ファイルを見つけられません。", filename); - perror(buf); +} + +void display_user_name() { + char buf[128]; + FILE *p = popen("whoami", "r"); + if (!p) { + fprintf(stderr, "「whoami」コマンドを実効に失敗"); return; } - while (fgets(buf, sizeof(buf), f) != NULL) { + while (fgets(buf, sizeof(buf), p) != NULL) { + buf[strcspn(buf, "\n")] = '\0'; printf("%s", buf); } - fclose(f); + pclose(p); +} + +void display_user_host() { +#if defined(__OpenBSD__) + run_user_command("/etc/myname"); +#elif defined(__NetBSD__) + run_user_command("hostname"); +#elif defined(__FreeBSD__) + run_user_command("sysctl -n kern.hostname"); +#else + run_user_command("/etc/hostname"); #endif } -- cgit v1.2.3