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

.586
include B:\masm32\include\masm32rt.inc
 
.data?
hInstance dd ?
hFile dd ?
tmpV dd ?
fSize dd ?
hBuff dd ?
readBytes dd ?
 
argc dd ?
argv dd ?
.data
 
txt2 db "Blad otwarcia pliku.", NULL
txt3 db "Blad odczytu pliku.", NULL
txt4 db "Blad alokacji pamieci.", NULL 
fName db "plik.txt"
 
.code
Start:
INVOKE AllocConsole
INVOKE GetStdHandle, STD_OUTPUT_HANDLE
mov hInstance, eax
 
        
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, hInstance, 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, hInstance,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, hInstance,OFFSET txt3, eax,ADDR tmpV, NULL    
        .ELSE
            INVOKE WriteConsoleW, hInstance, hBuff, fSize,OFFSET tmpV, NULL
        .ENDIF
    .ENDIF
 
    INVOKE VirtualFree,eax,OFFSET fSize,MEM_RELEASE  
    INVOKE CloseHandle, hFile
 
 
@end:
 
 
INVOKE FreeConsole
INVOKE ExitProcess,0
 
END Start

.586P
.MODEL FLAT,stdcall
 
include    B:\masm32\include\masm32rt.inc
 
 
 
STD_OUTPUT_HANDLE                    EQU -11
INVALID_HANDLE_VALUE                 EQU -1
GENERIC_WRITE                        EQU 40000000h
OPEN_ALWAYS                          EQU 4
FILE_ATTRIBUTE_ARCHIVE               EQU 20h
 
 
 
 
 
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
 
 
 
 
 
.DATA?
WriteBytes  DD ?
HANDL       DWORD ?
HFILEZ      DD ?
ARGC        DD ?
ARGV        DD ?
LENS        DD ?
 
 
 
.DATA
NazwaPliku DW 300 DUP(0)
STR1       DB "Неправильное количество параметров",10,13,0
STR2       DB "Проблема с созданием файла",10,13,0
 
 
 
.CODE
 
 
Start:
 
 
CALL AllocConsole@0 
 
PUSH STD_OUTPUT_HANDLE  
 
CALL GetStdHandle@4
 
MOV HANDL,EAX
 
 
CALL GetCommandLineW@0
 
 
PUSH OFFSET ARGC
PUSH EAX
CALL CommandLineToArgvW@8
MOV  ARGV,EAX
 
 
MOV EBX,3
 
CMP ARGC,EBX
JNE ZLEPARAMETRY
 
 
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
 
JMP KONCZ
 
 
ZLEPARAMETRY:
 
PUSH OFFSET STR1
PUSH OFFSET STR1
CALL CharToOemA@8
 
MOV EBX ,OFFSET STR1
PUSH EBX
CALL lstrlenA@4
 
 
PUSH 0
PUSH OFFSET LENS
PUSH EAX
PUSH EBX
PUSH HANDL
CALL WriteConsoleA@20
 
KONCZ:
 
CALL FreeConsole@0
 
PUSH 0
 
CALL ExitProcess@4
 
RET
 
END Start

@@OneArg:
........................
jmp @@Exit

@@TwoArg:
........................
jmp @@Exit

.code
.......ТУТ, ТУТ, ТУТ
mov eax, [argc]

        ;получение указателя на командную строку
        invoke  GetCommandLineW
        mov     [lpCmdLine],    eax
        ;получение количества параметров командной строки
        ;и массива параметров (в виде указателей на Unicode строки)
        invoke  CommandLineToArgvW,     [lpCmdLine], ADDR argc
        mov     [argv], eax
        test    eax,    eax
        jnz     @f
                invoke  WriteConsole, hConsoleOutput, ADDR aszCmdLineToArgvError,\
                        LENGTHOF aszCmdLineToArgvError - 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]
        invoke  wsprintfW, ADDR Buffer, ADDR wszMsg0roArg, esi
        mov     [BufLen],       eax
        invoke  WriteConsoleW, hConsoleOutput, ADDR Buffer,\
                BufLen, ADDR BufLen, NULL
       mov     esi,    [argv]
        invoke  MessageBoxW, NULL, [esi], addr wszMsgBoxCapt, MB_OK

        mov     esi,    [argv]
        add     esi,    8
        mov     ebx,    [esi]
        invoke  wsprintfW, ADDR Buffer, ADDR wszMsgArgN, 2, ebx
        mov     [BufLen],       eax
        invoke  WriteConsoleW, hConsoleOutput, ADDR Buffer,
                BufLen, ADDR BufLen, NULL