Magical Dorothy ::

'Computer/System'에 해당되는 글 7건

2008/01/16 dr.Dorothy How To Get IP & Mac Address (C)
2008/01/14 dr.Dorothy 물리드라이브에 포함된 논리 드라이브 보기
2008/01/02 dr.Dorothy How To Get Physical Drive Size (2)
2008/01/01 dr.Dorothy How To Get Process List
2008/01/01 dr.Dorothy Change Security Descriptor (Registry)

◀ Prev 1 2 Next ▶

How To Get IP & Mac Address (C)

Computer/System | 2008/01/16 08:08

IP Address

int GetIPAddress(char Addr[]) {
	WSAData wsaData;
	char ac[80];

	if(WSAStartup(MAKEWORD(1, 1), &wsaData) != 0)
		return 0;

	if(gethostname(ac, sizeof(ac)) == SOCKET_ERROR)
		return 0;

	struct hostent* phe = gethostbyname(ac);

	if(phe == 0)
		return 0;

	for (int i = 0; phe->h_addr_list[i] != 0; i++) {
		struct in_addr addr;
		memcpy(&addr, phe->h_addr_list[i], sizeof(struct in_addr));
		wsprintfA(Addr, "%s", inet_ntoa(addr));
	}

	WSACleanup();

	return 1;
}

Mac Address

PIP_ADAPTER_INFO pAdapterInfo;
PIP_ADAPTER_INFO pinfo = NULL;

DWORD dwRetVal = 0;

pAdapterInfo = (IP_ADAPTER_INFO *) malloc( sizeof(IP_ADAPTER_INFO) );
ULONG ulOutBufLen = sizeof(IP_ADAPTER_INFO);

if (GetAdaptersInfo( pAdapterInfo, &ulOutBufLen) == ERROR_BUFFER_OVERFLOW) {
	free(pAdapterInfo);
	pAdapterInfo = (IP_ADAPTER_INFO *) malloc (ulOutBufLen); 
}

if ((dwRetVal = GetAdaptersInfo( pAdapterInfo, &ulOutBufLen)) == NO_ERROR) {
	pinfo = pAdapterInfo;
	wsprintf(temp,"%s (%02X:%02X:%02X:%02X:%02X:%02X)",
		pinfo->IpAddressList.IpAddress.String,
		pinfo->Address[0],
		pinfo->Address[1],
		pinfo->Address[2],
		pinfo->Address[3],
		pinfo->Address[4],
		pinfo->Address[5] );
	SetDlgItemText(IDC_ADDRESS, temp);
}

트랙백이 없고, 댓글이 없습니다.

How To Get Physical Drive Size

Computer/System | 2008/01/02 20:06

가끔씩 프로그램을 만들다 보면..
물리드라이브의 전체 용량이 필요한 경우가 있다..

이때 DeviceIoControl() 를 이용하면 쉽게 알 수 있다..

#include <windows.h>
#include <winioctl.h>
#include <stdio.h>

int main(void) {
	HANDLE hDisk;
	DISK_GEOMETRY pdg;

	ULONGLONG DiskSize;
	ULONGLONG TotalSize = 0;

	TCHAR temp[MAX_PATH];
	DWORD i,ret;

	for (i=0; i<4; i++) {
		wsprintf(temp, "\\\\.\\PHYSICALDRIVE%d", i);
		hDisk = CreateFile(	temp,
							GENERIC_READ,
							FILE_SHARE_READ,
							NULL,
							OPEN_EXISTING,
							FILE_FLAG_NO_BUFFERING,
							NULL );

		if (hDisk == INVALID_HANDLE_VALUE)
			break;

		DeviceIoControl(	hDisk,
							IOCTL_DISK_GET_DRIVE_GEOMETRY,
							NULL,
							0,
							&pdg,
							sizeof(pdg),
							&ret,
							(LPOVERLAPPED) NULL );

		DiskSize = pdg.Cylinders.QuadPart *
					(ULONG)pdg.TracksPerCylinder *
					(ULONG)pdg.SectorsPerTrack *
					(ULONG)pdg.BytesPerSector;

		TotalSize += DiskSize;

		CloseHandle(hDisk);

		printf(	"== Physical Drive %d ==\n"
				"Size : %I64d (Bytes)\n"
				"Size : %I64d (GB)\n"
				"Real : %I64d (GB)\n\n",
				i, DiskSize,
				DiskSize / (1000 * 1000 * 1000),
				DiskSize / (1024 * 1024 * 1024));
	}

	printf(	"== Total Size ==\n"
			"Size : %I64d (Bytes)\n"
			"Size : %I64d (GB)\n"
			"Real : %I64d (GB)\n\n",
			TotalSize,
			TotalSize / (1000 * 1000 * 1000),
			TotalSize / (1024 * 1024 * 1024));

	return 0;
}

트랙백이 없고, 댓글 2개가 달렸습니다.

프로세스 리스트의 목록이 필요한 경우..

1. CreateToolhelp32Snapshot 함수를 이용해서 획득하는 방법이 있고..
2. ZwQuerySystemInformation 함수를 이용해서 획득하는 방법이 있다..

전자의 경우는 #include <tlhelp32.h> 라이브러리를 사용하면 되고..
후자의 경우는 Windows Native API 를 사용해야 한다..

먼저.. CreateToolhelp32Snapshot 함수의 예를 보면..
... void ListProcessInfo(void) { HANDLE hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); PROCESSENTRY32 pe32; if (hProcessSnap == INVALID_HANDLE_VALUE) { _tprintf(TEXT("CreateToolhelp32Snapshot Error!!\n")); return ; } pe32.dwSize = sizeof(PROCESSENTRY32); if (!Process32First(hProcessSnap, &pe32)) { _tprintf(TEXT("Process32First Error!!\n")); CloseHandle(hProcessSnap); return ; } do { _tprintf(TEXT("%25s %5d\n"), pe32.szExeFile, pe32.th32ProcessID); } while (Process32Next(hProcessSnap, &pe32)); CloseHandle(hProcessSnap); } ...
이런 식으로 사용한다..

다음으로.. Native API를 사용하는 후자의 경우..
... void ListProcessInfo(void) { { ULONG dwAllocedSize, dwNeeded; PSYSTEM_PROCESSES pProcesses; NTSTATUS Status; dwAllocedSize = 0x1000; while(TRUE) { pProcesses = (PSYSTEM_PROCESSES) VirtualAlloc(NULL, dwAllocedSize, MEM_COMMIT, PAGE_READWRITE); Status = ZwQuerySystemInformation( SystemProcessesAndThreadsInformation, pProcesses, dwAllocedSize, &dwNeeded); if(Status == STATUS_INFO_LENGTH_MISMATCH) { VirtualFree(pProcesses, dwAllocedSize, MEM_DECOMMIT); if(dwNeeded > dwAllocedSize) dwAllocedSize = dwNeeded; else dwAllocedSize += 0x500; } else break; } while(pProcesses->NextEntryDelta != 0) { pProcesses = (PSYSTEM_PROCESSES)((char *)pProcesses + pProcesses->NextEntryDelta); printf("PID:%d - %.*ws\n", pProcesses->ProcessId, pProcesses->ProcessName.Length / 2, pProcesses->ProcessName.Buffer); } } ...
위에 보이는것 처럼.. 먼저, 메모리 크기를 잡아주고..
PSYSTEM_PROCESSES 구조체에서 다음 데이터 포인터를 통해 이동한다..

요즘은 Native API들을 일부로 자주 사용하고 있다..
의식적으로 유니코드에 대해서 항상 생각을 해줘야 하고..
윈도우의 내부함수를 사용한다는 재미가 쏠쏠하다..

트랙백이 없고, 댓글이 없습니다.

Change Security Descriptor (Registry)

Computer/System | 2008/01/01 15:47

안에 있을때 USB 사용시 흔적 지우는게 가장 귀찮았다..
자동으로 지워지게끔 프로그램을 구현하려 해도 보안권한에 막혀서,
키가 지워지지 않았다.
(사실 이때까지만 해도 윈도우 보안체계에는 관심이 없었다..)

그래서!!..

자동으로 지워주게끔 할 프로그램을 개발하면서..
윈도우의 보안체계에 대해서 열공한 결과..

USB흔적을 자동으로 지우는 프로그램을 개발해서 사용했다..

이때 참고한.. 보안 권한을 바꿔주는 코드이다..

#include "stdafx.h"

#include <windows.h>
#include <stdio.h>
#include <aclapi.h>

int __cdecl main( void )
{
	SID_IDENTIFIER_AUTHORITY sia = SECURITY_NT_AUTHORITY;
	PSID pInteractiveSid = NULL;
	PSID pAdministratorsSid = NULL;
	SECURITY_DESCRIPTOR sd;
	PACL pDacl = NULL;
	DWORD dwAclSize;
	HKEY hKey;
	LONG lRetCode;
	BOOL bSuccess = FALSE; // assume this function fails

	// 
	// open the performance key for WRITE_DAC access
	// 
	lRetCode = RegOpenKeyEx(
			HKEY_LOCAL_MACHINE,
			TEXT("System\\CurrentControlSet\\Enum\\USBSTOR"),
			0,
			WRITE_DAC,
			&hKey );

	if (lRetCode != ERROR_SUCCESS) {
		fprintf(stderr, "RegOpenKeyEx error! (rc=%lu)\n", lRetCode);
		return 1;
	}

	// 
	// prepare a Sid representing any Interactively logged-on user
	// 
	if (!AllocateAndInitializeSid(
			&sia,
			1,
			SECURITY_INTERACTIVE_RID,
			0, 0, 0, 0, 0, 0, 0,
			&pInteractiveSid ))
		goto cleanup;

	// 
	// prepare a Sid representing the well-known admin group
	// 
	if (!AllocateAndInitializeSid(
			&sia,
			2,
			SECURITY_BUILTIN_DOMAIN_RID,
			DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0,
			&pAdministratorsSid ))
		goto cleanup;

	// 
	// compute size of new acl
	// 
	dwAclSize = sizeof(ACL) + 2 * (sizeof(ACCESS_ALLOWED_ACE) - sizeof(DWORD))
		+ GetLengthSid(pInteractiveSid) + GetLengthSid(pAdministratorsSid);

	// 
	// allocate storage for Acl
	// 
	pDacl = (PACL)HeapAlloc(GetProcessHeap(), 0, dwAclSize);

	if (pDacl == NULL)
		goto cleanup;

	if (!InitializeAcl(pDacl, dwAclSize, ACL_REVISION))
		goto cleanup;

	// 
	// grant the Interactive Sid KEY_READ access to the perf key
	// 
	if (!AddAccessAllowedAce(
			pDacl,
			ACL_REVISION,
			KEY_ALL_ACCESS,
			pInteractiveSid ))
		goto cleanup;

	// 
	// grant the Administrators Sid GENERIC_ALL access to the perf key
	// 
	if (!AddAccessAllowedAce(
			pDacl,
			ACL_REVISION,
			KEY_ALL_ACCESS,
			pAdministratorsSid ))
		goto cleanup;

	if (!InitializeSecurityDescriptor(&sd, SECURITY_DESCRIPTOR_REVISION))
		goto cleanup;

	if (!SetSecurityDescriptorDacl(&sd, TRUE, pDacl, FALSE))
		goto cleanup;

	// 
	// apply the security descriptor to the registry key
	// 
	lRetCode = RegSetKeySecurity(
			hKey,
			(SECURITY_INFORMATION) DACL_SECURITY_INFORMATION,
			&sd );

	if (lRetCode != ERROR_SUCCESS)
		goto cleanup;

	bSuccess = TRUE; // indicate success

cleanup:

	RegCloseKey(hKey);
	RegCloseKey(HKEY_LOCAL_MACHINE);

	// 
	// free allocated resources
	// 
	if (pDacl != NULL)
		HeapFree(GetProcessHeap(), 0, pDacl);

	if (pInteractiveSid != NULL)
		FreeSid(pInteractiveSid);

	if (pAdministratorsSid != NULL)
		FreeSid(pAdministratorsSid);

	if (!bSuccess)
		return 1;

	return 0;
}

트랙백이 없고, 댓글이 없습니다.

◀ Prev 1 2 Next ▶

Magical Dorothy Do anything what you want. Nobody care who you are.

'Computer/System'에 해당되는 글 7건

How To Get IP & Mac Address (C)

트랙백을 보내세요

물리드라이브에 포함된 논리 드라이브 보기

트랙백을 보내세요

How To Get Physical Drive Size

트랙백을 보내세요

How To Get Process List

트랙백을 보내세요

Change Security Descriptor (Registry)

트랙백을 보내세요

Magical Dorothy - 최근 공지

Notice

Category

Recent Post

Recent Comment

Counter