Поиск символа в строке и замена - Assembler - Ответ 636852

.386
.model flat, stdcall
option casemap : none
includelib imp32i.lib
include    def32.inc
 
extrn   MessageBoxA : proc
extrn   ExitProcess : proc
extrn   SetCurrentDirectoryA : proc
extrn   FindFirstFileA : proc
extrn   CloseHandle : proc
extrn   GlobalAlloc : proc
extrn   GlobalLock : proc
extrn   GlobalUnlock : proc
extrn   GlobalFree : proc
extrn   CreateFileA : proc
extrn   ReadFile : proc
extrn   WriteFile : proc
extrn   GetFileSize : proc
extrn   GetModuleHandleA : proc
extrn   FindNextFileA : proc
 
MAX_PATH equ 260
FILE_ATTRIBUTE_NORMAL equ 80h
 
FIND_DATA STRUC
    dwFileAttributes dd ?
    ftCreationTime dq ?
    ftLastAccessTime dq ?
    ftLastWriteTime dq ?
    nFileSizeHigh dd ?
    nFileSizeLow dd ?
    Reserved0 dd ?
    Reserved1 dd ?
    cFileName db MAX_PATH dup(?)
    cAlternateFileName db 14 dup(?)
FIND_DATA ENDS
.data
MB_OK   equ 16
maska   db '*.txt',0
hfind dword ?
FindData FIND_DATA <?>
DIR db 'Тут пишешь нужную директорию',0
TR  db 'Вирус!',0
error3  db 'Операция успешно выполнена!',0
error         db 'Ошибка',0
error1        db 'Файл не найден',0
error2        db 'Операция не может быть выполнена!',0
fileName      dd 0
programHandle dd 0
fileHandle    dd 0
memoryHandle  dd 0
memoryOffset  dd 0
SizeRW        dd 0
.code
    jmp main
program_error proc
    push MB_OK
    push offset TR
    push eax
    push ebx
    call MessageBoxA
    push memoryHandle
    call GlobalUnlock
    push memoryHandle
    call GlobalFree
    push ebx
    call ExitProcess
    ret
program_error endp
CheckSymbol proc
    push edi
    mov al,'s'
    mov ecx,edi
    inc ecx
    mov edi,memoryOffset
loop_check:
    repne scasb
    test ecx,ecx
    jz end_check
    mov byte ptr [edi-1],'$'
    jmp loop_check
end_check:
    pop edi
    ret
CheckSymbol endp
main:
    xor ebx,ebx
    push ebx
    call GetModuleHandleA
    mov programHandle,eax
    push offset DIR
    call SetCurrentDirectoryA
    cmp eax,0
    je none1
    jnc file_ok
none:
    mov eax,offset error1
    call program_error
file_ok:
    push offset FindData
    push offset maska
    call FindFirstFileA
    mov hfind,eax
    cmp eax,-1
    je none
    cmp byte ptr [FindData.cFileName],"."
    je next
next:
    push offset FindData
    push hfind
    call FindNextFileA
    cmp eax,0
    je none
    cmp byte ptr [FindData.cFileName],"."
    je next
    jnc OK
OK:
    lea esi,[FindData.cFileName]
    mov fileName,esi
    push ebx
    push FILE_ATTRIBUTE_ARCHIVE
    push OPEN_EXISTING
    push ebx
    push FILE_SHARE_READ
    push GENERIC_READ
    push esi
    call CreateFileA
    add esp,12
    test eax,eax
    jnz file_opened_OK
none1:
    mov eax,offset error2
    call program_error 
file_opened_OK:
    mov fileHandle,eax
    push ebx
    push eax
    call GetFileSize
    mov edi,eax
    push eax
    push GMEM_MOVEABLE or GMEM_ZEROINIT
    call GlobalAlloc
    mov memoryHandle,eax
    push eax
    call GlobalLock
    mov memoryOffset,eax
    push ebx
    push offset SizeRW
    push edi
    push eax
    push fileHandle
    call ReadFile
    push fileHandle
    call CloseHandle
    call CheckSymbol
    push ebx
    push FILE_ATTRIBUTE_ARCHIVE
    push TRUNCATE_EXISTING
    push ebx
    push FILE_SHARE_WRITE
    push GENERIC_WRITE
    push fileName
    call CreateFileA
    push ebx
    push offset SizeRW
    push edi
    push memoryOffset
    push fileHandle
    call WriteFile
    push fileHandle
    call CloseHandle 
    jnc next        
end main