| | |
| | | #include "pc-identifiers.h" |
| | | #include <stdlib.h> |
| | | #include <string.h> |
| | | #include <stdio.h> |
| | | #include "base/base64.h" |
| | | #include "base/base.h" |
| | | #ifdef __linux__ |
| | |
| | | #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; |
| | |
| | | *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); |
| | |
| | | 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); |
| | | } |
| | |
| | | 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); |
| | | } |
| | |
| | | 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; |
| | |
| | | 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; |
| | | } |
| | |
| | | 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++; |
| | | } |
| | | } |
| | |
| | | //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; |
| | | } |