From 12de86415acf6e6a6de88565d4a0caf1cb89a924 Mon Sep 17 00:00:00 2001
From: Jan Willamowius <jan@willamowius.de>
Date: 周一, 22 4月 2019 00:56:23 +0800
Subject: [PATCH] intentionally ignore result

---
 src/library/pc-identifiers.c |  296 ++++++++++++++++++++++++++++++----------------------------
 1 files changed, 152 insertions(+), 144 deletions(-)

diff --git a/src/library/pc-identifiers.c b/src/library/pc-identifiers.c
index 07a362a..2a57eaa 100644
--- a/src/library/pc-identifiers.c
+++ b/src/library/pc-identifiers.c
@@ -2,147 +2,141 @@
  * pc-identifiers.c
  *
  *  Created on: Apr 16, 2014
- *      Author: devel
+ *
  */
 
 #include "os/os.h"
 #include "pc-identifiers.h"
 #include <stdlib.h>
 #include <string.h>
-#include <stdbool.h>
+#include <stdio.h>
 #include "base/base64.h"
+#include "base/base.h"
+#ifdef __linux__
+#include <stdbool.h>
 #include <valgrind/memcheck.h>
+#else
+#ifdef __MINGW32__
+#include <windows.h>
+#else
+#include <Windows.h>
+#endif
+#endif
 
-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_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, 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;
+	size_t adapter_num, disk_num;
+	FUNCTION_RETURN result_adapterInfos, result_diskinfos, function_return;
+	unsigned int caller_identifiers, i, j, k, array_index;
+	DiskInfo * diskInfoPtr;
+	OsAdapterInfo *adapterInfoPtr;
 
-	//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 && 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;
-	if (identifiers == NULL) {
-		*num_identifiers = required_id_size;
-		return OK;
-	} else if (required_id_size > defined_identifiers) {
-		return BUFFER_TOO_SMALL;
-	}
+	if (identifiers == NULL || *num_identifiers == 0) {
+		result_adapterInfos = getAdapterInfos(NULL, &adapter_num);
+		if (result_adapterInfos != FUNC_RET_OK) {
+			return generate_disk_pc_id(identifiers, num_identifiers, false);
+		}
+		result_diskinfos = getDiskInfos(NULL, &disk_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 {
+		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(adapterInfoPtr);
+			return generate_disk_pc_id(identifiers, num_identifiers, false);
+		}
+		diskInfoPtr = (DiskInfo*) malloc((*num_identifiers) * sizeof(DiskInfo));
+		disk_num = *num_identifiers;
+		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);
+		}
+		function_return = FUNC_RET_OK;
 
-	//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);
+		caller_identifiers = *num_identifiers;
 		for (i = 0; i < disk_num; i++) {
 			for (j = 0; j < adapter_num; j++) {
-				if (current_identifier > defined_identifiers) {
-					break;
+				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[current_identifier][k] = diskInfos[i].disk_sn[k
-							+ 2] ^ adapterInfos[j].mac_address[k + 2];
-				}
-				encodeStrategy(&identifiers[current_identifier], DEFAULT);
-				current_identifier++;
+				for (k = 0; k < 6; k++)
+					identifiers[array_index][k] = diskInfoPtr[i].disk_sn[k + 2]
+							^ adapterInfoPtr[j].mac_address[k + 2];
 			}
 		}
-		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);
+end:
+#ifdef _MSC_VER
+        *num_identifiers = min(*num_identifiers, adapter_num * disk_num);
+#else
+        *num_identifiers = cmin(*num_identifiers, adapter_num * disk_num);
+#endif
+		free(diskInfoPtr);
+		free(adapterInfoPtr);
 	}
-
-	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) {
-
+	return function_return;
 }
 
 static FUNCTION_RETURN generate_ethernet_pc_id(PcIdentifier * identifiers,
-		unsigned int * num_identifiers, bool use_mac) {
-	size_t adapters;
+		unsigned int * num_identifiers, int use_mac) {
 	FUNCTION_RETURN result_adapterInfos;
 	unsigned int j, k;
-	AdapterInfo *adapterInfos;
+	OsAdapterInfo *adapterInfos;
+	size_t defined_adapters, adapters = 0;
 
-	result_adapterInfos = getAdapterInfos(NULL, &adapters);
-	if (result_adapterInfos != OK) {
-		return result_adapterInfos;
-	}
-
-	int defined_adapters = *num_identifiers;
-	*num_identifiers = adapters;
-	if (identifiers == NULL) {
-		return OK;
-	} else if (adapters > defined_adapters) {
-		return BUFFER_TOO_SMALL;
-	}
-
-	adapterInfos = (AdapterInfo*) malloc(adapters * sizeof(AdapterInfo));
-	result_adapterInfos = getAdapterInfos(adapterInfos, &adapters);
-	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 {
-				//use ip
-				if (k < 4) {
-					identifiers[j][k] = adapterInfos[j].ipv4_address[k];
-				} else {
-					//padding
-					identifiers[j][k] = 42;
-				}
+	if (identifiers == NULL || *num_identifiers == 0) {
+		result_adapterInfos = getAdapterInfos(NULL, &adapters);
+		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));
+		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 {
+						//use ip
+						if (k < 4) {
+							identifiers[j][k] = adapterInfos[j].ipv4_address[k];
+						} else {
+							//padding
+							identifiers[j][k] = 42;
+						}
+					}
 			}
+			result_adapterInfos = (
+					adapters > defined_adapters ?
+							FUNC_RET_BUFFER_TOO_SMALL : FUNC_RET_OK);
+		}
+		free(adapterInfos);
 	}
-	free(adapterInfos);
-	return OK;
+	return result_adapterInfos;
 }
 
 static FUNCTION_RETURN generate_disk_pc_id(PcIdentifier * identifiers,
@@ -150,17 +144,18 @@
 	size_t disk_num, available_disk_info = 0;
 	FUNCTION_RETURN result_diskinfos;
 	unsigned int i, j;
+	int defined_identifiers;
 	char firstChar;
 	DiskInfo * diskInfos;
 
 	result_diskinfos = getDiskInfos(NULL, &disk_num);
-	if (result_diskinfos != OK) {
+	if (result_diskinfos != FUNC_RET_OK) {
 		return result_diskinfos;
 	}
 	diskInfos = (DiskInfo*) malloc(disk_num * sizeof(DiskInfo));
-//memset(diskInfos,0,disk_num * sizeof(DiskInfo));
+	memset(diskInfos,0,disk_num * sizeof(DiskInfo));
 	result_diskinfos = getDiskInfos(diskInfos, &disk_num);
-	if (result_diskinfos != OK) {
+	if (result_diskinfos != FUNC_RET_OK) {
 		free(diskInfos);
 		return result_diskinfos;
 	}
@@ -169,14 +164,14 @@
 		available_disk_info += firstChar == 0 ? 0 : 1;
 	}
 
-	int defined_identifiers = *num_identifiers;
+	defined_identifiers = *num_identifiers;
 	*num_identifiers = available_disk_info;
 	if (identifiers == NULL) {
 		free(diskInfos);
-		return OK;
+		return FUNC_RET_OK;
 	} else if (available_disk_info > defined_identifiers) {
 		free(diskInfos);
-		return BUFFER_TOO_SMALL;
+		return FUNC_RET_BUFFER_TOO_SMALL;
 	}
 
 	j = 0;
@@ -197,7 +192,7 @@
 		}
 	}
 	free(diskInfos);
-	return OK;
+	return FUNC_RET_OK;
 }
 
 /**
@@ -215,6 +210,7 @@
  * @param
  * @return
  */
+
 FUNCTION_RETURN generate_pc_id(PcIdentifier * identifiers,
 		unsigned int * array_size, IDENTIFICATION_STRATEGY strategy) {
 	FUNCTION_RETURN result;
@@ -234,14 +230,19 @@
 	case DISK_NUM:
 		result = generate_disk_pc_id(identifiers, array_size, false);
 		break;
-	case PLATFORM_SPECIFIC:
-		result = generate_platform_specific_pc_id(identifiers, array_size);
+	case DISK_LABEL:
+		result = generate_disk_pc_id(identifiers, array_size, true);
 		break;
 	default:
-		return ERROR;
+		return FUNC_RET_ERROR;
 	}
 
-	if (result == OK && identifiers != NULL) {
+	if (result == FUNC_RET_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 << 5;
@@ -278,27 +279,28 @@
 
 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];
+	char* b64_data = NULL;
 	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,
-			&b64_size);
+	b64_data = base64(concat_identifiers, concatIdentifiersSize, &b64_size);
 	if (b64_size > sizeof(PcSignature)) {
-		return BUFFER_TOO_SMALL;
+		free(b64_data);
+		return FUNC_RET_BUFFER_TOO_SMALL;
 	}
 	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;
+	return FUNC_RET_OK;
 }
 
 FUNCTION_RETURN parity_check_id(PcSignature pc_identifier) {
-	return OK;
+	return FUNC_RET_OK;
 }
 
 FUNCTION_RETURN generate_user_pc_signature(PcSignature identifier_out,
@@ -307,24 +309,28 @@
 	PcIdentifier* identifiers;
 	unsigned int req_buffer_size = 0;
 	result = generate_pc_id(NULL, &req_buffer_size, strategy);
-	if (result != OK) {
+	if (result != FUNC_RET_OK) {
 		return result;
 	}
 	if (req_buffer_size == 0) {
-		return ERROR;
+		return FUNC_RET_ERROR;
 	}
 	req_buffer_size = req_buffer_size < 2 ? 2 : req_buffer_size;
 	identifiers = (PcIdentifier *) malloc(
 			sizeof(PcIdentifier) * req_buffer_size);
 	result = generate_pc_id(identifiers, &req_buffer_size, strategy);
-	if (result != OK) {
+	if (result != FUNC_RET_OK) {
 		free(identifiers);
 		return result;
 	}
+#ifdef __linux__
 	VALGRIND_CHECK_VALUE_IS_DEFINED(identifiers[0]);
 	VALGRIND_CHECK_VALUE_IS_DEFINED(identifiers[1]);
+#endif
 	result = encode_pc_id(identifiers[0], identifiers[1], identifier_out);
+#ifdef __linux__
 	VALGRIND_CHECK_VALUE_IS_DEFINED(identifier_out);
+#endif
 	free(identifiers);
 	return result;
 }
@@ -338,9 +344,9 @@
  */
 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;
+	unsigned char * concat_identifiers = NULL;
 	char base64ids[17];
 	int identifiers_size;
 
@@ -348,13 +354,14 @@
 			&base64ids[8], &base64ids[12]);
 	concat_identifiers = unbase64(base64ids, 16, &identifiers_size);
 	if (identifiers_size > sizeof(PcIdentifier) * 2) {
-		return BUFFER_TOO_SMALL;
+		free(concat_identifiers);
+		return FUNC_RET_BUFFER_TOO_SMALL;
 	}
 	memcpy(identifier1_out, concat_identifiers, sizeof(PcIdentifier));
 	memcpy(identifier2_out, concat_identifiers + sizeof(PcIdentifier),
 			sizeof(PcIdentifier));
 	free(concat_identifiers);
-	return OK;
+	return FUNC_RET_OK;
 }
 
 static IDENTIFICATION_STRATEGY strategy_from_pc_id(PcIdentifier identifier) {
@@ -368,19 +375,20 @@
 	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
 	result = decode_pc_id(user_identifiers[0], user_identifiers[1], str_code);
-	if (result != OK) {
+	if (result != FUNC_RET_OK) {
 		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) {
+			free(calculated_identifiers);
 			return LICENSE_MALFORMED;
 		}
 		if (current_strategy_id != previous_strategy_id) {
@@ -398,9 +406,9 @@
 		for (j = 0; j < calc_identifiers_size; j++) {
 #ifdef _DEBUG
 			printf("generated id: %02x%02x%02x%02x%02x%02x index %d, user_supplied id %02x%02x%02x%02x%02x%02x idx: %d\n",
-					calculated_identifiers[j][0],calculated_identifiers[j][1],calculated_identifiers[j][2],
-					calculated_identifiers[j][3],calculated_identifiers[j][4],calculated_identifiers[j][5],j,
-					user_identifiers[i][0],user_identifiers[i][1],user_identifiers[i][2],user_identifiers[i][3],user_identifiers[i][4],user_identifiers[i][5],i);
+					calculated_identifiers[j][0], calculated_identifiers[j][1], calculated_identifiers[j][2],
+					calculated_identifiers[j][3], calculated_identifiers[j][4], calculated_identifiers[j][5], j,
+					user_identifiers[i][0], user_identifiers[i][1], user_identifiers[i][2], user_identifiers[i][3], user_identifiers[i][4], user_identifiers[i][5], i);
 
 #endif
 			if (!memcmp(user_identifiers[i], calculated_identifiers[j],

--
Gitblit v1.9.1