open-license-manager
2014-08-04 8af6b12ee8716d2004d13bbfc81281953975b466
src/library/pc-identifiers.c
@@ -11,56 +11,105 @@
#include <string.h>
#include <stdbool.h>
#include "base/base64.h"
#include <valgrind/memcheck.h>
FUNCTION_RETURN generate_ethernet_pc_id(PcIdentifier * identifiers,
      unsigned int * num_identifiers, bool use_mac);
static void encodeStrategy(PcIdentifier * identifier,
      IDENTIFICATION_STRATEGY strategy);
static FUNCTION_RETURN generate_platform_specific_pc_id(
      PcIdentifier * identifiers, unsigned int * num_identifiers);
static FUNCTION_RETURN generate_default_pc_id(PcIdentifier * identifiers,
      unsigned int * num_identifiers) {
   size_t adapter_num, disk_num;
   FUNCTION_RETURN result_adapterInfos, result_diskinfos;
   unsigned int required_id_size, i, j, k;
   size_t adapter_num, disk_num, plat_spec_id;
   FUNCTION_RETURN result_adapterInfos, result_diskinfos, result_plat_spec;
   unsigned int required_id_size, current_identifier, i, j, k;
   DiskInfo * diskInfos;
   AdapterInfo *adapterInfos;
   required_id_size = 0;
   result_adapterInfos = getAdapterInfos(NULL, &adapter_num);
   if (result_adapterInfos != OK) {
      //call generate_disk_pc_id;
      return result_adapterInfos;
   //just calculate the number of required identifiers
   result_plat_spec = generate_platform_specific_pc_id(NULL, &plat_spec_id);
   if (result_plat_spec == OK) {
      required_id_size += 1;
   }
   result_adapterInfos = getAdapterInfos(NULL, &adapter_num);
   result_diskinfos = getDiskInfos(NULL, &disk_num);
   if (result_diskinfos == OK) {
      required_id_size = disk_num * adapter_num;
   } else {
      required_id_size = disk_num;
   if (result_diskinfos == OK && result_adapterInfos == OK) {
      required_id_size += disk_num * adapter_num;
   } else if (result_adapterInfos == OK) {
      required_id_size += adapter_num;
   } else if (result_diskinfos == OK) {
      required_id_size += disk_num;
   }
   int defined_identifiers = *num_identifiers;
   *num_identifiers = required_id_size;
   if (identifiers == NULL) {
      *num_identifiers = required_id_size;
      return OK;
   } else if (required_id_size > defined_identifiers) {
      return BUFFER_TOO_SMALL;
   }
   diskInfos = (DiskInfo*) malloc(disk_num * sizeof(DiskInfo));
   result_diskinfos = getDiskInfos(diskInfos, &disk_num);
   adapterInfos = (AdapterInfo*) malloc(adapter_num * sizeof(AdapterInfo));
   result_adapterInfos = getAdapterInfos(adapterInfos, &adapter_num);
   for (i = 0; i < disk_num; i++) {
      for (j = 0; j < adapter_num; j++) {
         for (k = 0; k < 6; k++)
            identifiers[i * adapter_num + j][k] =
                  diskInfos[i].disk_sn[k + 2]
                        ^ adapterInfos[j].mac_address[k + 2];
   //calculate the identifiers
   current_identifier = 0;
   if (result_plat_spec == OK) {
      generate_platform_specific_pc_id(identifiers, 1);
      current_identifier += 1;
   }
   if (result_diskinfos == OK && result_adapterInfos == OK) {
      diskInfos = (DiskInfo*) malloc(disk_num * sizeof(DiskInfo));
      result_diskinfos = getDiskInfos(diskInfos, &disk_num);
      adapterInfos = (AdapterInfo*) malloc(adapter_num * sizeof(AdapterInfo));
      result_adapterInfos = getAdapterInfos(adapterInfos, &adapter_num);
      for (i = 0; i < disk_num; i++) {
         for (j = 0; j < adapter_num; j++) {
            if (current_identifier > defined_identifiers) {
               break;
            }
            for (k = 0; k < 6; k++) {
               identifiers[current_identifier][k] = diskInfos[i].disk_sn[k
                     + 2] ^ adapterInfos[j].mac_address[k + 2];
            }
            encodeStrategy(&identifiers[current_identifier], DEFAULT);
            current_identifier++;
         }
      }
      free(diskInfos);
      free(adapterInfos);
   } else if (result_adapterInfos == OK) {
      i=defined_identifiers-current_identifier;
      return generate_ethernet_pc_id(&identifiers[current_identifier],
            &i, true);
   } else if (result_diskinfos == OK) {
      i=defined_identifiers-current_identifier;
      return generate_disk_pc_id(&identifiers[current_identifier],
            &i, false);
   }
   free(diskInfos);
   free(adapterInfos);
   return OK;
}
static void encodeStrategy(PcIdentifier * identifier,
      IDENTIFICATION_STRATEGY strategy) {
   unsigned char strategy_num = strategy << 5;
   identifier[0][0] = (identifier[0][0] & 15)
               ++     | strategy_num;
}
static FUNCTION_RETURN generate_platform_specific_pc_id(
         * identifiers, unsigned int * num_identifiers) {
}
static FUNCTION_RETURN generate_ethernet_pc_id(PcIdentifier * identifiers,
      unsigned int * num_identifiers, bool use_mac) {
   size_t adapters;
   FUNCTION_RETURN result_adapterInfos;
   unsigned int i, j, k;
   unsigned int j, k;
   AdapterInfo *adapterInfos;
   result_adapterInfos = getAdapterInfos(NULL, &adapters);
@@ -98,33 +147,52 @@
static FUNCTION_RETURN generate_disk_pc_id(PcIdentifier * identifiers,
      unsigned int * num_identifiers, bool use_label) {
   size_t disk_num;
   size_t disk_num, available_disk_info = 0;
   FUNCTION_RETURN result_diskinfos;
   unsigned int i, k;
   unsigned int i, j;
   char firstChar;
   DiskInfo * diskInfos;
   result_diskinfos = getDiskInfos(NULL, &disk_num);
   if (result_diskinfos != OK) {
      return result_diskinfos;
   }
   diskInfos = (DiskInfo*) malloc(disk_num * sizeof(DiskInfo));
//memset(diskInfos,0,disk_num * sizeof(DiskInfo));
   result_diskinfos = getDiskInfos(diskInfos, &disk_num);
   if (result_diskinfos != OK) {
      free(diskInfos);
      return result_diskinfos;
   }
   for (i = 0; i < disk_num; i++) {
      firstChar = use_label ? diskInfos[i].label[0] : diskInfos[i].disk_sn[0];
      available_disk_info += firstChar == 0 ? 0 : 1;
   }
   int defined_identifiers = *num_identifiers;
   *num_identifiers = disk_num;
   *num_identifiers = available_disk_info;
   if (identifiers == NULL) {
      free(diskInfos);
      return OK;
   } else if (disk_num > defined_identifiers) {
   } else if (available_disk_info > defined_identifiers) {
      free(diskInfos);
      return BUFFER_TOO_SMALL;
   }
   diskInfos = (DiskInfo*) malloc(disk_num * sizeof(DiskInfo));
   result_diskinfos = getDiskInfos(diskInfos, &disk_num);
   j = 0;
   for (i = 0; i < disk_num; i++) {
      for (k = 0; k < 6; k++) {
         if (use_label) {
            identifiers[i][k] = diskInfos[i].label[k];
         } else {
            identifiers[i][k] = diskInfos[i].disk_sn[k + 2];
      if (use_label) {
         if (diskInfos[i].label[0] != 0) {
            memset(identifiers[j], 0, sizeof(PcIdentifier)); //!!!!!!!
            strncpy(identifiers[j], diskInfos[i].label,
                  sizeof(PcIdentifier));
            j++;
         }
      } else {
         if (diskInfos[i].disk_sn[0] != 0) {
            memcpy(identifiers[j], &diskInfos[i].disk_sn[2],
                  sizeof(PcIdentifier));
            j++;
         }
      }
   }
@@ -166,22 +234,17 @@
   case DISK_NUM:
      result = generate_disk_pc_id(identifiers, array_size, false);
      break;
   case DISK_LABEL:
      result = generate_disk_pc_id(identifiers, array_size, true);
   case PLATFORM_SPECIFIC:
      result = generate_platform_specific_pc_id(identifiers, array_size);
      break;
   default:
      return ERROR;
   }
   if (result == OK && identifiers != NULL) {
      strategy_num = strategy << 5;
      for (i = 0; i < *array_size; i++) {
         //encode strategy in the first three bits of the pc_identifier
         identifiers[i][0] = (identifiers[i][0] & 15) | strategy_num;
      }
      //fill array if larger
      for (i = *array_size; i < original_array_size; i++) {
         identifiers[i][0] = STRATEGY_UNKNOWN;
         identifiers[i][0] = STRATEGY_UNKNOWN << 5;
         for (j = 1; j < sizeof(PcIdentifier); j++) {
            identifiers[i][j] = 42; //padding
         }
@@ -215,11 +278,11 @@
FUNCTION_RETURN encode_pc_id(PcIdentifier identifier1, PcIdentifier identifier2,
      PcSignature pc_identifier_out) {
   //TODO base62 encoding, now uses base64
//TODO base62 encoding, now uses base64
   PcIdentifier concat_identifiers[2];
   int b64_size = 0;
   size_t concatIdentifiersSize = sizeof(PcIdentifier) * 2;
   //concat_identifiers = (PcIdentifier *) malloc(concatIdentifiersSize);
//concat_identifiers = (PcIdentifier *) malloc(concatIdentifiersSize);
   memcpy(&concat_identifiers[0], identifier1, sizeof(PcIdentifier));
   memcpy(&concat_identifiers[1], identifier2, sizeof(PcIdentifier));
   char* b64_data = base64(concat_identifiers, concatIdentifiersSize,
@@ -229,7 +292,7 @@
   }
   sprintf(pc_identifier_out, "%.4s-%.4s-%.4s-%.4s", &b64_data[0],
         &b64_data[4], &b64_data[8], &b64_data[12]);
   //free(concat_identifiers);
//free(concat_identifiers);
   free(b64_data);
   return OK;
}
@@ -258,7 +321,10 @@
      free(identifiers);
      return result;
   }
   VALGRIND_CHECK_VALUE_IS_DEFINED(identifiers[0]);
   VALGRIND_CHECK_VALUE_IS_DEFINED(identifiers[1]);
   result = encode_pc_id(identifiers[0], identifiers[1], identifier_out);
   VALGRIND_CHECK_VALUE_IS_DEFINED(identifier_out);
   free(identifiers);
   return result;
}
@@ -272,7 +338,7 @@
 */
static FUNCTION_RETURN decode_pc_id(PcIdentifier identifier1_out,
      PcIdentifier identifier2_out, PcSignature pc_signature_in) {
   //TODO base62 encoding, now uses base64
//TODO base62 encoding, now uses base64
   unsigned char * concat_identifiers;
   char base64ids[17];
@@ -302,7 +368,7 @@
   PcIdentifier* calculated_identifiers = NULL;
   unsigned int calc_identifiers_size = 0;
   int i = 0, j = 0;
   //bool found;
//bool found;
#ifdef _DEBUG
   printf("Comparing pc identifiers: \n");
#endif
@@ -311,7 +377,7 @@
      return result;
   }
   previous_strategy_id = STRATEGY_UNKNOWN;
   //found = false;
//found = false;
   for (i = 0; i < 2; i++) {
      current_strategy_id = strategy_from_pc_id(user_identifiers[i]);
      if (current_strategy_id == STRATEGY_UNKNOWN) {