ちんちん

2 files changed, 9 insertions, 2 deletions

diff --git a/src/cpu.c b/src/cpu.c
index bf02346..ee0448c 100644
--- a/src/cpu.c
+++ b/src/cpu.c
@@ -1,6 +1,11 @@
 #include "cpu.h"
 #include "common.h"
 
+#if defined(__linux__)
+#include <stdio.h>
+#include <stdlib.h>
+#endif
+
 const char *display_cpu() {
 #if defined(__NetBSD__)
   return run_command_s("sysctl -n machdep.cpu_brand | sed 's/(R)//' | "
@@ -28,7 +33,7 @@ const char *display_cpu() {
 
   if (fp == NULL) {
     perror("失敗");
-    return;
+    return NULL;
   }
 
   if (fgets(buf, sizeof(buf), fp) != NULL) {
@@ -38,7 +43,7 @@ const char *display_cpu() {
   } else {
     perror("失敗");
     fclose(fp);
-    return;
+    return NULL;
   }
 
   fclose(fp);
diff --git a/src/packages.c b/src/packages.c
index 3505815..01fb8e2 100644
--- a/src/packages.c
+++ b/src/packages.c
@@ -18,5 +18,7 @@ const char *display_packages() {
   } else if (access("/usr/bin/dpkg-query", F_OK) != -1) {
     return run_command_s("dpkg-query -f '.\n' -W | wc -l | sed \"s/ //g\"");
   }
+
+  return NULL;
 #endif
 }