WDK 왕초보를 위한 개발 가이드 #3 - 문자열 사용

 VC++에서는 문자열을 다룰 때 char나 wchar_t 로 선언합니다.

그러나 드라이버 개발을 하다보면 STRING이나 UNICODE_STRING 로 선언된 것을 많이 볼 수 있는데요.

// 구조체 typedef struct _UNICODE_STRING { USHORT Length; USHORT MaximumLength; wchar_t* Buffer; } UNICODE_STRING; typedef struct _STRING { USHORT Length; USHORT MaximumLength; char* Buffer; } STRING; // 선언 UNICODE_STRING UnicodeString;

특이하게도 문자열을 구조체로 표현합니다.

각각 길이, 최대 길이(NULL을 포함하는), 문자열 포인터를 표현합니다.

DbgPrint에서 구조체를 직접 출력할 수 있는 특이점이 있습니다. 아래 %wZ 출력을 확인해 주세요.

UNICODE_STRING UnicodeString; RtlInitUnicodeString (&UnicodeString, L"Hello"); DbgPrint("%d %d %ws %wZ", UnicodeString.Length, UnicodeString.MaximumLength, UnicodeString.Buffer, &UnicodeString);

문자열의 초기화는 RtlInitUnicodeString()를 사용하며, 아래와 같이 PSEUDO 코드를 작성할 수 있습니다.

UnicodeString->Buffer = 초기화 문자열;

UnicodeString->Length = wcslen(초기화 문자열) ;
UnicodeString->MaximumLength = UnicodeString->Length + sizeof(UNICODE_NULL) ;

아래 코드는 동적으로 할당 및 해제 하는 방법입니다. Buffer만 동적으로 할당하니 결국 malloc()과 같다고 볼 수 있습니다.

VOID AllocateUnicodeString(PUNICODE_STRING us, USHORT Size) { if (!us) return; __try { us->Length = 0; us->MaximumLength = 0; us->Buffer = (PWSTR)ExAllocatePool(NonPagedPool, Size); if (us->Buffer) { us->Length = 0; us->MaximumLength = Size; } } __except(EXCEPTION_EXECUTE_HANDLER) { } } VOID FreeUnicodeString(PUNICODE_STRING us) { if (!us) return; __try { if (us->MaximumLength > 0 && us->Buffer) { ExFreePool(us->Buffer); } us->Length = 0; us->MaximumLength = 0; } __except(EXCEPTION_EXECUTE_HANDLER) { } } PUNICODE_STRING us; AllocateUnicodeString(us, 100 * sizeof(WCHAR)); FreeUnicodeString(us);