; ------------------------------------------------------------------------
;
; ATTR.ASM  --        tiny DOS attrib clone (w/ .ASM src)
;
;    written with FASM 1.67.21          http://board.flatassembler.net
;    public domain, free for any use ("nenies proprajxo")
;
;    rugxulo CXE gmail PUNKTO com    (bug reports welcome)
;
; Monday, August 27, 2007   6:32pm
;
; ------------------------------------------------------------------------
;
; +==========+
; | HISTORY: |
; +==========+
;
; 0.6  : 955 (810) bytes, fixed "attr" to show +H +S files like "attr *.*",
;        .ZIP now includes a 624'd .COM (b/c AVG Free resident / heuristics
;         whines FALSELY, scan online at http://virustotal.com if in doubt)
; 0.5  : 939 (794) bytes, fixed "attr /d -r" to not act unnecessarily on
;           dirs, now keeps going if GET_ATTRIB failed (WinXP: "attr \*.*"),
;            plus lists attribs from before / after change ("????" = failed)
; 0.4  : 895 bytes, fixed the new "attr whatever\*.*" (aka, "attr ..\*.*")
;          bug, plus now assembles to only 750 bytes if TINYHELP > 0
; 0.3  : 899 bytes, "attr /d" bugfix, fixed accidental 386+ jumps
;            (doh! sorry ...), plus it's MUCH faster on ignorable files
;        (see "speed" section below)
; 0.2  : 900 bytes, supports wildcards (default '*.*'), says fname + attrib,
;        options /Q (quiet), /D (dirs in wildcards), and /H /? (help
;            screen), plus some basic error checking   
; 0.1  : 168 bytes, first release, only very VERY minimum functionality
;
; ------------------------------------------------------------------------
;
;  +==========+
;  | THANKS : |             )   B-)   =o)       
;  +==========+
;
; -  Thanks to Jesus Christ for his help (in this and everything)!   
; -  Mom? Dad? Beyond words ... so much kindness and patience.             
; -  FASM forum r0x0rz!   ^_^                                            
;
; -  greets to Octavio Vega Fernandez, Jason Hood, David Lindauer,
;    Eric Auer, and Stefan Weber (among many others ...).
;
; ------------------------------------------------------------------------
;
; +=========+
; | NOTES : |
; +=========+
;
;    If this doesn't work fully for you, please contact me, and tell me
;    what happened, what specific OS (PTS-DOS? OS/2? Win NT?) and version
;    that you used it on, and how / if you can duplicate the problem, etc.
;
; -  I used the FASM dialect of assembly language and did a few minor (heh)
;    annoying things, so you may find it harder to translate to NASM (or
;    whatever) than usual. If you can't understand the code, disassemble it
;    or just e-mail me asking me to translate it to NASM, Octasm, etc.
;
; -  Usage:  'attr +sa a*.*'  or  'attr /q -Rh t:*.asm'  or  'attr /q /d -S'
;
; -  the following filespecs should work correctly:
;
;       "t:*.*"   "c:\utils\*.com"   "\*.bat"   "*.bk?"   ""
;
; ------------------------------------------------------------------------
;
; +=========+
; | SPEED : |
; +=========+
;
; [ DR-DOS ] Thu  7-19-2007>attr +r u:attr*.com
; A..R      u:attr1.com
; A..R       u:attr.com
; 
; [ DR-DOS ] Thu  7-19-2007>attr +r u:v.com
; A..R    u:v.com
; 
; [ DR-DOS ] Thu  7-19-2007>runtime fdattrib -r u:\*.com | find "seconds"
; 01.70 seconds elapsed
; 
; [ DR-DOS ] Thu  7-19-2007>attr +r u:v.com
; A..R       u:v.com
; 
; [ DR-DOS ] Thu  7-19-2007>attr +r u:attr*.com
; A..R   u:attr1.com
; A..R       u:attr.com
; 
; [ DR-DOS ] Thu  7-19-2007>runtime attr3 -r u:\*.com
; A... u:\v.com
; A... u:\attr1.com
; A...     u:\attr.com
; 00.11 seconds elapsed
; 
; [ DR-DOS ] Thu  7-19-2007>scrndump
;
; ------------------------------------------------------------------------
;
; +========+
; | TODO : |
; +========+
;
; 1).  make usage of PLEX strlens conditional (if ~ UNDOC, use normal getlen)
;      = would be nice, just in case ...
; 2).  report "Not found" if attr changes nothing (but be silent if /Q)
;      = so far, it's always silent if no change (slightly confusing??)
; 3).  LFN support (int 21h,71..h), which FreeDOS ATTRIB also lacks
;      = shouldn't be hard, see my LFNDIR   
; 4).  perhaps "attr =ar" to set / unset in one invocation ??
;
; -  lacks the following features (not very important for me):
;
;   -  set and unset in one cmd             'attr -s +r *.asm'
;       -  set/unset after fname                'attr *.asm -sr'
;         -  set/unset on multiple filespecs      'attr +r *.asm *.txt'
;    -  set/unset in subdirs                 'attr /s +r c:\code\*.asm'
;
;  -  Not a huge deal, IMO, since I *very* seldomly use such. Use the
;     'for' keyword in COMMAND.COM to simulate some of those.
;
;  -  Or get FreeDOS' ATTRIB instead if you must have such abilities
;  (only 5,044-byte .COM, aPACK'd; GPL, written in C).
;
;  -  Or get Charles Dye's nice clone (5,107-byte .COM, UPX'd; GPL, NASM):
;       http://www.highfiber.com/~raster/freeware.htm
;
; ------------------------------------------------------------------------

TINYHELP=0

; set %INCLUDE% or put "Include=" in your FASMW.INI file (under Environment)

include 'only8086.inc'                 ; not needed, just a precaution
                                     ;   (I didn't write this, only
                                 ;    found it on FASM's forum)

VERSION equ '0.6'

EMAIL equ 'rugxulo AT gmail DOT com'  ; (anti-spam trick for forum)

DEFAULT=0                            ; default options

OPT_Q=1                                ; quiet (no attrib + fname reporting)
OPT_D=2                                ; don't ignore dirs in wildcards

CR=13
LF=10
CRLF=0A0Dh
TAB=9

ARCHIVE=32
DIRECTORY=16
VOLUME=8
SYSTEM=4
HIDDEN=2
READONLY=1

ALL=ARCHIVE or DIRECTORY or SYSTEM or HIDDEN or READONLY and not VOLUME

SET_DTA=1Ah
FINDFIRST=4Eh
FINDNEXT=4Fh

GET_ATTRIB=4300h
SET_ATTRIB=4301h

FILEATTR equ dta+15h
FILENAME equ dta+1Eh
DOS equ int 21h
PLEX equ int 2Fh

macro zero reg { xor reg,reg }

macro uppercase whatever { and whatever,0FFh and not 20h }
macro lowercase whatever { or whatever,20h }

org 100h

Komenco:
       mov ah,SET_DTA
      mov dx,dta                     ; change DTA to avoid PSP:80h clash
      DOS

     mov si,80h
  lodsb
       cbw
 xchg ax,bp                     ; BP = length of cmdline used

    mov byte [si+bp],0             ; make filename ASCIIZ

   cmp byte [si],CR
    jz FinoFino

     mov di,si
   call skip_space

 call find_options

       mov dx,filespec

 cmp byte [di-2],'/'
       jz Zero_it
  cmp byte [di],0
     jnz @f
      cmp byte [di-1],0
   jnz @f
      mov cx,ALL
  test byte [options],OPT_D
   jz .maskoff
 jmp Find
.maskoff:
   and cl,not DIRECTORY
        jmp Find
@@:
 mov al,byte [di-1]
  test al,al
  jz Before
   cmp al,'+'
        jz @f
       cmp al,'-'
        jz @f
       dec di
      jmp Get_filename
@@:
 mov byte [flag],al             ; save plus '+' or minus '-'
Zero_it:
     zero dx                        ; start w/ clear attribute
Search:
    zero bx
.begin:
      cmp byte [di],' '
 jz Get_filename
     cmp byte [di],0
     jz Get_filename
     mov al,byte [attrib_list+bx]
        uppercase byte [di]
 cmp byte [di],al               ; is valid attribute symbol found?
   jz Found
    inc bx                         ; if not, goto next symbol
   cmp bx,attrib_list.size-1
   ja Get_filename                ; > means we've tried all attribs
        jmp .begin                     ;   with no luck!  
Found:
 call modify_value              ; tweak value (set or unset)
 inc di
      jmp Search

FinoFino:
     jmp Fino

Get_filename:
   mov byte [attrib],dl           ; save final attribute
       call skip_space
     dec di
      mov word [file_with_path],di

    mov ax,1212h                   ; DOS 3.0+ (undocumented)
    PLEX                           ;  get length of ASCIIZ in ES:DI

     mov si,di
   mov di,filespec
     push si cx
  cmp byte [si],0
     jz @f
       rep movsb
@@:
        pop cx di

       mov bx,di
   add di,cx
   std
 mov al,'\'
       repnz scasb
 cld

; \mydir
; t:myfiles*.com
; t:\*.asm

 cmp bx,di
   pushf
       inc di
      inc di
      popf
        jnz Before
  cmp byte [bx],'\'
        jz .fix2
    cmp byte [bx+1],':'
   jz Before
.fix:
      dec di
.fix2:
        dec di
Before:
       push di
     push bx
     mov dx,filespec
     mov bx,ALL

      push dx
     mov di,dx
   mov ax,1212h
        PLEX
        mov al,'?'
@@:
     push cx
     repnz scasb
 test cx,cx
  pop cx
      jnz .wildcard_found
 cmp al,'*'
        jz .no_wildcard
     pop di
      push di
     mov dx,di
   mov al,'*'
        jmp @b
.wildcard_found:
      and bl,not DIRECTORY
.no_wildcard:
   mov cx,bx                      ; setup attribs to search for
        pop dx                         ; restore filespec
   pop bx
      pop di
Find:
 mov ah,FINDFIRST               ; find matching files (via wildcards)
        test byte [options],OPT_D
   jz @f
       or cl,DIRECTORY
@@:
  DOS
 jc Fino

Change:
  mov si,FILENAME
     cmp byte [si],'.'            ; skip '.' and '..' (useless filenames)
  jz Next

 push di
     mov ax,1225h                   ; DOS 3.0+ (undocumented)
    PLEX                           ;   get length of ASCII in DS:SI
 rep movsb
   mov dx,word [file_with_path]
        pop di

  mov cl,byte [FILEATTR]

.okay:
    and cl,not DIRECTORY
        mov bx,cx
@@:
        mov ax,SET_ATTRIB
   call adjust_attrib
  cmp byte [flag],' '
       jz @f
       cmp bx,cx                      ; skip files which don't need changing
      jz Next                        ;   for better speed
@@:
      DOS

.output:
     call output_info
    jc Fino
Next:
        mov ah,FINDNEXT
     mov dx,filespec
     DOS
 jc Clear

        mov si,FILENAME
     cmp byte [si],'.'
 jz Next

 jmp Change
Clear:
    zero al
Fino:
        test byte [options],OPT_Q
   jnz Adios

       push ax
     cmp al,3
    jnz @f
      mov si,no_path
      call error_msg
@@:
   cmp al,15
   jnz @f
      mov si,no_drive
     call error_msg
@@:
   cmp al,18
   jnz @f
      mov si,no_file
      call print_str
@@:
   pop ax

Adios:
    mov ah,4Ch
  DOS

; output_info
output_info:
        lahf

    mov bx,cx

       test byte [options],OPT_Q
   jnz .ret

        push dx

 push ax bx
  mov cl,byte [FILEATTR]
      call print_attrib
   pop bx ax

       cmp byte [flag],' '
       jz .mark_dir

    push ax
     mov ax,GET_ATTRIB
   DOS
 pop ax
      jnc .after

      mov cx,bx
   sahf
        jnc .after

      mov ax,'??'
       call fill_attrib_str
        call print_attrib.print
     jmp .mark_dir
.after:
        call print_attrib

.mark_dir:
     mov dl,'['
    call is_dir
 mov si,word [file_with_path]
        call print_str2                ; print fname (lowercase if not DIR)
                                ;    and \n
 mov dl,']'
    call is_dir
.crlf:
   call print_crlf
     pop dx
.ret:
 ret

; print_attrib
print_attrib:
      mov bx,-1
.test:
     inc bx
      cmp bl,attrib_list.size-1
   ja .print
   test cl,[values+bx]
 jz .test
.okay:
      mov dl,[attrib_list+bx]
     mov [attrib_str+bx],dl
      jmp .test

.print:
        mov si,attrib_str
   call print_str                 ; print file's attribute string,

.clear:
 mov ax,'__'
       call fill_attrib_str
.ret:
   ret

; error_msg
error_msg:
    call print_str
      mov si,not_found
    call print_str
.ret:
 ret

; fill_attrib_str
fill_attrib_str:
        push di                        ;   then clear string for later
      mov di,attrib_str
   stosw
       stosw
       pop di
.ret:
 ret

; find_options
find_options:
      cmp byte [di-1],'/'
       jnz .ret
    mov al,byte [di]
    cmp al,'?'
        jnz @f
      jmp .help
@@:
        call toupper
        cmp al,'H'
        jnz @f
.help:
        mov si,help_msg
     call print_str
      jmp Clear
@@:
        cmp al,'Q'
        jnz @f
      xor byte [options],OPT_Q
@@:
 cmp al,'D'
        jnz .next
   xor byte [options],OPT_D

.next:
  cmp byte [di+1],0
   jnz .okay
   inc di
      ret
.okay:
   mov cx,bp
   mov al,' '
        repnz scasb
 dec di
      mov cx,bp
   rep scasb
   jmp find_options
.ret:
       ret

; print_str2
print_str2:
  lodsb
       test al,al
  jz .ret
     test byte [FILEATTR],DIRECTORY
      jnz @f
      call tolower
@@:
     xchg ax,dx
  call print_char
     jmp print_str2
.ret:
 ret

; print_str
print_str:
    push ax
.begin:
      lodsb
       test al,al
  jz .bye
     cmp al,LF
   jnz @f
      call print_crlf
     jmp .begin
@@:
       xchg ax,dx
  call print_char
     jmp .begin
.bye:
     pop ax
.ret:
 ret

; print_crlf
print_crlf:
  mov dl,CR
   call print_char
     mov dl,LF
   DOS
.ret:
    ret

; print_char
print_char:
  mov ah,2
    DOS
.ret:
    ret

; toupper
toupper:
        cmp al,'a'
        jb .ret
     cmp al,'z'
        ja .ret
     uppercase al
.ret:
   ret

; tolower
tolower:
        cmp al,'A'
        jb .ret
     cmp al,'Z'
        ja .ret
     lowercase al
.ret:
   ret

; is_dir
is_dir:
  test byte [FILEATTR],DIRECTORY
      jz .ret
     call print_char
.ret:
        ret

; skip_space
skip_space:
  mov al,' '
        mov cx,bp
   inc cx
      rep scasb
.ret:
      ret

; modify_value
modify_value:
      cmp byte [flag],'+'
       jnz .unset
  or dl,byte [values+bx]
      ret
.unset:
  cmp byte [flag],'-'
       jnz .ret
    push ax
     mov al,byte [values+bx]
     not al
      and al,ARCHIVE or SYSTEM or HIDDEN or READONLY
      test dl,dl
  jnz .and
    or dl,al
    jmp .bye
.and:
       and dl,al
.bye:
      pop ax
.ret:
 ret

; adjust_attrib
adjust_attrib:
    cmp byte [flag],'+'
       jnz .unset
  or cl,byte [attrib]
 ret
.unset:
  cmp byte [flag],'-'
       jnz .ret
    and cl,byte [attrib]
.ret:
   ret

if ~ TINYHELP

help_msg db LF,'ATTR ',VERSION,' -- tiny DOS attrib (w/ .ASM src)',LF
    db TAB,' -- ',EMAIL,LF,LF
         db '/q',TAB,'Quiet (don''t show attribs & fnames)',LF
     db '/d',TAB,'don''t ignore Dirs in wildcard',LF
       db '/h /?',TAB,'this Help',LF,0

else

help_msg db LF,'ATTR ',VERSION,' : /q /d',LF,0

end if

no_path db LF,'Path',0
no_drive db LF,'Drive',0
no_file db LF,'File'

if ~ TINYHELP

not_found db ' not found!',LF,0

else

not_found db ' ??',LF,0

end if

attrib_str db '____  ',0
attrib_list db 'ASHR'
.size=$-attrib_list
values db ARCHIVE,SYSTEM,HIDDEN,READONLY
flag db ' '                         ; set '+' (plus) or unset '-' (minus)
options db DEFAULT

file_with_path dw FILENAME

filespec: db '*.*',0
.len=$-filespec

filespec_data rb 128-filespec.len
attrib rb 1
dta rb 43

; EOF
    

FreeDOS' attrib.com is only 5,044 bytes UPX'd

Charles Petzold's ATTR.COM was 626 bytes way back in 1986, with support for wildcards and elementaty help. Despite PC Magazines no-dist policy it's still around, with source.

Even my dumb, wimpy (buggy?) DOS ATTR.COM is (still) detected as "suspicious unknown EXE / COM virus" by AVG Free. (And it doesn't even open / read / create / modify / write / close / delete any files or stay resident!!)
There are three possible ways around it:
* pack your .COMs and .EXEs (624, aPACK, 32Lite, UPX)
* turn off heuristics in AVG Free Resident Shield (under Properties)
* send the .ZIP (encrypted) w/ password in body to virus AT grisoft DOT com

are more bugs ("attr \*.*" on XP stops after only 4 files for me b/c "Access denied" error #5 on HIBERFIL.SYS)

Quote:

are more bugs ("attr \*.*" on XP stops after only 4 files for me b/c "Access denied" error #5 on HIBERFIL.SYS)

I don't get this "bug" on DOS - maybe not a bug at all

Is much better now.

Remaining issues:

- Display attr before and after change (as APACK'ed bloat does)
- ATTRIB /*.* +s is buggy - should accept /*.* +s as well as +s /*.*