-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathinit.asm
1346 lines (1271 loc) · 25.9 KB
/
init.asm
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
include compiler.h
include version.h
include mem.h
include find.h
include stdio.h
.code
extrn Break10: near ; U4.ASM
extrn Break20: near
extrn Break21: near
extrn BreakCC: near
extrn BreakEP: near
extrn BreakGS: near
extrn BreakGI: near
extrn BreakQT: near
extrn BreakDZ: near
extrn Info: near
extrn SetDTA: near
extrn SegData: word
extrn OldInt21: word
extrn Warning: near
extrn Quit: near
extrn WriteASCIIZ: near
extrn WriteLn: near
extrn WriteLnASCIIZ: near
extrn WriteLongInt: near
extrn FindNext: near
.data
extrn ME_BasicOp: tbyte ; U4.ASM
extrn DefAlignSize: word
extrn Command: byte
extrn DTA_Filename: tbyte
extrn DTABlock: tbyte
extrn File1NmePtr: word
extrn FilenameCnt: byte
extrn Filename1: tbyte
extrn Filename2: tbyte
extrn Filename3: tbyte
extrn Filename4: tbyte
extrn FilesFound: word
extrn HookedInts: tbyte
extrn HookedIntsCnt: abs
extrn IntsCopied: byte
extrn MemStrategy: word
extrn OldInts: tbyte
extrn SegFiles: MEMBLOCK
extrn SegHMAStart: word
extrn SegPSP: word
extrn SwitchB: byte
extrn SwitchI: byte
extrn SwitchN: byte
extrn SwitchP: byte
extrn SwitchV: byte
extrn SwitchS: byte
extrn Switches: byte
extrn SwitchesCnt: abs
extrn SwitchHlp: byte
extrn UMBState: word
DataEnd label
public DataEnd
public Init
.code
Init: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
mov ds,SegData
ASSUME ds:DGROUP
mov SegPSP,es
call GetInt21
call WriteTitle
call GetDefaults
call HandleCLI
call BuildTempfile
call CheckDCmmnd
call HandleOptions
call SetDefCmmnd
call SetDTA
call CheckFilenames
call FreeConvMem
call SetupMemory
call ReadFilenames
call RelocateMem
ASSUME ds:NOTHING,es:NOTHING
call HookInts
call Specials
call DebugInfo
mov es,SegData
ASSUME es:DGROUP
mov ds,SegData
ASSUME ds:DGROUP
mov dx,offset Filename1
call FindFirst
mov es,SegData
ASSUME es:DGROUP
mov si,offset Filename1
call FindFilename
mov File1NmePtr,di
mov si,offset DTA_Filename
mov cx,13
repnz movsb
mov al,FALSE
call SetExecOnly ; clears ExecOnly flag
mov cx,DataSegSz
mov es,NewSegData
xor si,si
xor di,di
shl cx,3
mov bx,ProgSegSz
ret
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
WriteTitle: ASSUME ds:NOTHING, es:NOTHING
mov si,offset Titlebar
call WriteLnASCIIZ
IFDEF REGISTERED
mov si,offset Subbar
call WriteLnASCIIZ
.data
;Subbar db 'Registered to ',REGISTERED, ', distribution prohibited.',0
Subbar db 'Special version registered to ',REGISTERED, ', distribution prohibited.',0
.code
ENDIF
call WriteLn
ret
.data
Titlebar db LF,'UNP ',Version
db ' Executable file restore utility, written by Ben Castricum, ',??date,0
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
GetDefaults: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
mov es,SegData
ASSUME es:DGROUP
mov ds,SegPSP
mov ds,ds:[002Ch]
xor si,si
xor ax,ax
ScanEnv:
cmp ds:[si],ax
je GotEndEnv
inc si
jmp ScanEnv
GotEndEnv:
add si,4
mov di,offset Filename1
mov dx,di
mov cx,64
repnz movsw
mov ds,SegData
ASSUME ds:DGROUP
call OpenFile
mov dx,offset Filename3
mov cx,1Ch
call ReadFile
mov cx,100h
call ReadFile
mov cx,ax
call CloseFile
mov si,dx
ScanToDefaults:
lodsb
cmp al,EOF
je GotDefaults
loop ScanToDefaults
jmp EndDefaults
GotDefaults:
call AnalyzeCLI
EndDefaults:
ret
CheckDCmmnd: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
push ds
push es
mov ds,SegData
ASSUME ds:DGROUP
mov es,SegData
ASSUME es:DGROUP
cmp Command,'D'
jne CD_done
mov cx,SwitchesCnt -1 ; skip the '-' switch
mov bp,offset Switches
mov si,offset SwitchHlp
mov di,offset MyCLISz+1
cmp FilenameCnt,0
je CD_NextSwitch
mov si,offset NoNamesReq
call WriteLnASCIIZ
mov al,ECINVALIDCL
jmp Quit
CD_NextSwitch:
lodsb
mov ah,[bp]
inc bp
mov al,'-'
stosb
mov al,ah
stosb
cmp ah,'K'
mov al,ds:[si-1]
mov bl,al
xor bh,bh
mov al,Swtch2Asc[bx]
stosb
mov al,' '
stosb
loop CD_NextSwitch
mov al,CR
dec di
stosb
mov cx,di
sub cx,offset DefTxt
push cx
sub di,offset MyCLISz
mov ax,di
mov MyCLISz,al
mov FilenameCnt,2
mov si,offset Filename4
mov di,offset Filename2
mov cx,40h
repnz movsw
mov dx,offset Filename2
call CreateFile
mov dx,offset DefTxt
pop cx
call WriteFile
call CloseFile
mov Command,'M'
CD_done:
pop es
pop ds
ret
.data
DefTxt db CR,'UNP ', Version, ' Executable file restore utility,'
db ' written by Ben Castricum.',CR,LF,EOF
MyCLISz db 0
db 128 dup (0)
.code
HandleCLI: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
mov es,SegData
ASSUME es:DGROUP
push ds
mov ds,SegPSP
ASSUME ds:NOTHING
mov si,80h
call AnalyzeCLI
mov ds:[0080h],0D00h ; clear cli
cmp SwitchHlp,FALSE
jne Help
EndCLI:
pop ds
ret
AnalyzeCLI:
push es
mov es,SegData
ASSUME es:DGROUP
mov cl,ds:[si]
mov ch,0
inc si
ReadCLI:
or cx,cx
jle EndLine
call ReadUpCase
cmp al,' '
jbe ReadCLI
cmp al,'-'
je GotSwitch
cmp al,'/'
je GotSwitch
cmp byte ptr ds:[si],' '
jbe GotCommand
GotFilename:
mov bl,FilenameCnt
cmp bl,2
je Help
mov bh,0
push ax
mov ax,128
mul bx
add ax,offset Filename1
mov di,ax
pop ax
WriteFilename:
stosb ; copy filename
call ReadUpCase
cmp al,' '
ja WriteFilename
or bl,bl
jne Name2
cmp byte ptr es:[di-1],'\' ; is name path ?
jne Name2
cmp byte ptr es:[di-1],':' ; is name drive ?
jne Name2
mov ax,'.*' ; append *.*
stosw
mov ah,0
stosw
Name2:
mov al,0
stosb
inc FilenameCnt
jmp ReadCLI
GotSwitch:
call ReadUpCase
cmp al,' '
jbe ReadCLI
cmp al,'-'
je ChildCL
cmp al,'/'
je ChildCL
mov di,offset Switches
push cx
mov cx,SwitchesCnt
repnz scasb ; check for valid switch
jne Help
sub di,offset Switches +1
pop cx
mov ah,al ; remember switch
call ReadUpCase
mov bl,TRUE
cmp al,'+'
je SetSwitch
mov bl,ASK
cmp al,'?'
je SetSwitch
mov bl,FALSE
cmp al,'-'
je SetSwitch
cmp ah,'P'
jne Toggle
push ax
and al,0F0h
cmp al,30h
pop ax
jne Toggle
and ax,000Fh
mov DefAlignSize,ax
mov SwitchP,FALSE
GetNextDigit:
call ReadUpCase
push ax
and al,0F0h
cmp al,30h
pop ax
jne ReadCLI
push ax
mov ax,DefAlignSize
mov bx,10
mul bx
pop bx
and bx,000Fh
add ax,bx
mov DefAlignSize,ax
jmp GetNextDigit
Toggle:
mov bl,byte ptr [SwitchHlp+di]
xor bl,1 ; toggle switch
dec si ; char might be other switch
inc cx
SetSwitch:
mov byte ptr [SwitchHlp+di],bl
jmp GotSwitch
GotCommand:
cmp Command,0
jne GotFilename
mov di,offset Commands
push cx
mov cx,CommandsCnt
repnz scasb ; check for valid command
pop cx
jne GotFilename ; if not valid then filename
mov Command,al
jmp ReadCLI
EndLine:
pop es
ret
ChildCL: ASSUME ds:NOTHING, es:DGROUP
push ds
pop es
ASSUME es:NOTHING
mov di,81h
ShrinkCL:
lodsb
stosb
cmp al,0Dh
jne ShrinkCL
xchg ax,di
sub ax,82h
mov ds:[80h],al
jmp EndLine
Help: ASSUME ds:NOTHING, es:DGROUP
mov ds,SegData
ASSUME ds:DGROUP
mov si,offset HelpText
call WriteLnASCIIZ
mov ch,CommandsCnt
shr ch,1
adc ch,0
mov cl,0
WriteComHelp:
mov si,offset CommandHelp
mov ax,HalveLineSz
mul cl
add si,ax
call WriteASCIIZ
mov si,offset space
call WriteASCIIZ
mov si,offset CommandHelp
mov ax,HalveLineSz
mul ch
add si,ax
call WriteLnASCIIZ
add cx,0101h
mov al,ch
and al,0FEh
cmp al,CommandsCnt
jl WriteComHelp
mov si,offset MoreHelpText
call WriteLnASCIIZ
mov ch,SwitchesCnt
shr ch,1
adc ch,0
mov cl,0
WriteHelp:
mov si,offset SwitchHelp
mov ax,HalveLineSz
mul cl
add si,ax
mov bl,cl
mov bh,0
mov bl,SwitchHlp[bx]
mov al,Swtch2Asc[bx]
mov ds:[si+2],al
call WriteASCIIZ
mov si,offset space
call WriteASCIIZ
mov si,offset SwitchHelp
mov ax,HalveLineSz
mul ch
add si,ax
mov bl,ch
mov bh,0
mov bl,SwitchHlp[bx]
mov al,Swtch2Asc[bx]
mov ds:[si+2],al
call WriteLnASCIIZ
add cx,0101h
mov al,ch
and al,0FEh
cmp al,SwitchesCnt
jl WriteHelp
EndHelp:
mov al,ECHELP
jmp Quit
.data
Commands db 'CDEIL'
IFDEF MARKEXE
db 'M'
ENDIF
db 'OSTX'
CommandsCnt equ $ - offset Commands
HelpText db 'usage: UNP command [options] [[d:][\path]Infile] [[d:][\path]Outfile]',CR,LF,LF
db ' commands:',0
CommandHelp label
db 'c = convert to COM file ',0
db 'd = make current options default ',0
db 'e = expand compressed file (default) ',0
db 'i = show info only ',0
db 'l = load and save ',0
IFDEF MARKEXE
db 'm = MarkEXE, insert a file in header ',0
ENDIF
db 'o = copy overlay ',0
db 's = search for compressed files ',0
db 't = trace executable ',0
db 'x = convert to EXE file ',0
db ' ',0
MoreHelpText label
db LF,' options followed by their current setting',0
SwitchHelp label
db '-? = help (this screen) ',0
HalveLineSz equ $ - offset SwitchHelp
db '-a = automatic retry ',0
db '-b = make backup .BAK file of original ',0
db '-c = ask for confirmation before action',0
db '-f = optimize fixups (like HDROPT.EXE) ',0
db '-g = merge overlay into image ',0
db '-h = remove irrelevant header data ',0
db '-i = interception of I/O interrupts ',0
db '-k = [-|+|?] pklite signature handling ',0
db '-l = use large memory block ',0
db '-m = MORE alike output ',0
db '-n = numbered Outfiles ',0
db '-o = overwrite output file if it exists',0
db '-p = align header data on a page ',0
db '-r = remove overlay data ',0
IFDEF TBAV
db '-s = scan file for virus with TbScanX ',0
ENDIF ;TBAV
db '-u = update file time/date ',0
db '-v = verbose ',0
db '-- = program''s command line ',0
space db ' ',0
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
HandleOptions: ASSUME ds:NOTHING, es:NOTHING
push ds
mov ds,SegData
ASSUME ds:DGROUP
cmp SwitchHlp,FALSE
jne Help
cmp SwitchP,FALSE
je EndOptions
mov DefAlignSize,0200h
EndOptions:
cmp FilenameCnt,0
jne EndFiles
cmp Command,0
je Help
mov di,offset Filename1
mov ax,'.*' ; append *.*
stosw
mov ah,0
stosw
inc FilenameCnt
EndFiles:
pop ds
ret
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
ReadUpCase: ASSUME ds:NOTHING, es:NOTHING
lodsb
dec cx
cmp al,'a'
jb EndUpCase
cmp al,'z'
ja EndUpCase
and al,11011111b
EndUpCase:
ret
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
CheckFilenames: ASSUME ds:NOTHING, es:NOTHING
mov ax,AT_ARCHIVE+AT_READONLY
call SetSrchAttrib
push ds
push es
mov ds,SegData
ASSUME ds:DGROUP
mov es,SegData
ASSUME es:DGROUP
CheckIfAny:
mov dx,offset Filename1 ; check for matching files
call FindFirst
jc CheckIfDir ; none, check if dir
call FindNext
mov al,0
adc al,0
xor al,1
mov MoreThan1,al
or al,al
jne CheckIgnore
mov dx,offset Filename1 ; Reload matching file
call FindFirst
mov si,offset Filename1 ; replace wildcard with name
call FindFilename
mov si,offset DTA_Filename
mov cx,13
repnz movsb
CheckIgnore:
mov al,Command ; ignore outfile commands?
cmp al,'I'
je OnlyInfile
cmp al,'S'
je OnlyInfile
cmp FilenameCnt,1
je NoOutfile
mov si,offset Filename2 ; outfile contains ? or *
call CheckWild
jc OutfileError
mov si,offset Filename1 ; check if name1 = name 2
mov di,offset Filename3
mov ah,60h
int 21h
mov di,offset Filename2
cmp byte ptr [Filename3],0
je CompareName
mov si,di
mov di,offset Filename5
mov ah,60h
int 21h
mov si,offset Filename3
CompareName:
cmpsb
jne HaveOutfile
cmp byte ptr [si-1],0
jne CompareName
mov si,offset WAR_SameName ; delete outfile
call Warning
dec FilenameCnt
NoOutfile:
cmp Command,'O' ; no outfile
je No2Names
cmp Command,'M' ; markexe
jne BNConflict
No2Names:
mov si,offset TwoNamesReq
call WriteLnASCIIZ
mov al,ECINVALIDCL
jmp Quit
HaveOutfile:
cmp MoreThan1,1
je ManyInOne
cmp SwitchB,FALSE
je BNConflict
mov si,offset WAR_IgnoreB
call Warning
mov SwitchB,FALSE
jmp BNConflict
CheckIfDir:
mov si,offset Filename1
call CheckWild
jc CheckExec
mov ax,AT_ARCHIVE+AT_READONLY + AT_DIRECTORY
call SetSrchAttrib
mov dx,offset Filename1
call FindFirst
pushf
mov ax,AT_ARCHIVE+AT_READONLY
call SetSrchAttrib
popf
jc CheckExec
mov di,offset Filename1
mov cx,-1
mov al,0
repnz scasb
dec di
mov ax,'*\'
stosw
mov ax,'*.'
stosw
mov al,0
stosb
jmp CheckIfAny
CheckExec:
mov si,offset Filename1
call FindExtention
cmp byte ptr [di-1],':'
je IsPath
cmp byte ptr [di-1],'\'
je IsPath
cmp byte ptr [di],'.'
je NoFilesFound
mov ax,'*.'
stosw
mov al,0
stosb
mov al,TRUE
call SetExecOnly ; set STDIOs ExecOnly flag
jmp CheckIfAny
IsPath:
mov ax,'.*'
stosw
mov ah,0
stosw
jmp CheckIfAny
NoFilesFound:
mov si,offset NoFiles
call WriteASCIIZ
mov si,offset Filename1
call WriteLnASCIIZ
mov al,ECINVALIDCL
jmp Quit
ManyInOne:
mov si,offset ManyFiles
call WriteLnASCIIZ
mov al,ECINVALIDCL
jmp Quit
OutfileError:
mov si,offset WildcardErr
call WriteLnASCIIZ
mov al,ECINVALIDCL
jmp Quit
OnlyInfile:
mov FilenameCnt,1
BNConflict:
cmp SwitchB,FALSE
je OutfileOk
cmp SwitchN,FALSE
je OutfileOk
mov si,offset WAR_BNconflict
call Warning
mov SwitchB,FALSE
OutfileOk:
pop es
ASSUME es:NOTHING
pop ds
ASSUME ds:NOTHING
ret
CheckWild:
lodsb
cmp al,0
je NoWildCards
cmp al,'*'
je GotWildcards
cmp al,'?'
jne CheckWild
GotWildcards:
stc
ret
NoWildCards:
clc
ret
.data
Filename5 db 128 dup (0)
MoreThan1 db FALSE
NoFiles db 'FATAL ERROR - No files found matching ',0
WAR_SameName db 'Infile and Outfile are same, Outfile ignored.',0
WAR_IgnoreB db 'Outfile specified, -B option ignored.',0
WAR_BNconflict db '-N option overrules -B option, -B option ignored',0
ManyFiles db 'FATAL ERROR - Decompressing many files into one.',0
WildcardErr db 'FATAL ERROR - Output path/file must not contain ''*'' or ''?''.',0
TwoNamesReq db 'FATAL ERROR - Outfile required for specified command.',0
NoNamesReq db 'FATAL ERROR - Specified command does not require filenames.',0
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
SetDefCmmnd: ASSUME ds:NOTHING, es:NOTHING
push ds
mov ds,SegData
ASSUME ds:DGROUP
cmp Command,0
jne GotSomething
cmp FilenameCnt,0
je Help
mov Command,'E'
GotSomething:
pop ds
ASSUME ds:NOTHING
ret
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
SetupMemory: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
push ds
mov ds,SegData
ASSUME ds:DGROUP
call GetUMBState ; get UMB chain state
mov ah,1
mov UMBState,ax
call GetMemStrategy ; get mem allocation strategy
mov ah,1h
mov MemStrategy,ax
mov bl,01h ; inlude UMB in memory chain
call SetUMBState
mov bl,MSHLBestFit ; set new allocation strategy
call SetMemStrategy
call GetFreeMem
call AllocateMem
add bx,ax
mov SegHMAStart,bx
push es
mov es,ax
call FreeMem
pop es
pop ds
ret
.data
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
BuildTempfile: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
push ds ; save DS
push es
mov ax,cs
sub ax,10h
mov es,ax
ASSUME es:NOTHING
mov es,word ptr es:[002Ch]
ASSUME es:NOTHING
mov ds,SegData
xor di,di
mov al,0
CheckEnv:
cmp es:[di],al ; end of environment ?
je NoTempEnv
mov cx,VarTempSz ; check if "TEMP=" found
mov si,offset VarTemp
repz cmpsb
je CopyTempEnv
mov ch,10h
repnz scasb ; go to next variable
jmp CheckEnv
CopyTempEnv:
push es
pop ds ; DS=environment segment
ASSUME ds:NOTHING
mov es,SegData
ASSUME es:DGROUP
mov si,di
mov di,offset Filename4 ; ES:DI=filename location
CopyTempChar:
call ReadUpCase
or al,al
je AppendName
stosb
jmp CopyTempChar
NoTempEnv:
mov es,SegData
ASSUME es:DGROUP
mov di,offset Filename4 ; ES:DI=filename location
jmp CopyTempName
AppendName:
cmp byte ptr es:[di-1],';' ; special code for
jne TempOk ; the ; guy
dec di ; allow trailing ";"s in
jmp AppendName ; TEMP variable
TempOk:
mov al,'\'
cmp byte ptr es:[di-1],al
je CopyTempName
cmp byte ptr es:[di-1],':'
je CopyTempName
stosb
CopyTempName:
mov ds,SegData
ASSUME ds:DGROUP
mov si,offset TempName
mov cx,TempNameSz
repnz movsb
mov ax,cs
sub ax,10h
mov es,ax
ASSUME es:NOTHING
mov es,word ptr es:[002Ch]
ASSUME es:NOTHING
call FreeMem ; release environment
pop es
ASSUME es:NOTHING
pop ds ; restore DS
ASSUME ds:NOTHING
ret
.data
VarTemp db 'TEMP='
VarTempSz equ $ - offset VarTemp
TempName db TMPFile
TempNameSz equ $ - offset TempName
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
FreeConvMem: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
push es
mov ax,SegData
mov bx,cs
sub bx,10h
mov es,bx
ASSUME es:NOTHING
sub ax,bx
mov bx,offset EndInitData
shr bx,4
add bx,ax
inc bx
call ResizeMem
pop es
ASSUME es:NOTHING
ret
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
ReadFilenames: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
push ds
push es
mov bx,1000h
mov si,offset MemFiles
call AllocateMem
mov ds,SegData
ASSUME ds:DGROUP
mov es,ax
ASSUME es:NOTHING
xor di,di
mov SegFiles.A,ax
mov dx,offset Filename1
call FindFirst
xor bp,bp
StoreName:
inc bp
call FindNext
jc FilesSegAdj
mov si,offset DTA_Filename
mov cx,12
CopyChar:
lodsb
stosb
or al,al
loopne CopyChar
jmp StoreName
FilesSegAdj:
mov ax,di
mov bx,ax
shr bx,4
and ax,0Fh
cmp ah,al
adc bx,0
mov SegFiles.S,bx
call ResizeMem
mov FilesFound,bp
pop es
ASSUME es:NOTHING
pop ds
ASSUME ds:NOTHING
ret
.data
MemFiles db 'to store filenames.',0
DataSegSz dw 0
ProgSegSz dw 0
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
.code
RelocateMem: ASSUME ds:NOTHING, es:NOTHING, ss:NOTHING
push ds
push es
mov ds,SegData
ASSUME ds:DGROUP
mov ax,offset DataEnd ; get data size
mov bx,ax
shr bx,4
and ax,0Fh
cmp ah,al
adc bx,0
mov DataSegSz,bx
mov ax,SegData
call FindMem
mov NewSegData,ax
mov cx,offset Init
mov bx,cx
shr bx,4
and cx,0Fh
cmp ch,cl
adc bx,0
cmp ax,SegData ; no move ?
jne CalcSize
mov NewSegData,bx
mov ax,cs
add NewSegData,ax
add bx,DataSegSz
CalcSize:
add bx,10h ; size PSP in paragraphs
mov ProgSegSz,bx
mov bx,SegFiles.S
mov ax,SegFiles.A
call FindMem
cmp ax,SegFiles.A
je NoFilesMove
mov es,ax
ASSUME es:NOTHING
mov cx,SegFiles.S