open-license-manager
2014-08-04 cd53d13fd12f8dd86514b2894e62ee6612566975
src/library/os/linux/os-linux.c
@@ -4,12 +4,13 @@
#include <netdb.h>
#include <ifaddrs.h>
#include <stdio.h>
#include <ctype.h>
#include <stdlib.h>
#include <unistd.h>
#include <linux/if_link.h>
#include <sys/socket.h>
#include <netpacket/packet.h>
#include <valgrind/memcheck.h>
#include <paths.h>
#include <stdlib.h>
@@ -30,6 +31,9 @@
#include <dirent.h>
#include <stdio.h>
#include <dbus-1.0/dbus/dbus.h>
#include <sys/utsname.h>
static int ifname_position(char *ifnames, char * ifname, int ifnames_max) {
   int i, position;
   position = -1;
@@ -48,15 +52,20 @@
   FUNCTION_RETURN f_return = OK;
   struct ifaddrs *ifaddr, *ifa;
   int family, i, s, n, if_name_position;
   int family, i, n, if_name_position;
   unsigned int if_num, if_max;
   char host[NI_MAXHOST];
   //char host[NI_MAXHOST];
   char *ifnames;
   if (getifaddrs(&ifaddr) == -1) {
      perror("getifaddrs");
      return ERROR;
   }
   if (adapterInfos != NULL) {
      memset(adapterInfos, 0, (*adapter_info_size) * sizeof(AdapterInfo));
   }
   /* count the maximum number of interfaces */
   for (ifa = ifaddr, if_max = 0; ifa != NULL; ifa = ifa->ifa_next, n++) {
      if (ifa->ifa_addr == NULL) {
@@ -110,8 +119,8 @@
          */
#ifdef _DEBUG
         s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in), host,
         NI_MAXHOST,
         NULL, 0, NI_NUMERICHOST);
               NI_MAXHOST,
               NULL, 0, NI_NUMERICHOST);
         if (s != 0) {
            printf("getnameinfo() failed: %s\n", gai_strerror(s));
         }
@@ -155,6 +164,7 @@
      f_return = BUFFER_TOO_SMALL;
   }
   freeifaddrs(ifaddr);
   free(ifnames);
   return f_return;
}
/**
@@ -163,23 +173,55 @@
 *@param uuid uuid as read in /dev/disk/by-uuid
 *@param buffer_out: unsigned char buffer[8] output buffer for result
 */
static void parseUUID(const char *uuid, unsigned char* buffer_out) {
static void parseUUID(const char *uuid, unsigned char* buffer_out,
      unsigned int out_size) {
   size_t len;
   unsigned int i, j;
   char * hexuuid;
   unsigned char cur_character;
   //remove characters not in hex set
   len = strlen(uuid);
   hexuuid = (char *) malloc(sizeof(char) * strlen(uuid));
   memset(buffer_out, 0, out_size);
   memset(hexuuid, 0, sizeof(char) * strlen(uuid));
   for (i = 0, j = 0; i < len; i++) {
      if (isxdigit(uuid[i])) {
         hexuuid[j] = uuid[i];
         j++;
      } else {
         //skip
         continue;
      }
   }
   if (j % 2 == 1) {
      hexuuid[j++] = '0';
   }
   hexuuid[j] = '\0';
   for (i = 0; i < j / 2; i++) {
      sscanf(&hexuuid[i * 2], "%2hhx", &cur_character);
      buffer_out[i % out_size] = buffer_out[i % out_size] ^ cur_character;
   }
   free(hexuuid);
}
#define MAX_UNITS 20
FUNCTION_RETURN getDiskInfos(DiskInfo * diskInfos, size_t * disk_info_size) {
   struct stat mount_stat, sym_stat;
   /*static char discard[1024];
    char device[64], name[64], type[64];
    */
   char path[MAX_PATH], cur_dir[MAX_PATH];
   char cur_dir[MAX_PATH];
   struct mntent *ent;
   int maxDrives, currentDrive, stat_result, i, len;
   int maxDrives, currentDrive, i, drive_found;
   __ino64_t *statDrives;
   DiskInfo *tmpDrives;
   FILE *aFile;
   DIR *disk_by_uuid_dir, *disk_by_label;
   struct dirent *dir;
   FUNCTION_RETURN result;
   if (diskInfos != NULL) {
      maxDrives = *disk_info_size;
@@ -188,17 +230,15 @@
      maxDrives = MAX_UNITS;
      tmpDrives = (DiskInfo *) malloc(sizeof(DiskInfo) * maxDrives);
   }
   memset(tmpDrives, 0, sizeof(DiskInfo) * maxDrives);
   statDrives = (__ino64_t *) malloc(maxDrives * sizeof(__ino64_t ));
   memset(statDrives, 0, sizeof(__ino64_t ) * maxDrives);
   ;
   aFile = setmntent("/proc/mounts", "r");
   if (aFile == NULL) {
      perror("setmntent");
      exit(1);
   }
   disk_by_uuid_dir = opendir("/dev/disk/by-uuid");
   if (disk_by_uuid_dir == NULL) {
      printf("errreeee!!!");
      /*proc not mounted*/
      return ERROR;
   }
   currentDrive = 0;
@@ -209,49 +249,92 @@
            && ent->mnt_fsname != NULL
            && strncmp(ent->mnt_fsname, "/dev/", 5) == 0) {
         if (stat(ent->mnt_fsname, &mount_stat) == 0) {
            printf("mntent: %s %s %d\n", ent->mnt_fsname, ent->mnt_dir,
                  mount_stat.st_ino);
            strcpy(tmpDrives[currentDrive].device, ent->mnt_fsname);
            statDrives[currentDrive] = mount_stat.st_ino;
            if (strcmp(ent->mnt_dir, "/") == 0) {
               tmpDrives[currentDrive].preferred = true;
            drive_found = -1;
            for (i = 0; i < currentDrive; i++) {
               if (statDrives[i] == mount_stat.st_ino) {
                  drive_found = i;
               }
            }
            currentDrive++;
            if (drive_found == -1) {
#ifdef _DEBUG
               printf("mntent: %s %s %d\n", ent->mnt_fsname, ent->mnt_dir,
                     mount_stat.st_ino);
#endif
               strcpy(tmpDrives[currentDrive].device, ent->mnt_fsname);
               statDrives[currentDrive] = mount_stat.st_ino;
               drive_found = currentDrive;
               currentDrive++;
            }
            if (strcmp(ent->mnt_dir, "/") == 0) {
               strcpy(tmpDrives[drive_found].label, "root");
#ifdef _DEBUG
               printf("drive %s set to preferred\n", ent->mnt_fsname);
#endif
               tmpDrives[drive_found].preferred = true;
            }
         }
      }
   }
   endmntent(aFile);
   while ((dir = readdir(disk_by_uuid_dir)) != NULL) {
      strcpy(cur_dir, "/dev/disk/by-uuid/");
      strcat(cur_dir, dir->d_name);
      if (stat(cur_dir, &sym_stat) == 0) {
         for (i = 0; i < currentDrive; i++) {
            if (sym_stat.st_ino == statDrives[i]) {
               parseUUID(dir->d_name, tmpDrives[i].disk_sn);
               printf("uuid %d %s %s\n", i, tmpDrives[i].device, path);
            }
         }
   if (diskInfos == NULL) {
      *disk_info_size = currentDrive;
      free(tmpDrives);
      result = OK;
   } else if (*disk_info_size >= currentDrive) {
      disk_by_uuid_dir = opendir("/dev/disk/by-uuid");
      if (disk_by_uuid_dir == NULL) {
#ifdef _DEBUG
         printf("Open /dev/disk/by-uuid fail");
#endif
         free(statDrives);
         return ERROR;
      }
   }
   closedir(disk_by_uuid_dir);
   disk_by_label = opendir("/dev/disk/by-label");
   if (disk_by_label != NULL) {
      while ((dir = readdir(disk_by_label)) != NULL) {
         strcpy(cur_dir, "/dev/disk/by-label/");
         strcat(cur_dir, dir->d_name);
      result = OK;
      *disk_info_size = currentDrive;
      while ((dir = readdir(disk_by_uuid_dir)) != NULL) {
         strcpy(cur_dir, "/dev/disk/by-uuid/");
         strncat(cur_dir, dir->d_name, 200);
         if (stat(cur_dir, &sym_stat) == 0) {
            for (i = 0; i < currentDrive; i++) {
               if (sym_stat.st_ino == statDrives[i]) {
                  strncpy(tmpDrives[i].label, dir->d_name, 255);
                  printf("label %d %s %s\n", i, tmpDrives[i].label,
                        tmpDrives[i].device);
                  parseUUID(dir->d_name, tmpDrives[i].disk_sn,
                        sizeof(tmpDrives[i].disk_sn));
#ifdef _DEBUG
                  VALGRIND_CHECK_VALUE_IS_DEFINED(tmpDrives[i].device);
                  printf("uuid %d %s %02x%02x%02x%02x\n", i,
                        tmpDrives[i].device,
                        tmpDrives[i].disk_sn[0],
                        tmpDrives[i].disk_sn[1],
                        tmpDrives[i].disk_sn[2],
                        tmpDrives[i].disk_sn[3]);
#endif
               }
            }
         }
      }
      closedir(disk_by_label);
      closedir(disk_by_uuid_dir);
      disk_by_label = opendir("/dev/disk/by-label");
      if (disk_by_label != NULL) {
         while ((dir = readdir(disk_by_label)) != NULL) {
            strcpy(cur_dir, "/dev/disk/by-label/");
            strcat(cur_dir, dir->d_name);
            if (stat(cur_dir, &sym_stat) == 0) {
               for (i = 0; i < currentDrive; i++) {
                  if (sym_stat.st_ino == statDrives[i]) {
                     strncpy(tmpDrives[i].label, dir->d_name, 255);
                     printf("label %d %s %s\n", i, tmpDrives[i].label,
                           tmpDrives[i].device);
                  }
               }
            }
         }
         closedir(disk_by_label);
      }
   } else {
      result = BUFFER_TOO_SMALL;
   }
   /*
    FILE *mounts = fopen(_PATH_MOUNTED, "r");
@@ -269,7 +352,8 @@
    }
    }
    */
   return OK;
   free(statDrives);
   return result;
}
void os_initialize() {
@@ -280,6 +364,28 @@
      ERR_load_crypto_strings();
      OpenSSL_add_all_algorithms();
   }
}
static void _getCpuid(unsigned int* p, unsigned int ax) {
   __asm __volatile
   ( "movl %%ebx, %%esi\n\t"
         "cpuid\n\t"
         "xchgl %%ebx, %%esi"
         : "=a" (p[0]), "=S" (p[1]),
         "=c" (p[2]), "=d" (p[3])
         : "0" (ax)
   );
}
FUNCTION_RETURN getCpuId(unsigned char identifier[6]) {
   unsigned int i;
   unsigned int cpuinfo[4] = { 0, 0, 0, 0 };
   _getCpuid(cpuinfo, 0);
   for (i = 0; i < 3; i++) {
      identifier[i * 2] = cpuinfo[i] & 0xFF;
      identifier[i * 2 + 1] = (cpuinfo[i] & 0xFF00) >> 8;
   }
   return OK;
}
VIRTUALIZATION getVirtualization() {
@@ -305,3 +411,22 @@
   return NONE;
}
FUNCTION_RETURN getMachineName(unsigned char identifier[6]) {
   static struct utsname u;
   if (uname(&u) < 0) {
      return ERROR;
   }
   memcpy(identifier, u.nodename, 6);
   return OK;
}
FUNCTION_RETURN getOsSpecificIdentifier(unsigned char identifier[6]) {
   char* dbus_id = dbus_get_local_machine_id();
   if (dbus_id == NULL) {
      return ERROR;
   }
   memcpy(identifier, dbus_id, 6);
   dbus_free(dbus_id);
   return OK;
}