Обработка параметров командной строки

.486
.model flat, stdcall
option casemap :none
 
        include \masm32\include\windows.inc
 
        include \masm32\include\user32.inc
        include \masm32\include\kernel32.inc
        include \masm32\include\shell32.inc
        include \masm32\include\masm32.inc
 
        include \masm32\macros\macros.asm
 
        includelib \masm32\lib\user32.lib
        includelib \masm32\lib\kernel32.lib
        includelib \masm32\lib\shell32.lib
        includelib \masm32\lib\masm32.lib
.const
        UCSTR   wszCmdLineToArgvError, "Command Line To Argv Error", 0
        UCSTR   wszMsgCmdLine,  "Command Line: '%ls'", 0Dh, 0Ah, 0
        UCSTR   wszMsgARGC,     "Params count: %d", 0Dh, 0Ah, 0
        UCSTR   wszMsgArgN,     "  argv[%d]: '%ls'", 0Dh, 0Ah, 0
        UCSTR   wszWrongNumberOfArgs, 0Dh, 0Ah, "Wrong number of parameters", 0Dh, 0Ah, 0
        UCSTR   wszPressLeftAlt, 0Dh, 0Ah, 0Dh, 0Ah, "Press Left Alt to exit", 0
        ;
        txt2    db      "Blad otwarcia pliku.", NULL
        txt3    db      "Blad odczytu pliku.", NULL
        txt4    db      "Blad alokacji pamieci.", NULL
        fName   db      "plik.txt"
        ;
        STR2    DB      "Проблема с созданием файла",10,13,0
.data
.data?
        ;описатели консольного ввода и вывода
        hConsoleOutput  HANDLE  ?
        hConsoleInput   HANDLE  ?
        ;переменные, касающиеся командной строки
        lpCmdLine       LPWSTR  ?               ;указатель на командную строку
        argc            dd      ?               ;число параметров в командной строке
        argv            LPWSTR  ?               ;указатель на массив указателей параметров командной строки
        ;буфер для ввода и вывода, а также вспомогательная переменная
        Buffer          db      1024 dup(?)
        BufLen          dd      ?
        ;
        hFile           dd      ?
        tmpV            dd      ?
        fSize           dd      ?
        hBuff           dd      ?
        readBytes       dd      ?
        ;
        WriteBytes      DD      ?
        HANDL           DWORD   ?
        HFILEZ          DD      ?
        LENS            DD      ?
.code
 
OneArg  proc
INVOKE CreateFile,ADDR fName, GENERIC_READ, 0, NULL, OPEN_EXISTING,FILE_ATTRIBUTE_ARCHIVE,0
mov hFile, eax
 
    .IF eax==INVALID_HANDLE_VALUE
        INVOKE lstrlen, OFFSET txt2
        INVOKE WriteConsole, hConsoleOutput, OFFSET txt2, eax,ADDR tmpV, NULL
        jmp @end
    .ENDIF
 
    INVOKE GetFileSize,hFile,0
    mov fSize, eax
 
    INVOKE VirtualAlloc,0,eax,MEM_COMMIT,PAGE_READWRITE
    .IF eax == NULL
        INVOKE lstrlen, OFFSET txt4
        INVOKE WriteConsole, hConsoleOutput,OFFSET txt4, eax,OFFSET tmpV, NULL
    .ELSE
        mov hBuff, eax
        INVOKE ReadFile, hFile, eax, fSize, OFFSET readBytes, 0
 
        .IF eax == -1
            INVOKE lstrlen, OFFSET txt3
            INVOKE WriteConsole, hConsoleOutput,OFFSET txt3, eax,ADDR tmpV, NULL
        .ELSE
            INVOKE WriteConsoleW, hConsoleOutput, hBuff, fSize,OFFSET tmpV, NULL
        .ENDIF
    .ENDIF
 
    INVOKE VirtualFree,eax,OFFSET fSize,MEM_RELEASE
    INVOKE CloseHandle, hFile
@end:
        ret
OneArg  endp
 
TwoArgs proc
 
 
EXTERN AllocConsole@0:NEAR
EXTERN FreeConsole@0:NEAR
EXTERN GetStdHandle@4:NEAR
EXTERN ExitProcess@4:NEAR
EXTERN GetCommandLineW@0:NEAR
EXTERN CommandLineToArgvW@8:NEAR
EXTERN WriteConsoleA@20:NEAR
EXTERN lstrlenA@4:NEAR
EXTERN lstrlenW@4:NEAR
EXTERN CreateFileW@28:NEAR
EXTERN CloseHandle@4:NEAR
EXTERN WriteFile@20:NEAR
EXTERN CharToOemA@8:NEAR
 
MOV EDI,argv
ADD EDI,4
MOV EAX,[EDI]
 
PUSH 0
PUSH FILE_ATTRIBUTE_ARCHIVE
PUSH OPEN_ALWAYS
PUSH 0
PUSH 0
PUSH GENERIC_WRITE
PUSH EAX
CALL CreateFileW@28
 
MOV HFILEZ,EAX
 
 
CMP EAX,INVALID_HANDLE_VALUE
JNE PLIKOK
 
 
PUSH OFFSET STR2
PUSH OFFSET STR2
CALL CharToOemA@8
 
 
MOV EBX ,OFFSET STR2
PUSH EBX
CALL lstrlenA@4
 
 
 
PUSH 0
PUSH OFFSET LENS
PUSH EAX
PUSH EBX
PUSH HANDL
CALL WriteConsoleA@20
JMP KONCZ
 
PLIKOK:
 
 
;считываем второй параметр
 
 
MOV EDI,argv
ADD EDI,8
MOV EBX,[EDI]
 
PUSH EBX
CALL lstrlenW@4
 
XOR ESI,ESI
MOV ESI,2
MUL ESI
 
PUSH 0
PUSH OFFSET WriteBytes
PUSH EAX
PUSH EBX
PUSH HFILEZ
CALL WriteFile@20
 
 
 
PUSH HFILEZ
CALL CloseHandle@4
 
KONCZ:
        ret
TwoArgs endp
 
start:
 
        ; получение описателей ввода и вывода консоли
        invoke  GetStdHandle,   STD_INPUT_HANDLE
        mov     hConsoleInput,  eax
        invoke  GetStdHandle,   STD_OUTPUT_HANDLE
        mov     hConsoleOutput, eax
        ;очистка экрана
        invoke  ClearScreen
 
        ;получение указателя на командную строку
        invoke  GetCommandLineW
        mov     [lpCmdLine],    eax
        ;получение количества параметров командной строки
        ;и массива параметров (в виде указателей на Unicode строки)
        invoke  CommandLineToArgvW,     [lpCmdLine], ADDR argc
        mov     [argv], eax
        test    eax,    eax
        jnz     @f
                invoke  WriteConsole, hConsoleOutput, ADDR wszCmdLineToArgvError,\
                        LENGTHOF wszCmdLineToArgvError - 1, ADDR BufLen, NULL
                jmp     @@Exit
        @@:
 
        ;информационный вывод
        ; - командная строка
        ; - количество параметров в командной строке
        ; - список параметров
        invoke  wsprintfW, ADDR Buffer, ADDR wszMsgCmdLine, [lpCmdLine]
        mov     [BufLen],       eax
        invoke  WriteConsoleW, hConsoleOutput, ADDR Buffer,\
                BufLen, ADDR BufLen, NULL
 
        invoke  wsprintfW, ADDR Buffer, ADDR wszMsgARGC, [argc]
        mov     [BufLen],       eax
        invoke  WriteConsoleW, hConsoleOutput, ADDR Buffer,\
                BufLen, ADDR BufLen, NULL
 
        mov     esi,    [argv]
        mov     ecx,    0
        @@:
                push    ecx
                push    esi
                mov     ebx,    [esi+4*ecx]
                invoke  wsprintfW, ADDR Buffer, ADDR wszMsgArgN, ecx, ebx
                mov     [BufLen],       eax
                invoke  WriteConsoleW, hConsoleOutput, ADDR Buffer,\
                        BufLen, ADDR BufLen, NULL
                pop     esi
                pop     ecx
                inc     ecx
                cmp     ecx,    [argc]
        jb      @b
 
        ;распределение задач в зависимости от числа параметров в командной строке
        ;число в argc на 1 больше количества параметров командной строки,
        ;т.к. имя самой программы - это первый параметр (argv[0]) и он всегда имеется
        mov     eax,    [argc]
        .IF eax==2
                call    OneArg
        .ELSEIF eax==3
                call    TwoArgs
        .ELSEIF
                invoke  WriteConsoleW, hConsoleOutput, ADDR wszWrongNumberOfArgs,\
                        LENGTHOF wszWrongNumberOfArgs - 1, ADDR BufLen, NULL
        .ENDIF
 
@@FreeAndExit:
        ;освободить память от массива указателей на строки параметров
        invoke  LocalFree, [argv]
@@Exit:
        ;пауза - ожидание нажатия Left Alt
        invoke  WriteConsoleW, hConsoleOutput, ADDR wszPressLeftAlt,\
                LENGTHOF wszPressLeftAlt - 1, ADDR BufLen, NULL
        @@WaitForLAlt:
                invoke  GetAsyncKeyState, VK_LMENU
                and     eax,    8000h
        jz      @@WaitForLAlt
        ;завершение программы
        invoke  ExitProcess, 0
 
end start