From 1922d8c2cf7dcaacafd17394849577794b1f10eb Mon Sep 17 00:00:00 2001
From: open-license-manager <rillf@maildrop.cc>
Date: 周六, 13 9月 2014 06:24:55 +0800
Subject: [PATCH] signature windows

---
 src/library/os/win/os-win.c |  152 ++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 144 insertions(+), 8 deletions(-)

diff --git a/src/library/os/win/os-win.c b/src/library/os/win/os-win.c
index 088b97b..6fd115d 100644
--- a/src/library/os/win/os-win.c
+++ b/src/library/os/win/os-win.c
@@ -1,28 +1,164 @@
 #include <Windows.h>
+#include <iphlpapi.h>
+//definition of size_t
+#include <stdlib.h>
+#include "../../base/logger.h"
 #include"../os.h"
+#pragma comment(lib, "IPHLPAPI.lib")
 
 FUNCTION_RETURN getOsSpecificIdentifier(unsigned char identifier[6]){
-
+	return FUNC_RET_NOT_AVAIL;
 }
 
 FUNCTION_RETURN getMachineName(unsigned char identifier[6]) {
-	char* buffer[MAX_COMPUTERNAME_LENGTH + 1];
+	FUNCTION_RETURN result = FUNC_RET_ERROR;
+	char buffer[MAX_COMPUTERNAME_LENGTH + 1];
 	int bufsize = MAX_COMPUTERNAME_LENGTH + 1;
 	BOOL cmpName = GetComputerName(
-		buffer,&bufsize);
-	strncpy(identifier, buffer, 6);
+		buffer, &bufsize);
+	if (cmpName){
+		strncpy(identifier, buffer, 6);
+		result = FUNC_RET_OK;
+	}
+	return result;
 }
 
 FUNCTION_RETURN getCpuId(unsigned char identifier[6]) {
+	return FUNC_RET_NOT_AVAIL;
 }
 
 void os_initialize() {
 }
 
-#define MAX_UNITS 20
+//http://www.ok-soft-gmbh.com/ForStackOverflow/EnumMassStorage.c
+//http://stackoverflow.com/questions/3098696/same-code-returns-diffrent-result-on-windows7-32-bit-system
+#define MAX_UNITS 30
 FUNCTION_RETURN getDiskInfos(DiskInfo * diskInfos, size_t * disk_info_size) {
+	DWORD FileMaxLen;
+	int ndrives = 0;
+	DWORD FileFlags;
+	char volName[_MAX_FNAME], FileSysName[_MAX_FNAME];
+	char* szSingleDrive;
+	DWORD volSerial = 0;
+	BOOL success;
+	UINT driveType;
+	DWORD dwSize = MAX_PATH;
+	char szLogicalDrives[MAX_PATH] = { 0 };
+
+	FUNCTION_RETURN return_value;
+	DWORD dwResult = GetLogicalDriveStrings(dwSize, szLogicalDrives);
+
+	if (dwResult > 0 && dwResult <= MAX_PATH)
+	{
+		return_value = FUNC_RET_OK;
+		szSingleDrive = szLogicalDrives;
+		while (*szSingleDrive && ndrives < MAX_UNITS)
+		{
+
+			// get the next drive
+			driveType = GetDriveType(szSingleDrive);
+			if (driveType == DRIVE_FIXED){
+				success = GetVolumeInformation(szSingleDrive, volName, MAX_PATH, &volSerial,
+					&FileMaxLen, &FileFlags, FileSysName, MAX_PATH);
+				if (success) {
+					LOG_INFO("drive         : %s\n", szSingleDrive);
+					LOG_INFO("Volume Name   : %s\n", volName);
+					LOG_INFO("Volume Serial : 0x%x\n", volSerial);
+					LOG_DEBUG("Max file length : %d\n", FileMaxLen);
+					LOG_DEBUG("Filesystem      : %s\n", FileSysName);
+					if (diskInfos != NULL && * disk_info_size < ndrives){
+						strncpy(diskInfos[ndrives].device, volName, MAX_PATH);
+						strncpy(diskInfos[ndrives].label, FileSysName, MAX_PATH);
+						diskInfos[ndrives].id = ndrives;
+						diskInfos[ndrives].preferred = (strncmp(szSingleDrive, "C", 1) != 0);
+
+					}
+					ndrives++;
+				}
+				else {
+					LOG_WARN("Unable to retrieve information of '%s'\n", szSingleDrive);
+				}
+			}
+			LOG_INFO("This volume is not fixed : %s, type: %d\n", szSingleDrive);
+			szSingleDrive += strlen(szSingleDrive) + 1;
+		}
+	}
+
+	if (*disk_info_size >= ndrives){
+		return_value = FUNC_RET_BUFFER_TOO_SMALL;
+	}
+	return return_value;
 }
 
-FUNCTION_RETURN getAdapterInfos(AdapterInfo * adapterInfos,
-	size_t * adapter_info_size) {
-}
\ No newline at end of file
+static int translate(char ipStringIn[16], unsigned char ipv4[4]) {
+	char *str2;
+
+	size_t index = 0;
+
+	str2 = ipStringIn; /* save the pointer */
+	while (*str2) {
+		if (isdigit((unsigned char)*str2)) {
+			ipv4[index] *= 10;
+			ipv4[index] += *str2 - '0';
+		}
+		else {
+			index++;
+		}
+		str2++;
+	}
+	return 0;
+}
+
+//http://stackoverflow.com/questions/18046063/mac-address-using-c
+//TODO: count only interfaces with type (MIB_IF_TYPE_ETHERNET IF_TYPE_IEEE80211)
+FUNCTION_RETURN getAdapterInfos(OsAdapterInfo * adapterInfos, size_t * adapter_info_size) {
+	DWORD dwStatus;
+	unsigned int i = 0;
+	FUNCTION_RETURN result;
+	PIP_ADAPTER_INFO pAdapterInfo, pAdapter = NULL;
+	//IP_ADAPTER_INFO AdapterInfo[20];              // Allocate information for up to 16 NICs
+	DWORD dwBufLen = 20; //sizeof(AdapterInfo);         // Save the memory size of buffer
+
+	i = 3;
+	do{
+		pAdapterInfo = (PIP_ADAPTER_INFO)malloc(sizeof(IP_ADAPTER_INFO)*dwBufLen);
+		dwStatus = GetAdaptersInfo(               // Call GetAdapterInfo
+			pAdapterInfo, // [out] buffer to receive data
+			&dwBufLen   // [in] size of receive data buffer
+			);
+		dwBufLen = dwBufLen / sizeof(IP_ADAPTER_INFO);
+		if (dwStatus != NO_ERROR){
+			free(pAdapterInfo);
+		}
+	} while (dwStatus == ERROR_BUFFER_OVERFLOW && i-- > 0);
+
+	if (dwStatus != ERROR_BUFFER_OVERFLOW){
+		return FUNC_RET_ERROR;
+	}
+	if (adapterInfos == NULL || *adapter_info_size == 0){
+		*adapter_info_size = dwBufLen;
+		free(pAdapterInfo);
+		return FUNC_RET_BUFFER_TOO_SMALL;
+	}
+
+	memset(adapterInfos, 0, *adapter_info_size);
+	pAdapter = pAdapterInfo;
+	i = 0;
+	result = FUNC_RET_OK;
+	while (pAdapter) {
+		strncpy(adapterInfos[i].description, pAdapter->Description, min(sizeof(adapterInfos->description), MAX_ADAPTER_DESCRIPTION_LENGTH));
+		memcpy(adapterInfos[i].mac_address, pAdapter->Address, 8);
+		translate(pAdapter->IpAddressList.IpAddress.String, adapterInfos[i].ipv4_address);
+		adapterInfos[i].type = IFACE_TYPE_ETHERNET;
+		i++;
+		pAdapter = pAdapter->Next;
+		if (i == *adapter_info_size){
+			result = FUNC_RET_BUFFER_TOO_SMALL;
+			break;
+		}
+	}
+	free(pAdapterInfo);
+	*adapter_info_size = i;
+	return result;
+}
+

--
Gitblit v1.9.1