Поиск файлов

.MODEL Tiny
.286
.CODE
ORG 100h
LOCALS
 
union_string macro s1,s2
        mov ah,2h
        mov dx,10
        int 21h
 
        mov ah,9h
        mov dx,offset s1
        int 21h
 
        mov ah,2h
        mov dx,10
        int 21h
 
        mov ah,0Ah
        mov dx,offset s2
        int 21h
        jmp start_command
endm
 
 
PathLen     =   200            ; ???????????? ???? ? ?????/????????
 
;????? ????????? ??????/?????????
faReadOnly  =   1
faHidden    =   2
faSystem    =   4
;faVolume   =   8
faDirectory =   10h
faArchive   =   20h
faAnyFile   =   faReadOnly+faHidden+faSystem+faArchive
faAnyDir    =   faAnyFile+faDirectory
;faAnithing =   faAnyDir+faVolume
 
;???????? ?????? DTA
DTAFileInfo struc
dtaReserved db  21 dup (?)
dtaAttr     db  ?
dtaTime     dw  ?
dtaDate     dw  ?
dtaSize     dd  ?
dtaName     db  13 dup (?)
ends
 
Start:
        mov ah,9h
        mov dx,offset message
        int 21h
    start_command:
        mov ah,2h
        mov dx,10
        int 21h
 
        mov ah,9h
        mov dx,offset enter_command
        int 21h
 
        mov ah,2h
        mov dx,10
        int 21h
 
        mov ah,0Ah
        mov dx,offset command_buf
        int 21h
 
        mov bl, [command_buf+1]
        cmp bl,3
        ;je extation
        cmp bl,4
        je go_to_read_only
        cmp bl,5
        je temp_start_search
        cmp bl,6
        je go_to_hidden_system
        cmp bl,7
        je go_to_archive_
        cmp bl,8
        ;je file_name
        cmp bl,9
 
 
        mov ah,2h
        mov ah,10
        int 21h
    und:
        mov ah,2h
        mov dx,10
        int 21h
 
        mov ah,9h            ;??????? ?? ????? ?????? 'Undefind command'
        mov dx,offset error_command
        int 21h
        jmp start_command
    go_to_read_only:
        jmp read_only
    go_to_hidden_system:
        jmp hidden_system
    go_to_archive_:
        jmp archive_
    temp_start_search:
        jmp start_search
    extation:
        mov di,offset ExtationCom   ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,3        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        jne und
        union_string enter_ext,ExtString,NameString
    file_name:
        mov di,offset NameCom   ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,8        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        jne und
        union_string enter_name,NameString
 
    change_dir:
        mov di,offset ChangeDir     ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,9        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        jne go_to_und
        union_string enter_dir,StartPath_
 
        jmp start_command
 
    read_only:
        mov di,offset Exit  ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,4        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        je go_to_end_progr
 
        mov di,offset Read  ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,4        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        jne go_to_und
        mov Atribute,1
        jmp start_command
    go_to_und:
        jmp und
    hidden_system:
        mov di,offset Hidden    ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,6        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        jne system_
        mov Atribute,2
        jmp start_command
 
    system_:
        mov di,offset System    ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,6        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        jne go_to_und
        mov Atribute,3
        jmp start_command
    archive_:
        mov di,offset Archive   ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,7        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        jne go_to_und
        mov Atribute,4
        jmp start_command
    go_to_end_progr:
        jmp end_progr
 
 
    start_search:
        mov di,offset StartSearch   ;?????- ?????????? ??????
        mov si,offset command_buf+2h ;? si ? di ?????? ?????
        mov cx,5        ;???????????? ?????
        cld             ;??????????? - ??????
        repe cmpsb          ;?????????
        jne go_to_und
        mov ah,2h
        mov dx,10
        int 21h
 
 
    start_search_:
        ;push Atribute
        push    offset MyFunc
 
        call ScanTree
 
        int 20h
 
MyFunc      proc
        ;lea di,ScanPath
        ;mov    cx,[di+2h]
        lea di,ScanPath   ; DI = ????? ?????? ScanPath
        push    di
        mov cx, PathLen
        xor ax,ax
        ;mov al,'$'
        repne   scasb          ; ???? ????? ??????
        dec di
        mov word ptr [di-1],0A0Dh ; ?????????? ???? CR,LF...
        mov byte ptr [di+1],'$' ; ...? ?????? ??????? (??? ?-?? 9)
        mov ah,9
        pop dx
        int 21h            ; ??????? ?????? ?? ?????
 
        mov byte ptr [di-1],'\'
        ;mov    word ptr [di],0A0Dh ; ?????????? ???? CR,LF...
        mov byte ptr [di],0; ??????????????? ?????? ScanPath
                               ; (?.?. ???????? ?? ??????)
        ret
 
 
MyFunc      endp
 
 
 
; ????????? ?????? ???? ???????????? ????????? ????????.
; ????????? ????? ???????? ? ? ?????????? ??????? DOS EXE !!!
; ?? ?????:
; - ?????????? StartPath (??? ?? ScanPath) ???????? ??? ?????????? ????????.
  ; ??? ???????? ?????? ????????????? ???????? ?????? ('\') !
; - ? ????? ???????? (word) near-???????, ??????? ????? ?????????? ??? ???????
  ; ?????????? ????????. ??? ???? ??? ????? ???????????? ASCIIZ-?????? ScanPath,
  ; ??????? ???????? ?????? ??? ?????????? ???????? ?? ?????? ?? ?????, ?? ??
  ; ?????? ???????? ??? ??????! ??? ????????????? ??????? ?????? ???????????
  ; ?????? ? ?????-?????? ?????, ??? ????? ?????? ? ??? ???, ??? ????? :)
  ; ??????? ????? ?? ?????? ???????????? ????????????? DTA, ??? ?????????????
  ; ????? ??????? ???? ????? DTA. ????? ???????? ?????? ??????????????? ???
  ; ?????? ?? ?????, ??? ? ???????? DS, ES, ???? ??? ???? ????????.
; ?? ??????:
; - DTA ? ???????? DS,BP ?????????.
; - ?????? StartPath/ScanPath ???????? ?????? ??? ?????????? ?????????? ????????.
; - ES:DI ????????? ?? ????? ?????? StartPath/ScanPath, ??????? ???? ??? ??????
  ; ??????? (?.?. ????? ??????? xor al,al / stosb ? ????? ??????? ScanTree).
ScanTree    proc pascal
arg     Func:word
local       OldDTA:dword, NewDTA:DTAFileInfo, atr:word
; ES ?????? ????? ??????????????? DS !!
 
 
        ;pop atr
        push    ds             ; ????????? DS
        mov ah,2Fh
        int 21h            ; ???????? ????? DTA
        mov word ptr OldDTA[0],bx ; ????????? ???...
        mov word ptr OldDTA[2],es ; ...? ?????????? OldDTA
        pop es             ; ??????????????? DS ? ??????? ES (ES=DS)
 
        lea di,ScanPath
        mov cx,PathLen
        xor ax,ax
        ;mov al, '$'
        cld                    ; ?????? ??????? ?????? scasb
        repne   scasb          ; ???? ? ????? ???????? ??????? ???? (????? ??????)
        dec di
        push    di             ; ????????? ????????? ?? $
 
        push    ds es          ; ????????? DS ? ES
        call    Func           ; ???????? ??????? ??? ?????????? ????????
 
        mov ah,1Ah
        push    ss
        pop ds             ; DS = SS (??????? NewDTA)
 
        lea dx,NewDTA
        int 21h            ; ????????????? ???? DTA
        pop es ds          ; ??????????????? DS ? ES (ES=DS)
 
        mov ah,4Eh         ; ????????? ? ?????? ??????
        xor cx,cx
        mov cx,faAnyDir   ; ???????? (???, ????? Volume label)
        lea dx,ScanPath
        lea si,ScanMask    ; SI = ????? ????? ??????
    no_scan_mask:
 
        pop di             ; ??????????????? DI (????????? ?? ????? ??????)...
        push    di             ; ...? ??? ?? ????????? ???
        cld                    ; ?????? ??????? ?????? lodsb/stosb
@@CopyMask: lodsb                  ; ?????? ? AL ?????? ?? ?????? [SI]
        stosb                  ; ? ?????????? ??? ? ????? ScanPath
        or  al,al          ; ????? ?????? [SI] (AL=0) ?
        ;cmp al,'$'
        jne @@CopyMask     ; ???, ???????? ????????? ??????
@@FindNext:
        int 21h            ; ????????/?????????? ?????
        jc  go_to_@@NotFound     ; ???????, ???? ?????? ?? ??????? (CF=1)
 
        push    ss
        pop ds             ; DS = SS (??????? NewDTA)
 
        push si
        push ax
        lea si,Atribute
        lodsw
        mov atr,ax
        pop ax
        pop si
 
        cmp atr,1
        je testing0
 
        cmp atr,2
        je testing1
 
        cmp atr,3
        je testing2
 
        cmp atr,4
        je testing3
 
        cmp atr,5
        je testing4
 
        cmp atr,6
        je testing5
 
        cmp atr,7
        je testing6
 
        ;ret
 
 
testing0:
        test    NewDTA.dtaAttr,faReadOnly
        jmp continue
 
testing1:
        test    NewDTA.dtaAttr,faHidden
        jmp continue
 
testing2:
        test    NewDTA.dtaAttr,faSystem
        jmp continue
 
testing3:
        test    NewDTA.dtaAttr,faArchive
        jmp continue
 
testing4:
        test    NewDTA.dtaAttr,faAnyFile
        jmp continue
 
testing5:
        test    NewDTA.dtaAttr,faAnyDir
        jmp continue
 
go_to_@@NotFound:
        jmp @@NotFound
 
testing6:
        test    NewDTA.dtaAttr,faDirectory; ????????? - ??????? ?? ??? ?????? !!!!!!!
 
 
continue:
        jz  @@PrepareNext  ; ???? ???, ????????? ?????????? ?????
        lea si,NewDTA.dtaName ; ????? ????? ?????????? ?????/????????
        lodsb                  ; ?????? ?????? ??????
        cmp al,'.'         ; ??? ???????? '.' ? '..'? (??? ????? ?? ?????)
        je  @@PrepareNext  ; ???? ??? ???, ????????? ?????????? ?????
 
        pop di             ; ??????????????? DI (????????? ?? ????? ??????)...
        push    di             ; ...? ??? ?? ????????? ???
@@CopyPath:
        stosb                  ; ?????????? ?????? ?????? [SI] ? ????? ScanPath
        lodsb                  ; ?????? ? AL ?????? ?? ?????? [SI]
        or  al,al          ; ????? ?????? [SI] (AL=0) ?
        ;cmp al,'$'
        jne @@CopyPath     ; ???, ???????? ????????? ??????
        ;mov ah,'$'
        mov ax,'\'         ;  AH = 0, AL = '\'
        stosw                  ; ?????????? ???? ? ???? ? ????? ??????
 
        push    es
        pop ds             ; DS = ES
 
 
        push    Func
        call    ScanTree       ; ???????? ???? ??????????
        push    ss
        pop ds             ; DS = SS (??????? NewDTA)
 
@@PrepareNext:
        mov ah,4Fh         ; ????????? ? ??????? 4Fh
        lea dx,NewDTA
        jmp @@FindNext     ; ?????????? ?????
@@NotFound:
 
        pop di             ; ??????? ?? ????? DI
        mov ah,1Ah
        lds dx,OldDTA
        int 21h            ; ??????????????? DTA
 
        push    es
        pop ds             ; ??????????????? ?????????????? DS
 
        ;call Func
        ret                    ; ??????? ?? ?????????
ScanTree    endp
end_progr:
 
        ; mov ax,4c00h
        ; int 21h
        mov ah,4ch
        int 21h
.DATA
NameString  db  10, ?, 8 dup (?)
ExtString   db  5, ?, 3 dup (?)
 
command_buf db  12, ?, 12 dup (?)
ExtationCom db  'ext'                ; ??????? ??????? ??????????
NameCom     db  'filename'           ; ??????? ??????? ?????
Hidden      db  'hidden'             ; ?????? ??????? ?????
Read        db  'read'               ; ?????? ????????? ?????? ??? ?????? ?????
System      db  'system'             ; ?????? ????????? ?????
Archive     db  'archive'            ; ?????? ???????? ?????
StartSearch db  'start'              ; ?????? ??????
Exit        db  'exit'               ; ????? ??????
ChangeDir   db  'changedir'             ; ????? ???????? ??????
error_command db 'Undefined command','$'
enter_command db 'Enter command: ','$'
enter_ext   db 'Enter extation of file:','$'
enter_name  db 'Enter name of file: ','$'
enter_disk  db 'Enter disk to search: ','$'
enter_dir   db 'Enter direction to search: ','$'
ScanMask    db  '*.*',0                ; ????? ????????? ??? ??????
noScanMask  db  '*.*',0                ; ????? ????????? ??? ??????
AllPath     db  PathLen dup (?)        ; ?????? ???? ?????
StartPath   db  'C:\',0
StartPath_  db  100, ?, 100 dup (?)        ; ????????? ???????
ORG StartPath                          ; ????????? StartPath ? ScanPath
ScanPath    db  PathLen dup (?); ASCIIZ-??? ?????????? ????????
 
Atribute    dw  5h
 
message  db 78 dup (' '), 10, 13
         db 26, '  ', 'Welcome to File Finder', '  ', 27, 10, 13
         db 78 dup (' '), 10, 13
         ;db 218, 27 dup (196), 191, 10, 13
         db 'Results will be save in results.txt in program dirrectory. (Warning! If results.txt already exists it will be rewrite)',  10, 13
         db 78 dup (' '), 10, 13
         db 'Comands:', 10, 13
         db 'filename - enter name of file', 10, 13
         db 'ext - enter extation of file', 10, 13
         db 'read - read only files', 10, 13
         db 'system - system files and catalogs', 10, 13
         db 'hidden - hidden files and catalogs', 10, 13
         db 'archive - ordinary files and catalogs', 10, 13
         db 'archive - ordinary files and catalogs', 10, 13
         db 'changedir - change disk and directory (format c:\directory1\)', 10, 13
         db 'start - start search', 10, 13
         db 78 dup (' '), 36
 
END     Start

; tasm /m scantree.asm
; tlink /t /x scantree.obj
 
.MODEL Tiny
.286
.CODE
ORG 100h
 
LOCALS
 
PathLen     =   100            ; Максимальный путь к файлу/каталогу
                                       ; (уменьшать это значение нежелательно)
Start:
 
        push    offset MyFunc
        call    ScanTree
 
        int 20h
 
MyFunc      proc
        lea di,ScanPath    ; DI = адрес строки ScanPath
        push    di
        mov cx,PathLen
        xor al,al
        repne   scasb          ; Ищем конец строки
        dec di
        mov word ptr [di],0A0Dh ; Записываем туда CR,LF...
        mov byte ptr [di+2],'$' ; ...и символ доллара (для ф-ии 9)
 
        mov ah,9
        pop dx
        int 21h            ; Выводим строку на экран
 
        mov byte ptr [di],0; Восстанавливаем строку ScanPath
                               ; (т.к. изменять ее нельзя)
        ret
MyFunc      endp
 
; Маски атрибутов файлов/каталогов
faReadOnly  =   1
faHidden    =   2
faSystem    =   4
faVolume    =   8
faDirectory =   10h
faArchive   =   20h
faAnyFile   =       faReadOnly+faHidden+faSystem+faArchive
faAnyDir    =   faAnyFile+faDirectory
faAnithing  =   faAnyDir+faVolume
 
; Описание буфера DTA
DTAFileInfo struc
dtaReserved db  21 dup (?)
dtaAttr     db  ?
dtaTime     dw  ?
dtaDate     dw  ?
dtaSize     dd  ?
dtaName     db  13 dup (?)
ends
 
; Процедура обхода всех подкаталогов заданного каталога.
; Процедура может работать и в программах формата DOS EXE !!!
; На входе:
; - Переменная StartPath (она же ScanPath) содержит имя стартового каталога.
;   Имя каталога должно заканчиваться обратным слэшем ('\') !
; - В стеке смещение (word) near-функции, которая будет вызываться для каждого
;   найденного каталога. При этом она может использовать ASCIIZ-строку ScanPath,
;   которая содержит полное имя найденного каталога со слэшем на конце, но не
;   должна изменять эту строку! При необходимости функция должна скопировать
;   строку в какой-нибудь буфер, где может делать с ней все, что нужно :)
;   Функция также не должна использовать установленный DTA, при необходимости
;   нужно создать свой буфер DTA. Адрес прежнего буфера восстанавливать при
;   выходе не нужно, как и регистры DS, ES, если они были изменены.
; На выходе:
; - DTA и регистры DS,BP сохранены.
; - Строка StartPath/ScanPath содержит полное имя последнего найденного каталога.
; - ES:DI указывает на конец строки StartPath/ScanPath, которая была при вызове
;   функции (т.е. можно сделать xor al,al / stosb и снова вызвать ScanTree).
ScanTree    proc    pascal
arg     Func:word
local       OldDTA:dword, NewDTA:DTAFileInfo
; ES всегда равно первоначальному DS !!!
 
        push    ds             ; Сохраняем DS
        mov ah,2Fh
        int 21h            ; Получаем адрес DTA
        mov word ptr OldDTA[0],bx ; Сохраняем его...
        mov word ptr OldDTA[2],es ; ...в переменной OldDTA
        pop es             ; Восстанавливаем DS в регистр ES (ES=DS)
 
        lea di,ScanPath
        mov cx,PathLen
        xor al,al
        cld                    ; Прямой порядок поиска scasb
        repne   scasb          ; Ищем в имени каталога нулевой байт (конец строки)
        dec di
        push    di             ; Сохраняем указатель на ноль
 
        push    ds es          ; Сохраняем DS и ES
        call    Func           ; Вызываем функцию для стартового каталога
 
        mov ah,1Ah
        push    ss
        pop ds             ; DS = SS (сегмент NewDTA)
        lea dx,NewDTA
        int 21h            ; Устанавливаем свой DTA
        pop es ds          ; Восстанавливаем DS и ES (ES=DS)
 
        mov ah,4Eh         ; Готовимся в началу поиска
        mov cx,faAnyDir    ; Атрибуты (все, кроме Volume label)
        lea dx,ScanPath
 
        lea si,ScanMask    ; SI = адрес макси поиска
        pop di             ; Восстанавливаем DI (указатель на конец строки)...
        push    di             ; ...и тут же сохраняем его
        cld                    ; Прямой порядок работы lodsb/stosb
@@CopyMask: lodsb                  ; Читаем в AL символ из строки [SI]
        stosb                  ; И записываем его в конец ScanPath
        or  al,al          ; Конец строки [SI] (AL=0) ?
        jnz @@CopyMask     ; Нет, копируем следующий символ
@@FindNext:
        int 21h            ; Начинаем/продолжаем поиск
        jc  @@NotFound     ; Выходим, если ничего не найдено (CF=1)
 
        push    ss
        pop ds             ; DS = SS (сегмент NewDTA)
 
        test    NewDTA.dtaAttr,faDirectory ; Проверяем - каталог ли был найден
        jz  @@PrepareNext  ; Если нет, готовимся продолжать поиск
        lea si,NewDTA.dtaName ; Адрес имени найденного файла/каталога
        lodsb                  ; Читаем первый символ
        cmp al,'.'         ; Это каталоги '.' и '..'? (нам такие не нужны)
        je  @@PrepareNext  ; Если это они, готовимся продолжать поиск
 
        pop di             ; Восстанавливаем DI (указатель на конец строки)...
        push    di             ; ...и тут же сохраняем его
@@CopyPath: 
        stosb                  ; Записываем символ строки [SI] в конец ScanPath
                lodsb                  ; Читаем в AL символ из строки [SI]
        or  al,al          ; Конец строки [SI] (AL=0) ?
        jnz @@CopyPath     ; Нет, копируем следующий символ
        mov ax,'\'         ; AH = 0, AL = '\'
        stosw                  ; Записываем слэш и ноль в конец строки
 
        push    es
        pop ds             ; DS = ES
        push    Func
        call    ScanTree       ; Вызываем себя рекурсивно
        push    ss
        pop ds             ; DS = SS (сегмент NewDTA)
@@PrepareNext:
        mov ah,4Fh         ; Готовимся к функции 4Fh
        lea dx,NewDTA
        jmp @@FindNext     ; Продолжаем поиск
@@NotFound:
        pop di             ; Убираем из стека DI
 
        mov ah,1Ah
        lds dx,OldDTA
        int 21h            ; Восстанавливаем DTA
 
        push    es
        pop ds             ; Восстанавливаем первоначальный DS
        ret                    ; Выходим из процедуры
ScanTree    endp
 
.DATA
 
ScanMask    db  '*.*',0        ; Маска каталогов для поиска
                                       
StartPath   db  'C:\',0        ; Стартовый каталог
ORG StartPath                          ; Совмещаем StartPath и ScanPath
ScanPath    db  PathLen dup (?) ; ASCIIZ-имя найденного каталога
 
END     Start