| | |
| | | #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) { |
| | |
| | | |
| | | 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++; |
| | | } |
| | | } |
| | | } |
| | |
| | | } |
| | | //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 |
| | | } |
| | |
| | | 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; |
| | | } |