X-Git-Url: http://git.grml.org/?a=blobdiff_plain;f=compile%2Fvmware-detect.c;h=8afe2d1dd18c5a39096085b50989e62cf80a82ae;hb=f385c47b680aa903afbdce8ac8441dead20d9457;hp=7662a5a9983fb815891020551edf723ec93c164d;hpb=61261381f0fd824c577b0eb5ebf924e47c2d1659;p=grml-scripts.git

diff --git a/compile/vmware-detect.c b/compile/vmware-detect.c
index 7662a5a..8afe2d1 100644
--- a/compile/vmware-detect.c
+++ b/compile/vmware-detect.c
@@ -1,3 +1,11 @@
+/* Filename:      vmware-detect.c
+*  Purpose:       Detect if running inside vmware
+*  Authors:       grml-team (grml.org), (c) Michael Gebetsroither <gebi@grml.org>
+*  Bug-Reports:   see http://grml.org/bugs/
+*  License:       This file is licensed under the GPL v2.
+*******************************************************************************/
+// return 0 if running inside vmware, 1 otherwise
+
 #include "string.h"
 #include "unistd.h"
 #include "stdio.h"
@@ -43,19 +51,23 @@ void printIdtr(const unsigned char* idtr, unsigned size)
     for(i=0; i<size; ++i) {
         char out[4] = {0};
         fmt_xlong(out, idtr[i]);
+        if(strlen(out) == 1)
+            WRITE("0");
         WRITE(out);
     }
     WRITE("\n");
 }
 
-// i386 {{{
-#if defined (__i386__)
+// i386 and x86-64 {{{
+#if defined (__i386__) || defined (__x86_64__)
 int checkVmware(const int debug)
 {
-    unsigned char idtr[6] = {0};
+    unsigned char idtr[10] = {0};
     asm("sidt %0" : "=m" (idtr));
     if(debug)
-        printIdtr(idtr, 6);
+        printIdtr(idtr, sizeof(idtr));
+    // should normally be the case on amd64, but does not work
+    //return (0xff==idtr[9]) ? 1 : 0;
     return (0xff==idtr[5]) ? 1 : 0;
 }
 int checkVmwareIO()
@@ -80,20 +92,6 @@ int checkVmwareIO()
 }
 // }}}
 
-// x86-64 {{{
-#elif defined (__x86_64__)
-// only guessed, possible need to check against 0xffff?
-int checkVmware(const int debug)
-{
-    unsigned char idtr[10];
-    asm("sidt %0" : "=m" (idtr));
-    if(debug)
-        printIdtr(idtr, 10);
-    return (0xff==idtr[9]) ? 1 : 0;
-}
-int checkVmwareIO() { return 0; }
-// }}}
-
 // others {{{
 #else
 // vmware runs only on the archs above
@@ -102,6 +100,8 @@ int checkVmwareIO() { return 0; }
 #endif
 // }}}
 
+static int Killed = FALSE;
+
 // returns 0 if running inside vmware, 1 otherwise
 int main(int argc, char* argv[]) {
     int debug = FALSE;
@@ -120,16 +120,24 @@ int main(int argc, char* argv[]) {
         DWRITE("true\n");
 
     // never returns if not running under vmware
-    void dummy() { DWRITE("false\n"); exit(1); }
+    void dummy() { Killed=TRUE; DWRITE("false\n"); exit(1); }
     signal(SIGSEGV, dummy);
     DWRITE("ioport-check: ");
     b = checkVmwareIO();
     if(b) {
         DWRITE("true\n");
         return EXIT_SUCCESS;
+    } else {
+        if(!Killed) {
+            // check unuseable or not implemented
+            DWRITE("false\n");
+            DWRITE("Check not implemented, yet!\n");
+            return a ? EXIT_SUCCESS : EXIT_FAILURE;
+        } else {
+            // never reached
+            WRITE("Error: IO check hasn't killed the program but no vmware found either!\n");
+            return EXIT_FAILURE;
+        }
     }
-    // never reached
-    WRITE("Error: IO check hasn't killed the program but no vmware found either!\n");
-    return EXIT_FAILURE;
 }
 // vim: foldmethod=marker