гос / sswlect / Лекції / sswlect

Tools for Windows та lib-файлами зі складу Windows Driver Kit.

Заголовний файл windows64.inc можна отримати трохи виправивши windows32.inc зі складу MASM32, зокрема, перевизначивши деякі типи даних. Транслятор ml64.exe не підтримує INVOKE і тому inc-файли з прототипами Win API не потрібні. Деяка частина макровизначень Strings.mac із складу KMD Kit прекрасно обробляється препроцесором ml64.exe і залишається працездатною. Доведется, звичайно, відмовитися від функцій стандартної бібліотеки masm32.lib. Транслятор ml64.exe не підтримує синтаксисичні засоби високого рівня .IF-.ELSEIF-.ELSE-ENDIF, .WHILE-.ENDW, .REPEAT-.UNTIL,

SWITCH-CASE-ENDSW.

; Файл lst64.inc

UNICODE equ 1 include windows64.inc

include C:\masm64\macros\Strings.mac includelib shell32.lib

extrn AllocConsole:proc extrn GetStdHandle:proc

extrn GetCommandLineW:proc extrn CommandLineToArgvW:proc extrn FindFirstFileW:proc

extrn LocalFree:proc

extrn FileTimeToSystemTime:proc

61

mov ecx, STD_OUTPUT_HANDLE call GetStdHandle

mov hStdOut, rax

call GetCommandLineW lea rdx,argc

mov rcx,rax

call CommandLineToArgvW mov rsi,[rax+8]

mov rdi,rax lea rdx,DirInfo or rsi,rsi

jnz @F

mov rsi,$CTW0("*.*") @@: mov rcx,rsi

call FindFirstFileW mov hFindFile,rax mov rcx,rdi

62

call LocalFree

cmp hFindFile,INVALID_HANDLE_VALUE jnz next

mov rcx,-1 jmp exit

next: mov edx,DirInfo.dwFileAttributes lea rsi,FileAttr

lea rdi,FileInfoLine mov bx,002Dh ; "-" mov rcx,8

@@: lodsw shl dl,1

cmovnc ax,bx stosw

loop @B mov ax,20h stosw

lea rdx,DateTimeInfo

lea rcx,DirInfo.ftCreationTime call FileTimeToSystemTime movzx eax, DateTimeInfo.wDay mov dl,10

div dl

or ax,3030h shl eax,8 shr ax,8

mov dword ptr [rdi],eax mov word ptr [rdi+4],2Fh

movzx eax, DateTimeInfo.wMonth mov dl,10

div dl

or ax,3030h shl eax,8

63

shr ax,8

mov dword ptr [rdi+6],eax mov word ptr [rdi+10],2Fh lea rdi,[rdi+12+6]

mov ax,DateTimeInfo.wYear mov ecx,4

mov bx,10 @@: xor edx,edx

div bx

or dx,30h mov [rdi],dx sub rdi, 2 or ax,ax

jnz @B

lea rdi,[rdi+10] mov rsi,rdi mov ax,20h mov rcx,11 rep stosw

mov eax,DirInfo.nFileSizeLow mov edx,DirInfo.nFileSizeHigh xor rbx,rbx

@@: or edx,edx jz @F

inc rbx

shrd eax,edx,10 shr edx,10

jmp @B

@@: lea rcx,FileSizeNorm mov rbx,[rcx+rbx*8] mov [rdi],rbx

lea rdi,[rsi+20] mov ecx,10

64

@@: xor edx,edx div ecx

or dx,30h mov [rdi],dx sub rdi, 2 or eax,eax jnz @B

lea rbx,FileInfoLine

lea rdx,DirInfo.cFileName mov rcx,rbx

call lstrcatW mov rcx,rbx call lstrlenW

mov qword ptr [rbx+rax*2],0A000Dh mov qword ptr [rsp+20h],NULL

lea r9, NumberOfBytesWritten lea r8,[rax+2]

mov rdx,rbx mov rcx,hStdOut

call WriteConsoleW lea rdx,DirInfo mov rcx,hFindFile call FindNextFileW or rax,rax

jnz next xor rcx, rcx

exit: call ExitProcess main endp

end

Трансляція і лінкування

ML64.EXE /c /nologo /IC:\MASM64\INCLUDE lst64.asm

LINK.EXE /NOLOGO /ENTRY:main /SUBSYSTEM:CONSOLE /RELEASE

/DEFAULTLIB:kernel32.lib /LIBPATH:"C:\MASM64\LIB" lst64.obj

65

3.2.2.3 Основні функції для роботи з файлами

Під основними функціями тут розуміються ті, які дозволяють управляти файловою системою й одержувати доступ до вмісту файлів. Найперша з них це функція CreateFile, опис всіх параметрів якої займе з десяток сторінок. Тут будуть наведені тільки описи прототипів набору основних функцій для роботи з файлами з коротким описом параметрів. За детальним описом всіх параметрів варто звертатися до літератури або MSDN, хоча в багатьох випадках призначення того або іншого параметра цілком очевидно.

За допомогою функції CreateFile можна виконувати створення нового файлу, відкриття існуючого файлу або каталогу, зміну довжини існуючого файлу, відкриття фізичного диска, тому, консолі, послідовного порту, каналу, поштового слота, пристрою (в Windows 9x завантаження драйвера пристрою).

66

__out_opt LPDWORD lpNumberOfBytesWritten, // адреса слова, у якому

// буде збережена кількість // записаних байт

__inout_opt LPOVERLAPPED lpOverlapped ); // адреса структури типу

// OVERLAPPED

Функція SetFilePointer дозволяє виконати установку поточної файлової позиції для виконання операції вводу/виводу:

__out_opt LPDWORD lpFileSizeHigh); // адреса старшого слова

// для розміру файлу повертає молодше 32-розрядне слово 64-розрядного розміру файлу з

ідентифікатором hFile. Старше слово розміру файлу записується в змінну типу DWORD, адреса якої передається функції через параметр lpFileSizeHigh. При помилці GetFileSize повертає -1 (INVALID_VALUE_HANDLE)

67

Приклад 3 Консольний додаток CatFile.exe, що виконує конкатенацію 2-х файлів (додавання вмісту другого файлу в кінець першого). Параметри передають в командному рядку: CatFile.exe [path]file1 [path]file2.

Загальний алгоритм

1.Отримати параметри командного рядка.

2.Оскільки у file1 буде виконуватися запис, то зняти можливий атрибут «тільки для читання».

3.Відкрити file1 і переставити файловий покажчик в кінець файла

4.Відкрити file2.

5.Виконувати послідовне читання з file2 в буфер та запис з буфера в file1.

6.Закрити файли.

Очевидно, що під час виконання операцій читання/запису найбільш зручною кількістю байтів для обміну з дисковою підсистемою буде число, кратне одиниці розподілу дискового простору (кластера). Отже, і розмір буфера в програмі треба визначати таким що дорівнює або кратним розміру кластера. Але цей розмір заздалегідь невідомий і до того ж може розрізнятися для різних томів. Можна визначити в секції неініціалізованих даних дуже великий буфер, а потім, після одержання розміру кластера, використовувати лише його частину. Але в цій програмі взагалі ніяких буферів під файлові операції не резервується, а необхідна пам'ять буде отримана динамічно за допомогою виклику VirtualAlloc після визначення розміру кластера того диска, на якому розташовано файл, до якого буде виконуватися запис.

; CatFile.inc

include windows.inc include kernel32.inc include masm32.inc

include C:\MASM32\MACROS\strings.mac includelib kernel32.lib

includelib masm32.lib Main PROTO

.data?

69

; Буфери для отримання і зберігання 1-го та 2-го параметрів командного рядка

; CatFile.asm

.386

.model flat, stdcall option casemap:none include CatFile.inc

.code

start:

invoke Main

invoke ExitProcess,0 Main proc

2-й параметр командного рядка в буфер szFileName2

invoke GetCL,2, ADDR szFileName2

70