open-license-manager
2014-07-31 6b97710009d2830fa2ca86438a8eb83e3abf5810
src/library/pc-identifiers.c
@@ -11,6 +11,7 @@
#include <string.h>
#include <stdbool.h>
#include "base/base64.h"
#include <valgrind/memcheck.h>
static FUNCTION_RETURN generate_default_pc_id(PcIdentifier * identifiers,
      unsigned int * num_identifiers) {
@@ -98,33 +99,50 @@
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, k, 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];
         if(diskInfos[i].label[0]!=0){
            memset(identifiers[j],0,sizeof(PcIdentifier)); //!!!!!!!
            strncpy(identifiers[j],diskInfos[i].label,sizeof(PcIdentifier));
            j++;
         }
         } else {
            identifiers[i][k] = diskInfos[i].disk_sn[k + 2];
         if(diskInfos[i].disk_sn[0]!=0){
            memcpy(identifiers[j],&diskInfos[i].disk_sn[2],sizeof(PcIdentifier));
            j++;
         }
      }
   }
@@ -181,7 +199,7 @@
      }
      //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
         }
@@ -258,7 +276,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;
}