open-license-manager
2014-10-14 fc983a64e86a0b756c91db501d660bc02edd1609
src/library/pc-identifiers.c
@@ -9,6 +9,7 @@
#include "pc-identifiers.h"
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include "base/base64.h"
#include "base/base.h"
#ifdef __linux__
@@ -18,7 +19,12 @@
#include <Windows.h>
#endif
//FIXME if adapterinfo or diskinfo fail the function fails
static FUNCTION_RETURN generate_disk_pc_id(PcIdentifier * identifiers,
      unsigned int * num_identifiers, bool use_label);
static FUNCTION_RETURN generate_ethernet_pc_id(PcIdentifier * identifiers,
      unsigned int * num_identifiers, int use_mac);
static FUNCTION_RETURN generate_default_pc_id(PcIdentifier * identifiers,
                                   unsigned int * num_identifiers) {
   size_t adapter_num, disk_num;
@@ -39,17 +45,20 @@
      *num_identifiers = disk_num * adapter_num;
      function_return = FUNC_RET_OK;
   } else {
      adapterInfoPtr = (OsAdapterInfo*)malloc((*num_identifiers) * sizeof(OsAdapterInfo));
      adapterInfoPtr = (OsAdapterInfo*) malloc(
            (*num_identifiers) * sizeof(OsAdapterInfo));
      adapter_num = *num_identifiers;
      result_adapterInfos = getAdapterInfos(adapterInfoPtr, &adapter_num);
      if (result_adapterInfos != FUNC_RET_OK && result_adapterInfos != FUNC_RET_BUFFER_TOO_SMALL) {
         free(diskInfoPtr);
      if (result_adapterInfos != FUNC_RET_OK
            && result_adapterInfos != FUNC_RET_BUFFER_TOO_SMALL) {
         free(adapterInfoPtr);
         return generate_disk_pc_id(identifiers, num_identifiers, false);
      }
      diskInfoPtr = (DiskInfo*)malloc((*num_identifiers) * sizeof(DiskInfo));
      disk_num = *num_identifiers;
      result_diskinfos = getDiskInfos(adapterInfoPtr, &disk_num);
      if (result_diskinfos != FUNC_RET_OK && result_diskinfos != FUNC_RET_BUFFER_TOO_SMALL) {
      result_diskinfos = getDiskInfos(diskInfoPtr, &disk_num);
      if (result_diskinfos != FUNC_RET_OK
            && result_diskinfos != FUNC_RET_BUFFER_TOO_SMALL) {
         free(diskInfoPtr);
         free(adapterInfoPtr);
         return generate_ethernet_pc_id(identifiers, num_identifiers, true);
@@ -66,13 +75,12 @@
               goto end;
            }
            for (k = 0; k < 6; k++)
               identifiers[array_index][k] =
               diskInfoPtr[i].disk_sn[k + 2]
               identifiers[array_index][k] = diskInfoPtr[i].disk_sn[k + 2]
               ^ adapterInfoPtr[j].mac_address[k + 2];
         }
      }
end:
      *num_identifiers = min(*num_identifiers, adapter_num * disk_num);
      *num_identifiers = cmin(*num_identifiers, adapter_num * disk_num);
      free(diskInfoPtr);
      free(adapterInfoPtr);
   }
@@ -82,39 +90,41 @@
static FUNCTION_RETURN generate_ethernet_pc_id(PcIdentifier * identifiers,
                                    unsigned int * num_identifiers, int use_mac) {
   FUNCTION_RETURN result_adapterInfos;
   unsigned int i, j, k;
   unsigned int j, k;
   OsAdapterInfo *adapterInfos;
   size_t defined_adapters, adapters = 0;
   if (identifiers == NULL || *num_identifiers == 0) {
      result_adapterInfos = getAdapterInfos(NULL, &adapters);
      if (result_adapterInfos == FUNC_RET_OK || result_adapterInfos == FUNC_RET_BUFFER_TOO_SMALL) {
      if (result_adapterInfos == FUNC_RET_OK
            || result_adapterInfos == FUNC_RET_BUFFER_TOO_SMALL) {
         *num_identifiers = adapters;
         result_adapterInfos=FUNC_RET_OK;
      }
   } else {
      defined_adapters = adapters = *num_identifiers;
      adapterInfos = (OsAdapterInfo*)malloc(adapters * sizeof(OsAdapterInfo));
      adapterInfos = (OsAdapterInfo*) malloc(
            adapters * sizeof(OsAdapterInfo));
      result_adapterInfos = getAdapterInfos(adapterInfos, &adapters);
      if (result_adapterInfos == FUNC_RET_BUFFER_TOO_SMALL || result_adapterInfos == FUNC_RET_OK){
      if (result_adapterInfos == FUNC_RET_BUFFER_TOO_SMALL
            || result_adapterInfos == FUNC_RET_OK) {
         for (j = 0; j < adapters; j++) {
            for (k = 0; k < 6; k++)
            if (use_mac) {
               identifiers[j][k] = adapterInfos[j].mac_address[k + 2];
            }
            else {
               } else {
               //use ip
               if (k < 4) {
                  identifiers[j][k] = adapterInfos[j].ipv4_address[k];
               }
               else {
                  } else {
                  //padding
                  identifiers[j][k] = 42;
               }
            }
         }
         result_adapterInfos = (adapters>defined_adapters ? FUNC_RET_BUFFER_TOO_SMALL : FUNC_RET_OK);
         result_adapterInfos = (
               adapters > defined_adapters ?
                     FUNC_RET_BUFFER_TOO_SMALL : FUNC_RET_OK);
      }
      free(adapterInfos);
   }
@@ -125,7 +135,7 @@
                                 unsigned int * num_identifiers, bool use_label) {
   size_t disk_num, available_disk_info = 0;
   FUNCTION_RETURN result_diskinfos;
   unsigned int i, k, j;
   unsigned int i, j;
   int defined_identifiers;
   char firstChar;
   DiskInfo * diskInfos;
@@ -151,8 +161,7 @@
   if (identifiers == NULL) {
      free(diskInfos);
      return FUNC_RET_OK;
   }
   else if (available_disk_info > defined_identifiers) {
   } else if (available_disk_info > defined_identifiers) {
      free(diskInfos);
      return FUNC_RET_BUFFER_TOO_SMALL;
   }
@@ -162,13 +171,14 @@
      if (use_label){
         if (diskInfos[i].label[0] != 0){
            memset(identifiers[j], 0, sizeof(PcIdentifier)); //!!!!!!!
            strncpy(identifiers[j], diskInfos[i].label, sizeof(PcIdentifier));
            strncpy(identifiers[j], diskInfos[i].label,
                  sizeof(PcIdentifier));
            j++;
         }
      }
      else{
      } else {
         if (diskInfos[i].disk_sn[0] != 0){
            memcpy(identifiers[j], &diskInfos[i].disk_sn[2], sizeof(PcIdentifier));
            memcpy(identifiers[j], &diskInfos[i].disk_sn[2],
                  sizeof(PcIdentifier));
            j++;
         }
      }
@@ -269,8 +279,7 @@
   //concat_identifiers = (PcIdentifier *) malloc(concatIdentifiersSize);
   memcpy(&concat_identifiers[0], identifier1, sizeof(PcIdentifier));
   memcpy(&concat_identifiers[1], identifier2, sizeof(PcIdentifier));
   b64_data = base64(concat_identifiers, concatIdentifiersSize,
                 &b64_size);
   b64_data = base64(concat_identifiers, concatIdentifiersSize, &b64_size);
   if (b64_size > sizeof(PcSignature)) {
      return FUNC_RET_BUFFER_TOO_SMALL;
   }