| | |
| | | #include <Windows.h> |
| | | #endif |
| | | |
| | | //FIXME if adapterinfo or diskinfo fail the function fails |
| | | 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, defined_identifiers, i, j, k; |
| | | DiskInfo * diskInfos; |
| | | OsAdapterInfo *adapterInfos; |
| | | FUNCTION_RETURN result_adapterInfos, result_diskinfos, function_return; |
| | | unsigned int caller_identifiers, i, j, k, array_index; |
| | | DiskInfo * diskInfoPtr; |
| | | OsAdapterInfo *adapterInfoPtr; |
| | | |
| | | if (identifiers == NULL || *num_identifiers == 0) { |
| | | result_adapterInfos = getAdapterInfos(NULL, &adapter_num); |
| | | if (result_adapterInfos != FUNC_RET_OK) { |
| | | //call generate_disk_pc_id; |
| | | return result_adapterInfos; |
| | | return generate_disk_pc_id(identifiers, num_identifiers, false); |
| | | } |
| | | result_diskinfos = getDiskInfos(NULL, &disk_num); |
| | | if (result_diskinfos == FUNC_RET_OK) { |
| | | required_id_size = disk_num * adapter_num; |
| | | if (result_diskinfos != FUNC_RET_OK) { |
| | | return generate_ethernet_pc_id(identifiers, num_identifiers, true); |
| | | } |
| | | *num_identifiers = disk_num * adapter_num; |
| | | function_return = FUNC_RET_OK; |
| | | } else { |
| | | required_id_size = disk_num; |
| | | 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); |
| | | return generate_disk_pc_id(identifiers, num_identifiers, false); |
| | | } |
| | | defined_identifiers = *num_identifiers; |
| | | *num_identifiers = required_id_size; |
| | | if (identifiers == NULL) { |
| | | return FUNC_RET_OK; |
| | | } else if (required_id_size > defined_identifiers) { |
| | | return FUNC_RET_BUFFER_TOO_SMALL; |
| | | 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) { |
| | | free(diskInfoPtr); |
| | | free(adapterInfoPtr); |
| | | return generate_ethernet_pc_id(identifiers, num_identifiers, true); |
| | | } |
| | | diskInfos = (DiskInfo*) malloc(disk_num * sizeof(DiskInfo)); |
| | | result_diskinfos = getDiskInfos(diskInfos, &disk_num); |
| | | adapterInfos = (OsAdapterInfo*) malloc(adapter_num * sizeof(OsAdapterInfo)); |
| | | result_adapterInfos = getAdapterInfos(adapterInfos, &adapter_num); |
| | | function_return = FUNC_RET_OK; |
| | | |
| | | caller_identifiers = *num_identifiers; |
| | | for (i = 0; i < disk_num; i++) { |
| | | for (j = 0; j < adapter_num; j++) { |
| | | array_index = i * adapter_num + j; |
| | | if (array_index>=caller_identifiers) { |
| | | function_return = FUNC_RET_BUFFER_TOO_SMALL; |
| | | //sweet memories... |
| | | goto end; |
| | | } |
| | | for (k = 0; k < 6; k++) |
| | | identifiers[i * adapter_num + j][k] = |
| | | diskInfos[i].disk_sn[k + 2] |
| | | ^ adapterInfos[j].mac_address[k + 2]; |
| | | identifiers[array_index][k] = |
| | | diskInfoPtr[i].disk_sn[k + 2] |
| | | ^ adapterInfoPtr[j].mac_address[k + 2]; |
| | | } |
| | | } |
| | | |
| | | free(diskInfos); |
| | | free(adapterInfos); |
| | | return FUNC_RET_OK; |
| | | end: |
| | | *num_identifiers = min(*num_identifiers, adapter_num * disk_num); |
| | | free(diskInfoPtr); |
| | | free(adapterInfoPtr); |
| | | } |
| | | return function_return; |
| | | } |
| | | |
| | | static FUNCTION_RETURN generate_ethernet_pc_id(PcIdentifier * identifiers, |
| | | unsigned int * num_identifiers, int use_mac) { |
| | | size_t adapters; |
| | | int defined_adapters; |
| | | FUNCTION_RETURN result_adapterInfos; |
| | | unsigned int i, 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) { |
| | | return result_adapterInfos; |
| | | } |
| | | |
| | | defined_adapters = *num_identifiers; |
| | | if (result_adapterInfos == FUNC_RET_OK || result_adapterInfos == FUNC_RET_BUFFER_TOO_SMALL) { |
| | | *num_identifiers = adapters; |
| | | if (identifiers == NULL) { |
| | | return FUNC_RET_OK; |
| | | } else if (adapters > defined_adapters) { |
| | | return FUNC_RET_BUFFER_TOO_SMALL; |
| | | result_adapterInfos=FUNC_RET_OK; |
| | | } |
| | | |
| | | } else { |
| | | defined_adapters = adapters = *num_identifiers; |
| | | adapterInfos = (OsAdapterInfo*)malloc(adapters * sizeof(OsAdapterInfo)); |
| | | result_adapterInfos = getAdapterInfos(adapterInfos, &adapters); |
| | | 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); |
| | | } |
| | | free(adapterInfos); |
| | | return FUNC_RET_OK; |
| | | } |
| | | return result_adapterInfos; |
| | | } |
| | | |
| | | static FUNCTION_RETURN generate_disk_pc_id(PcIdentifier * identifiers, |
| | |
| | | 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; |
| | | } |
| | |
| | | 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)); |
| | | j++; |