f8f214a7053a6be4f80bc9963e0b5f721a6ab40b
[reactos.git] / subsystems / ntvdm / dos / dos32krnl / dos.c
1 /*
2 * COPYRIGHT: GPL - See COPYING in the top level directory
3 * PROJECT: ReactOS Virtual DOS Machine
4 * FILE: dos.c
5 * PURPOSE: VDM DOS Kernel
6 * PROGRAMMERS: Aleksandar Andrejevic <theflash AT sdf DOT lonestar DOT org>
7 */
8
9 /* INCLUDES *******************************************************************/
10
11 #define NDEBUG
12
13 #include "emulator.h"
14 #include "callback.h"
15
16 #include "dos.h"
17 #include "dos/dem.h"
18
19 #include "bios/bios.h"
20 #include "registers.h"
21
22 /* PRIVATE VARIABLES **********************************************************/
23
24 CALLBACK16 DosContext;
25
26 static WORD CurrentPsp = SYSTEM_PSP;
27 static WORD DosLastError = 0;
28 static DWORD DiskTransferArea;
29 /*static*/ BYTE CurrentDrive;
30 static CHAR LastDrive = 'E';
31 static CHAR CurrentDirectories[NUM_DRIVES][DOS_DIR_LENGTH];
32 static HANDLE DosSystemFileTable[DOS_SFT_SIZE];
33 static WORD DosSftRefCount[DOS_SFT_SIZE];
34 static BYTE DosAllocStrategy = DOS_ALLOC_BEST_FIT;
35 static BOOLEAN DosUmbLinked = FALSE;
36 static WORD DosErrorLevel = 0x0000;
37
38 /* PRIVATE FUNCTIONS **********************************************************/
39
40 /*
41 * Memory management functions
42 */
43 static VOID DosCombineFreeBlocks(WORD StartBlock)
44 {
45 PDOS_MCB CurrentMcb = SEGMENT_TO_MCB(StartBlock), NextMcb;
46
47 /* If this is the last block or it's not free, quit */
48 if (CurrentMcb->BlockType == 'Z' || CurrentMcb->OwnerPsp != 0) return;
49
50 while (TRUE)
51 {
52 /* Get a pointer to the next MCB */
53 NextMcb = SEGMENT_TO_MCB(StartBlock + CurrentMcb->Size + 1);
54
55 /* Check if the next MCB is free */
56 if (NextMcb->OwnerPsp == 0)
57 {
58 /* Combine them */
59 CurrentMcb->Size += NextMcb->Size + 1;
60 CurrentMcb->BlockType = NextMcb->BlockType;
61 NextMcb->BlockType = 'I';
62 }
63 else
64 {
65 /* No more adjoining free blocks */
66 break;
67 }
68 }
69 }
70
71 static WORD DosAllocateMemory(WORD Size, WORD *MaxAvailable)
72 {
73 WORD Result = 0, Segment = FIRST_MCB_SEGMENT, MaxSize = 0;
74 PDOS_MCB CurrentMcb, NextMcb;
75 BOOLEAN SearchUmb = FALSE;
76
77 DPRINT("DosAllocateMemory: Size 0x%04X\n", Size);
78
79 if (DosUmbLinked && (DosAllocStrategy & (DOS_ALLOC_HIGH | DOS_ALLOC_HIGH_LOW)))
80 {
81 /* Search UMB first */
82 Segment = UMB_START_SEGMENT;
83 SearchUmb = TRUE;
84 }
85
86 while (TRUE)
87 {
88 /* Get a pointer to the MCB */
89 CurrentMcb = SEGMENT_TO_MCB(Segment);
90
91 /* Make sure it's valid */
92 if (CurrentMcb->BlockType != 'M' && CurrentMcb->BlockType != 'Z')
93 {
94 DPRINT("The DOS memory arena is corrupted!\n");
95 DosLastError = ERROR_ARENA_TRASHED;
96 return 0;
97 }
98
99 /* Only check free blocks */
100 if (CurrentMcb->OwnerPsp != 0) goto Next;
101
102 /* Combine this free block with adjoining free blocks */
103 DosCombineFreeBlocks(Segment);
104
105 /* Update the maximum block size */
106 if (CurrentMcb->Size > MaxSize) MaxSize = CurrentMcb->Size;
107
108 /* Check if this block is big enough */
109 if (CurrentMcb->Size < Size) goto Next;
110
111 switch (DosAllocStrategy & 0x3F)
112 {
113 case DOS_ALLOC_FIRST_FIT:
114 {
115 /* For first fit, stop immediately */
116 Result = Segment;
117 goto Done;
118 }
119
120 case DOS_ALLOC_BEST_FIT:
121 {
122 /* For best fit, update the smallest block found so far */
123 if ((Result == 0) || (CurrentMcb->Size < SEGMENT_TO_MCB(Result)->Size))
124 {
125 Result = Segment;
126 }
127
128 break;
129 }
130
131 case DOS_ALLOC_LAST_FIT:
132 {
133 /* For last fit, make the current block the result, but keep searching */
134 Result = Segment;
135 break;
136 }
137 }
138
139 Next:
140 /* If this was the last MCB in the chain, quit */
141 if (CurrentMcb->BlockType == 'Z')
142 {
143 /* Check if nothing was found while searching through UMBs */
144 if ((Result == 0) && SearchUmb && (DosAllocStrategy & DOS_ALLOC_HIGH_LOW))
145 {
146 /* Search low memory */
147 Segment = FIRST_MCB_SEGMENT;
148 continue;
149 }
150
151 break;
152 }
153
154 /* Otherwise, update the segment and continue */
155 Segment += CurrentMcb->Size + 1;
156 }
157
158 Done:
159
160 /* If we didn't find a free block, return 0 */
161 if (Result == 0)
162 {
163 DosLastError = ERROR_NOT_ENOUGH_MEMORY;
164 if (MaxAvailable) *MaxAvailable = MaxSize;
165 return 0;
166 }
167
168 /* Get a pointer to the MCB */
169 CurrentMcb = SEGMENT_TO_MCB(Result);
170
171 /* Check if the block is larger than requested */
172 if (CurrentMcb->Size > Size)
173 {
174 /* It is, split it into two blocks */
175 NextMcb = SEGMENT_TO_MCB(Result + Size + 1);
176
177 /* Initialize the new MCB structure */
178 NextMcb->BlockType = CurrentMcb->BlockType;
179 NextMcb->Size = CurrentMcb->Size - Size - 1;
180 NextMcb->OwnerPsp = 0;
181
182 /* Update the current block */
183 CurrentMcb->BlockType = 'M';
184 CurrentMcb->Size = Size;
185 }
186
187 /* Take ownership of the block */
188 CurrentMcb->OwnerPsp = CurrentPsp;
189
190 /* Return the segment of the data portion of the block */
191 return Result + 1;
192 }
193
194 static BOOLEAN DosResizeMemory(WORD BlockData, WORD NewSize, WORD *MaxAvailable)
195 {
196 BOOLEAN Success = TRUE;
197 WORD Segment = BlockData - 1, ReturnSize = 0, NextSegment;
198 PDOS_MCB Mcb = SEGMENT_TO_MCB(Segment), NextMcb;
199
200 DPRINT("DosResizeMemory: BlockData 0x%04X, NewSize 0x%04X\n",
201 BlockData,
202 NewSize);
203
204 /* Make sure this is a valid, allocated block */
205 if ((Mcb->BlockType != 'M' && Mcb->BlockType != 'Z') || Mcb->OwnerPsp == 0)
206 {
207 Success = FALSE;
208 DosLastError = ERROR_INVALID_HANDLE;
209 goto Done;
210 }
211
212 ReturnSize = Mcb->Size;
213
214 /* Check if we need to expand or contract the block */
215 if (NewSize > Mcb->Size)
216 {
217 /* We can't expand the last block */
218 if (Mcb->BlockType != 'M')
219 {
220 Success = FALSE;
221 goto Done;
222 }
223
224 /* Get the pointer and segment of the next MCB */
225 NextSegment = Segment + Mcb->Size + 1;
226 NextMcb = SEGMENT_TO_MCB(NextSegment);
227
228 /* Make sure the next segment is free */
229 if (NextMcb->OwnerPsp != 0)
230 {
231 DPRINT("Cannot expand memory block: next segment is not free!\n");
232 DosLastError = ERROR_NOT_ENOUGH_MEMORY;
233 Success = FALSE;
234 goto Done;
235 }
236
237 /* Combine this free block with adjoining free blocks */
238 DosCombineFreeBlocks(NextSegment);
239
240 /* Set the maximum possible size of the block */
241 ReturnSize += NextMcb->Size + 1;
242
243 /* Maximize the current block */
244 Mcb->Size = ReturnSize;
245 Mcb->BlockType = NextMcb->BlockType;
246
247 /* Invalidate the next block */
248 NextMcb->BlockType = 'I';
249
250 /* Check if the block is larger than requested */
251 if (Mcb->Size > NewSize)
252 {
253 DPRINT("Block too large, reducing size from 0x%04X to 0x%04X\n",
254 Mcb->Size,
255 NewSize);
256
257 /* It is, split it into two blocks */
258 NextMcb = SEGMENT_TO_MCB(Segment + NewSize + 1);
259
260 /* Initialize the new MCB structure */
261 NextMcb->BlockType = Mcb->BlockType;
262 NextMcb->Size = Mcb->Size - NewSize - 1;
263 NextMcb->OwnerPsp = 0;
264
265 /* Update the current block */
266 Mcb->BlockType = 'M';
267 Mcb->Size = NewSize;
268 }
269 }
270 else if (NewSize < Mcb->Size)
271 {
272 DPRINT("Shrinking block from 0x%04X to 0x%04X\n",
273 Mcb->Size,
274 NewSize);
275
276 /* Just split the block */
277 NextMcb = SEGMENT_TO_MCB(Segment + NewSize + 1);
278 NextMcb->BlockType = Mcb->BlockType;
279 NextMcb->Size = Mcb->Size - NewSize - 1;
280 NextMcb->OwnerPsp = 0;
281
282 /* Update the MCB */
283 Mcb->BlockType = 'M';
284 Mcb->Size = NewSize;
285 }
286
287 Done:
288 /* Check if the operation failed */
289 if (!Success)
290 {
291 DPRINT("DosResizeMemory FAILED. Maximum available: 0x%04X\n",
292 ReturnSize);
293
294 /* Return the maximum possible size */
295 if (MaxAvailable) *MaxAvailable = ReturnSize;
296 }
297
298 return Success;
299 }
300
301 static BOOLEAN DosFreeMemory(WORD BlockData)
302 {
303 PDOS_MCB Mcb = SEGMENT_TO_MCB(BlockData - 1);
304
305 DPRINT("DosFreeMemory: BlockData 0x%04X\n", BlockData);
306
307 /* Make sure the MCB is valid */
308 if (Mcb->BlockType != 'M' && Mcb->BlockType != 'Z')
309 {
310 DPRINT("MCB block type '%c' not valid!\n", Mcb->BlockType);
311 return FALSE;
312 }
313
314 /* Mark the block as free */
315 Mcb->OwnerPsp = 0;
316
317 return TRUE;
318 }
319
320 static BOOLEAN DosLinkUmb(VOID)
321 {
322 DWORD Segment = FIRST_MCB_SEGMENT;
323 PDOS_MCB Mcb = SEGMENT_TO_MCB(Segment);
324
325 DPRINT("Linking UMB\n");
326
327 /* Check if UMBs are already linked */
328 if (DosUmbLinked) return FALSE;
329
330 /* Find the last block */
331 while ((Mcb->BlockType == 'M') && (Segment <= 0xFFFF))
332 {
333 Segment += Mcb->Size + 1;
334 Mcb = SEGMENT_TO_MCB(Segment);
335 }
336
337 /* Make sure it's valid */
338 if (Mcb->BlockType != 'Z') return FALSE;
339
340 /* Connect the MCB with the UMB chain */
341 Mcb->BlockType = 'M';
342
343 DosUmbLinked = TRUE;
344 return TRUE;
345 }
346
347 static BOOLEAN DosUnlinkUmb(VOID)
348 {
349 DWORD Segment = FIRST_MCB_SEGMENT;
350 PDOS_MCB Mcb = SEGMENT_TO_MCB(Segment);
351
352 DPRINT("Unlinking UMB\n");
353
354 /* Check if UMBs are already unlinked */
355 if (!DosUmbLinked) return FALSE;
356
357 /* Find the block preceding the MCB that links it with the UMB chain */
358 while (Segment <= 0xFFFF)
359 {
360 if ((Segment + Mcb->Size) == (FIRST_MCB_SEGMENT + USER_MEMORY_SIZE))
361 {
362 /* This is the last non-UMB segment */
363 break;
364 }
365
366 /* Advance to the next MCB */
367 Segment += Mcb->Size + 1;
368 Mcb = SEGMENT_TO_MCB(Segment);
369 }
370
371 /* Mark the MCB as the last MCB */
372 Mcb->BlockType = 'Z';
373
374 DosUmbLinked = FALSE;
375 return TRUE;
376 }
377
378 static VOID DosChangeMemoryOwner(WORD Segment, WORD NewOwner)
379 {
380 PDOS_MCB Mcb = SEGMENT_TO_MCB(Segment - 1);
381
382 /* Just set the owner */
383 Mcb->OwnerPsp = NewOwner;
384 }
385
386 static WORD DosCopyEnvironmentBlock(LPCVOID Environment, LPCSTR ProgramName)
387 {
388 PCHAR Ptr, DestBuffer = NULL;
389 ULONG TotalSize = 0;
390 WORD DestSegment;
391
392 Ptr = (PCHAR)Environment;
393
394 /* Calculate the size of the environment block */
395 while (*Ptr)
396 {
397 TotalSize += strlen(Ptr) + 1;
398 Ptr += strlen(Ptr) + 1;
399 }
400 TotalSize++;
401
402 /* Add the string buffer size */
403 TotalSize += strlen(ProgramName) + 1;
404
405 /* Allocate the memory for the environment block */
406 DestSegment = DosAllocateMemory((WORD)((TotalSize + 0x0F) >> 4), NULL);
407 if (!DestSegment) return 0;
408
409 Ptr = (PCHAR)Environment;
410
411 DestBuffer = (PCHAR)SEG_OFF_TO_PTR(DestSegment, 0);
412 while (*Ptr)
413 {
414 /* Copy the string */
415 strcpy(DestBuffer, Ptr);
416
417 /* Advance to the next string */
418 DestBuffer += strlen(Ptr);
419 Ptr += strlen(Ptr) + 1;
420
421 /* Put a zero after the string */
422 *(DestBuffer++) = 0;
423 }
424
425 /* Set the final zero */
426 *(DestBuffer++) = 0;
427
428 /* Copy the program name after the environment block */
429 strcpy(DestBuffer, ProgramName);
430
431 return DestSegment;
432 }
433
434 /* Taken from base/shell/cmd/console.c */
435 BOOL IsConsoleHandle(HANDLE hHandle)
436 {
437 DWORD dwMode;
438
439 /* Check whether the handle may be that of a console... */
440 if ((GetFileType(hHandle) & FILE_TYPE_CHAR) == 0) return FALSE;
441
442 /*
443 * It may be. Perform another test... The idea comes from the
444 * MSDN description of the WriteConsole API:
445 *
446 * "WriteConsole fails if it is used with a standard handle
447 * that is redirected to a file. If an application processes
448 * multilingual output that can be redirected, determine whether
449 * the output handle is a console handle (one method is to call
450 * the GetConsoleMode function and check whether it succeeds).
451 * If the handle is a console handle, call WriteConsole. If the
452 * handle is not a console handle, the output is redirected and
453 * you should call WriteFile to perform the I/O."
454 */
455 return GetConsoleMode(hHandle, &dwMode);
456 }
457
458 static WORD DosOpenHandle(HANDLE Handle)
459 {
460 BYTE i;
461 WORD DosHandle;
462 PDOS_PSP PspBlock;
463 LPBYTE HandleTable;
464
465 /* The system PSP has no handle table */
466 if (CurrentPsp == SYSTEM_PSP) return INVALID_DOS_HANDLE;
467
468 /* Get a pointer to the handle table */
469 PspBlock = SEGMENT_TO_PSP(CurrentPsp);
470 HandleTable = (LPBYTE)FAR_POINTER(PspBlock->HandleTablePtr);
471
472 /* Find a free entry in the JFT */
473 for (DosHandle = 0; DosHandle < PspBlock->HandleTableSize; DosHandle++)
474 {
475 if (HandleTable[DosHandle] == 0xFF) break;
476 }
477
478 /* If there are no free entries, fail */
479 if (DosHandle == PspBlock->HandleTableSize) return INVALID_DOS_HANDLE;
480
481 /* Check if the handle is already in the SFT */
482 for (i = 0; i < DOS_SFT_SIZE; i++)
483 {
484 /* Check if this is the same handle */
485 if (DosSystemFileTable[i] != Handle) continue;
486
487 /* Already in the table, reference it */
488 DosSftRefCount[i]++;
489
490 /* Set the JFT entry to that SFT index */
491 HandleTable[DosHandle] = i;
492
493 /* Return the new handle */
494 return DosHandle;
495 }
496
497 /* Add the handle to the SFT */
498 for (i = 0; i < DOS_SFT_SIZE; i++)
499 {
500 /* Make sure this is an empty table entry */
501 if (DosSystemFileTable[i] != INVALID_HANDLE_VALUE) continue;
502
503 /* Initialize the empty table entry */
504 DosSystemFileTable[i] = Handle;
505 DosSftRefCount[i] = 1;
506
507 /* Set the JFT entry to that SFT index */
508 HandleTable[DosHandle] = i;
509
510 /* Return the new handle */
511 return DosHandle;
512 }
513
514 /* The SFT is full */
515 return INVALID_DOS_HANDLE;
516 }
517
518 HANDLE DosGetRealHandle(WORD DosHandle)
519 {
520 PDOS_PSP PspBlock;
521 LPBYTE HandleTable;
522
523 /* The system PSP has no handle table */
524 if (CurrentPsp == SYSTEM_PSP) return INVALID_HANDLE_VALUE;
525
526 /* Get a pointer to the handle table */
527 PspBlock = SEGMENT_TO_PSP(CurrentPsp);
528 HandleTable = (LPBYTE)FAR_POINTER(PspBlock->HandleTablePtr);
529
530 /* Make sure the handle is open */
531 if (HandleTable[DosHandle] == 0xFF) return INVALID_HANDLE_VALUE;
532
533 /* Return the Win32 handle */
534 return DosSystemFileTable[HandleTable[DosHandle]];
535 }
536
537 static VOID DosCopyHandleTable(LPBYTE DestinationTable)
538 {
539 INT i;
540 PDOS_PSP PspBlock;
541 LPBYTE SourceTable;
542
543 /* Clear the table first */
544 for (i = 0; i < 20; i++) DestinationTable[i] = 0xFF;
545
546 /* Check if this is the initial process */
547 if (CurrentPsp == SYSTEM_PSP)
548 {
549 /* Set up the standard I/O devices */
550 for (i = 0; i <= 2; i++)
551 {
552 /* Set the index in the SFT */
553 DestinationTable[i] = (BYTE)i;
554
555 /* Increase the reference count */
556 DosSftRefCount[i]++;
557 }
558
559 /* Done */
560 return;
561 }
562
563 /* Get the parent PSP block and handle table */
564 PspBlock = SEGMENT_TO_PSP(CurrentPsp);
565 SourceTable = (LPBYTE)FAR_POINTER(PspBlock->HandleTablePtr);
566
567 /* Copy the first 20 handles into the new table */
568 for (i = 0; i < 20; i++)
569 {
570 DestinationTable[i] = SourceTable[i];
571
572 /* Increase the reference count */
573 DosSftRefCount[SourceTable[i]]++;
574 }
575 }
576
577 static BOOLEAN DosCloseHandle(WORD DosHandle)
578 {
579 BYTE SftIndex;
580 PDOS_PSP PspBlock;
581 LPBYTE HandleTable;
582
583 DPRINT("DosCloseHandle: DosHandle 0x%04X\n", DosHandle);
584
585 /* The system PSP has no handle table */
586 if (CurrentPsp == SYSTEM_PSP) return FALSE;
587
588 /* Get a pointer to the handle table */
589 PspBlock = SEGMENT_TO_PSP(CurrentPsp);
590 HandleTable = (LPBYTE)FAR_POINTER(PspBlock->HandleTablePtr);
591
592 /* Make sure the handle is open */
593 if (HandleTable[DosHandle] == 0xFF) return FALSE;
594
595 /* Decrement the reference count of the SFT entry */
596 SftIndex = HandleTable[DosHandle];
597 DosSftRefCount[SftIndex]--;
598
599 /* Check if the reference count fell to zero */
600 if (!DosSftRefCount[SftIndex])
601 {
602 /* Close the file, it's no longer needed */
603 CloseHandle(DosSystemFileTable[SftIndex]);
604
605 /* Clear the handle */
606 DosSystemFileTable[SftIndex] = INVALID_HANDLE_VALUE;
607 }
608
609 /* Clear the entry in the JFT */
610 HandleTable[DosHandle] = 0xFF;
611
612 return TRUE;
613 }
614
615 static BOOLEAN DosDuplicateHandle(WORD OldHandle, WORD NewHandle)
616 {
617 BYTE SftIndex;
618 PDOS_PSP PspBlock;
619 LPBYTE HandleTable;
620
621 DPRINT("DosDuplicateHandle: OldHandle 0x%04X, NewHandle 0x%04X\n",
622 OldHandle,
623 NewHandle);
624
625 /* The system PSP has no handle table */
626 if (CurrentPsp == SYSTEM_PSP) return FALSE;
627
628 /* Get a pointer to the handle table */
629 PspBlock = SEGMENT_TO_PSP(CurrentPsp);
630 HandleTable = (LPBYTE)FAR_POINTER(PspBlock->HandleTablePtr);
631
632 /* Make sure the old handle is open */
633 if (HandleTable[OldHandle] == 0xFF) return FALSE;
634
635 /* Check if the new handle is open */
636 if (HandleTable[NewHandle] != 0xFF)
637 {
638 /* Close it */
639 DosCloseHandle(NewHandle);
640 }
641
642 /* Increment the reference count of the SFT entry */
643 SftIndex = HandleTable[OldHandle];
644 DosSftRefCount[SftIndex]++;
645
646 /* Make the new handle point to that SFT entry */
647 HandleTable[NewHandle] = SftIndex;
648
649 /* Return success */
650 return TRUE;
651 }
652
653 static WORD DosCreateFile(LPWORD Handle, LPCSTR FilePath, WORD Attributes)
654 {
655 HANDLE FileHandle;
656 WORD DosHandle;
657
658 DPRINT("DosCreateFile: FilePath \"%s\", Attributes 0x%04X\n",
659 FilePath,
660 Attributes);
661
662 /* Create the file */
663 FileHandle = CreateFileA(FilePath,
664 GENERIC_READ | GENERIC_WRITE,
665 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
666 NULL,
667 CREATE_ALWAYS,
668 Attributes,
669 NULL);
670
671 if (FileHandle == INVALID_HANDLE_VALUE)
672 {
673 /* Return the error code */
674 return (WORD)GetLastError();
675 }
676
677 /* Open the DOS handle */
678 DosHandle = DosOpenHandle(FileHandle);
679
680 if (DosHandle == INVALID_DOS_HANDLE)
681 {
682 /* Close the handle */
683 CloseHandle(FileHandle);
684
685 /* Return the error code */
686 return ERROR_TOO_MANY_OPEN_FILES;
687 }
688
689 /* It was successful */
690 *Handle = DosHandle;
691 return ERROR_SUCCESS;
692 }
693
694 static WORD DosOpenFile(LPWORD Handle, LPCSTR FilePath, BYTE AccessMode)
695 {
696 HANDLE FileHandle;
697 ACCESS_MASK Access = 0;
698 WORD DosHandle;
699
700 DPRINT("DosOpenFile: FilePath \"%s\", AccessMode 0x%04X\n",
701 FilePath,
702 AccessMode);
703
704 /* Parse the access mode */
705 switch (AccessMode & 3)
706 {
707 case 0:
708 {
709 /* Read-only */
710 Access = GENERIC_READ;
711 break;
712 }
713
714 case 1:
715 {
716 /* Write only */
717 Access = GENERIC_WRITE;
718 break;
719 }
720
721 case 2:
722 {
723 /* Read and write */
724 Access = GENERIC_READ | GENERIC_WRITE;
725 break;
726 }
727
728 default:
729 {
730 /* Invalid */
731 return ERROR_INVALID_PARAMETER;
732 }
733 }
734
735 /* Open the file */
736 FileHandle = CreateFileA(FilePath,
737 Access,
738 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
739 NULL,
740 OPEN_EXISTING,
741 FILE_ATTRIBUTE_NORMAL,
742 NULL);
743
744 if (FileHandle == INVALID_HANDLE_VALUE)
745 {
746 /* Return the error code */
747 return (WORD)GetLastError();
748 }
749
750 /* Open the DOS handle */
751 DosHandle = DosOpenHandle(FileHandle);
752
753 if (DosHandle == INVALID_DOS_HANDLE)
754 {
755 /* Close the handle */
756 CloseHandle(FileHandle);
757
758 /* Return the error code */
759 return ERROR_TOO_MANY_OPEN_FILES;
760 }
761
762 /* It was successful */
763 *Handle = DosHandle;
764 return ERROR_SUCCESS;
765 }
766
767 WORD DosReadFile(WORD FileHandle, LPVOID Buffer, WORD Count, LPWORD BytesRead)
768 {
769 WORD Result = ERROR_SUCCESS;
770 DWORD BytesRead32 = 0;
771 HANDLE Handle = DosGetRealHandle(FileHandle);
772
773 DPRINT("DosReadFile: FileHandle 0x%04X, Count 0x%04X\n", FileHandle, Count);
774
775 /* Make sure the handle is valid */
776 if (Handle == INVALID_HANDLE_VALUE) return ERROR_INVALID_HANDLE;
777
778 /* Read the file */
779 if (!ReadFile(Handle, Buffer, Count, &BytesRead32, NULL))
780 {
781 /* Store the error code */
782 Result = (WORD)GetLastError();
783 }
784
785 /* The number of bytes read is always 16-bit */
786 *BytesRead = LOWORD(BytesRead32);
787
788 /* Return the error code */
789 return Result;
790 }
791
792 WORD DosWriteFile(WORD FileHandle, LPVOID Buffer, WORD Count, LPWORD BytesWritten)
793 {
794 WORD Result = ERROR_SUCCESS;
795 DWORD BytesWritten32 = 0;
796 HANDLE Handle = DosGetRealHandle(FileHandle);
797 WORD i;
798
799 DPRINT("DosWriteFile: FileHandle 0x%04X, Count 0x%04X\n",
800 FileHandle,
801 Count);
802
803 /* Make sure the handle is valid */
804 if (Handle == INVALID_HANDLE_VALUE) return ERROR_INVALID_HANDLE;
805
806 if (IsConsoleHandle(Handle))
807 {
808 for (i = 0; i < Count; i++)
809 {
810 /* Save AX and BX */
811 USHORT AX = getAX();
812 USHORT BX = getBX();
813
814 /* Set the parameters */
815 setAL(((PCHAR)Buffer)[i]);
816 setBL(DOS_CHAR_ATTRIBUTE);
817 setBH(Bda->VideoPage);
818
819 /* Call the BIOS INT 10h, AH=0Eh "Teletype Output" */
820 setAH(0x0E);
821 Int32Call(&DosContext, BIOS_VIDEO_INTERRUPT);
822
823 /* Restore AX and BX */
824 setBX(BX);
825 setAX(AX);
826
827 BytesWritten32++;
828 }
829 }
830 else
831 {
832 /* Write the file */
833 if (!WriteFile(Handle, Buffer, Count, &BytesWritten32, NULL))
834 {
835 /* Store the error code */
836 Result = (WORD)GetLastError();
837 }
838 }
839
840 /* The number of bytes written is always 16-bit */
841 *BytesWritten = LOWORD(BytesWritten32);
842
843 /* Return the error code */
844 return Result;
845 }
846
847 static WORD DosSeekFile(WORD FileHandle, LONG Offset, BYTE Origin, LPDWORD NewOffset)
848 {
849 WORD Result = ERROR_SUCCESS;
850 DWORD FilePointer;
851 HANDLE Handle = DosGetRealHandle(FileHandle);
852
853 DPRINT("DosSeekFile: FileHandle 0x%04X, Offset 0x%08X, Origin 0x%02X\n",
854 FileHandle,
855 Offset,
856 Origin);
857
858 /* Make sure the handle is valid */
859 if (Handle == INVALID_HANDLE_VALUE) return ERROR_INVALID_HANDLE;
860
861 /* Check if the origin is valid */
862 if (Origin != FILE_BEGIN && Origin != FILE_CURRENT && Origin != FILE_END)
863 {
864 return ERROR_INVALID_FUNCTION;
865 }
866
867 /* Move the file pointer */
868 FilePointer = SetFilePointer(Handle, Offset, NULL, Origin);
869
870 /* Check if there's a possibility the operation failed */
871 if (FilePointer == INVALID_SET_FILE_POINTER)
872 {
873 /* Get the real error code */
874 Result = (WORD)GetLastError();
875 }
876
877 if (Result != ERROR_SUCCESS)
878 {
879 /* The operation did fail */
880 return Result;
881 }
882
883 /* Return the file pointer, if requested */
884 if (NewOffset) *NewOffset = FilePointer;
885
886 /* Return success */
887 return ERROR_SUCCESS;
888 }
889
890 static BOOLEAN DosFlushFileBuffers(WORD FileHandle)
891 {
892 HANDLE Handle = DosGetRealHandle(FileHandle);
893
894 /* Make sure the handle is valid */
895 if (Handle == INVALID_HANDLE_VALUE) return FALSE;
896
897 /*
898 * No need to check whether the handle is a console handle since
899 * FlushFileBuffers() automatically does this check and calls
900 * FlushConsoleInputBuffer() for us.
901 */
902 // if (IsConsoleHandle(Handle))
903 // return (BOOLEAN)FlushConsoleInputBuffer(Handle);
904 // else
905 return (BOOLEAN)FlushFileBuffers(Handle);
906 }
907
908 static BOOLEAN DosChangeDrive(BYTE Drive)
909 {
910 WCHAR DirectoryPath[DOS_CMDLINE_LENGTH];
911
912 /* Make sure the drive exists */
913 if (Drive > (LastDrive - 'A')) return FALSE;
914
915 /* Find the path to the new current directory */
916 swprintf(DirectoryPath, L"%c\\%S", Drive + 'A', CurrentDirectories[Drive]);
917
918 /* Change the current directory of the process */
919 if (!SetCurrentDirectory(DirectoryPath)) return FALSE;
920
921 /* Set the current drive */
922 CurrentDrive = Drive;
923
924 /* Return success */
925 return TRUE;
926 }
927
928 static BOOLEAN DosChangeDirectory(LPSTR Directory)
929 {
930 BYTE DriveNumber;
931 DWORD Attributes;
932 LPSTR Path;
933
934 /* Make sure the directory path is not too long */
935 if (strlen(Directory) >= DOS_DIR_LENGTH)
936 {
937 DosLastError = ERROR_PATH_NOT_FOUND;
938 return FALSE;
939 }
940
941 /* Get the drive number */
942 DriveNumber = Directory[0] - 'A';
943
944 /* Make sure the drive exists */
945 if (DriveNumber > (LastDrive - 'A'))
946 {
947 DosLastError = ERROR_PATH_NOT_FOUND;
948 return FALSE;
949 }
950
951 /* Get the file attributes */
952 Attributes = GetFileAttributesA(Directory);
953
954 /* Make sure the path exists and is a directory */
955 if ((Attributes == INVALID_FILE_ATTRIBUTES)
956 || !(Attributes & FILE_ATTRIBUTE_DIRECTORY))
957 {
958 DosLastError = ERROR_PATH_NOT_FOUND;
959 return FALSE;
960 }
961
962 /* Check if this is the current drive */
963 if (DriveNumber == CurrentDrive)
964 {
965 /* Change the directory */
966 if (!SetCurrentDirectoryA(Directory))
967 {
968 DosLastError = LOWORD(GetLastError());
969 return FALSE;
970 }
971 }
972
973 /* Get the directory part of the path */
974 Path = strchr(Directory, '\\');
975 if (Path != NULL)
976 {
977 /* Skip the backslash */
978 Path++;
979 }
980
981 /* Set the directory for the drive */
982 if (Path != NULL)
983 {
984 strncpy(CurrentDirectories[DriveNumber], Path, DOS_DIR_LENGTH);
985 }
986 else
987 {
988 CurrentDirectories[DriveNumber][0] = '\0';
989 }
990
991 /* Return success */
992 return TRUE;
993 }
994
995 /* PUBLIC FUNCTIONS ***********************************************************/
996
997 VOID DosInitializePsp(WORD PspSegment, LPCSTR CommandLine, WORD ProgramSize, WORD Environment)
998 {
999 PDOS_PSP PspBlock = SEGMENT_TO_PSP(PspSegment);
1000 LPDWORD IntVecTable = (LPDWORD)((ULONG_PTR)BaseAddress);
1001
1002 ZeroMemory(PspBlock, sizeof(DOS_PSP));
1003
1004 /* Set the exit interrupt */
1005 PspBlock->Exit[0] = 0xCD; // int 0x20
1006 PspBlock->Exit[1] = 0x20;
1007
1008 /* Set the number of the last paragraph */
1009 PspBlock->LastParagraph = PspSegment + ProgramSize - 1;
1010
1011 /* Save the interrupt vectors */
1012 PspBlock->TerminateAddress = IntVecTable[0x22];
1013 PspBlock->BreakAddress = IntVecTable[0x23];
1014 PspBlock->CriticalAddress = IntVecTable[0x24];
1015
1016 /* Set the parent PSP */
1017 PspBlock->ParentPsp = CurrentPsp;
1018
1019 /* Copy the parent handle table */
1020 DosCopyHandleTable(PspBlock->HandleTable);
1021
1022 /* Set the environment block */
1023 PspBlock->EnvBlock = Environment;
1024
1025 /* Set the handle table pointers to the internal handle table */
1026 PspBlock->HandleTableSize = 20;
1027 PspBlock->HandleTablePtr = MAKELONG(0x18, PspSegment);
1028
1029 /* Set the DOS version */
1030 PspBlock->DosVersion = DOS_VERSION;
1031
1032 /* Set the far call opcodes */
1033 PspBlock->FarCall[0] = 0xCD; // int 0x21
1034 PspBlock->FarCall[1] = 0x21;
1035 PspBlock->FarCall[2] = 0xCB; // retf
1036
1037 /* Set the command line */
1038 PspBlock->CommandLineSize = (BYTE)min(strlen(CommandLine), DOS_CMDLINE_LENGTH - 1);
1039 RtlCopyMemory(PspBlock->CommandLine, CommandLine, PspBlock->CommandLineSize);
1040 PspBlock->CommandLine[PspBlock->CommandLineSize] = '\r';
1041 }
1042
1043 DWORD DosLoadExecutable(IN DOS_EXEC_TYPE LoadType,
1044 IN LPCSTR ExecutablePath,
1045 IN LPCSTR CommandLine,
1046 IN PVOID Environment,
1047 OUT PDWORD StackLocation OPTIONAL,
1048 OUT PDWORD EntryPoint OPTIONAL)
1049 {
1050 DWORD Result = ERROR_SUCCESS;
1051 HANDLE FileHandle = INVALID_HANDLE_VALUE, FileMapping = NULL;
1052 LPBYTE Address = NULL;
1053 WORD Segment = 0;
1054 WORD EnvBlock = 0;
1055 WORD MaxAllocSize;
1056 DWORD i, FileSize, ExeSize;
1057 PIMAGE_DOS_HEADER Header;
1058 PDWORD RelocationTable;
1059 PWORD RelocWord;
1060
1061 DPRINT1("DosLoadExecutable(%d, %s, %s, %s, 0x%08X, 0x%08X)\n",
1062 LoadType,
1063 ExecutablePath,
1064 CommandLine,
1065 Environment,
1066 StackLocation,
1067 EntryPoint);
1068
1069 if (LoadType == DOS_LOAD_OVERLAY)
1070 {
1071 DPRINT1("Overlay loading is not supported yet.\n");
1072 return ERROR_NOT_SUPPORTED;
1073 }
1074
1075 /* Open a handle to the executable */
1076 FileHandle = CreateFileA(ExecutablePath,
1077 GENERIC_READ,
1078 0,
1079 NULL,
1080 OPEN_EXISTING,
1081 FILE_ATTRIBUTE_NORMAL,
1082 NULL);
1083 if (FileHandle == INVALID_HANDLE_VALUE)
1084 {
1085 Result = GetLastError();
1086 goto Cleanup;
1087 }
1088
1089 /* Get the file size */
1090 FileSize = GetFileSize(FileHandle, NULL);
1091
1092 /* Create a mapping object for the file */
1093 FileMapping = CreateFileMapping(FileHandle,
1094 NULL,
1095 PAGE_READONLY,
1096 0,
1097 0,
1098 NULL);
1099 if (FileMapping == NULL)
1100 {
1101 Result = GetLastError();
1102 goto Cleanup;
1103 }
1104
1105 /* Map the file into memory */
1106 Address = (LPBYTE)MapViewOfFile(FileMapping, FILE_MAP_READ, 0, 0, 0);
1107 if (Address == NULL)
1108 {
1109 Result = GetLastError();
1110 goto Cleanup;
1111 }
1112
1113 /* Copy the environment block to DOS memory */
1114 EnvBlock = DosCopyEnvironmentBlock(Environment, ExecutablePath);
1115 if (EnvBlock == 0)
1116 {
1117 Result = ERROR_NOT_ENOUGH_MEMORY;
1118 goto Cleanup;
1119 }
1120
1121 /* Check if this is an EXE file or a COM file */
1122 if (Address[0] == 'M' && Address[1] == 'Z')
1123 {
1124 /* EXE file */
1125
1126 /* Get the MZ header */
1127 Header = (PIMAGE_DOS_HEADER)Address;
1128
1129 /* Get the base size of the file, in paragraphs (rounded up) */
1130 ExeSize = (((Header->e_cp - 1) * 512) + Header->e_cblp + 0x0F) >> 4;
1131
1132 /* Add the PSP size, in paragraphs */
1133 ExeSize += sizeof(DOS_PSP) >> 4;
1134
1135 /* Add the maximum size that should be allocated */
1136 ExeSize += Header->e_maxalloc;
1137
1138 /* Make sure it does not pass 0xFFFF */
1139 if (ExeSize > 0xFFFF) ExeSize = 0xFFFF;
1140
1141 /* Reduce the size one by one until the allocation is successful */
1142 for (i = Header->e_maxalloc; i >= Header->e_minalloc; i--, ExeSize--)
1143 {
1144 /* Try to allocate that much memory */
1145 Segment = DosAllocateMemory((WORD)ExeSize, NULL);
1146 if (Segment != 0) break;
1147 }
1148
1149 /* Check if at least the lowest allocation was successful */
1150 if (Segment == 0)
1151 {
1152 Result = ERROR_NOT_ENOUGH_MEMORY;
1153 goto Cleanup;
1154 }
1155
1156 /* Initialize the PSP */
1157 DosInitializePsp(Segment,
1158 CommandLine,
1159 (WORD)ExeSize,
1160 EnvBlock);
1161
1162 /* The process owns its own memory */
1163 DosChangeMemoryOwner(Segment, Segment);
1164 DosChangeMemoryOwner(EnvBlock, Segment);
1165
1166 /* Copy the program to Segment:0100 */
1167 RtlCopyMemory(SEG_OFF_TO_PTR(Segment, 0x100),
1168 Address + (Header->e_cparhdr << 4),
1169 min(FileSize - (Header->e_cparhdr << 4),
1170 (ExeSize << 4) - sizeof(DOS_PSP)));
1171
1172 /* Get the relocation table */
1173 RelocationTable = (PDWORD)(Address + Header->e_lfarlc);
1174
1175 /* Perform relocations */
1176 for (i = 0; i < Header->e_crlc; i++)
1177 {
1178 /* Get a pointer to the word that needs to be patched */
1179 RelocWord = (PWORD)SEG_OFF_TO_PTR(Segment + HIWORD(RelocationTable[i]),
1180 0x100 + LOWORD(RelocationTable[i]));
1181
1182 /* Add the number of the EXE segment to it */
1183 *RelocWord += Segment + (sizeof(DOS_PSP) >> 4);
1184 }
1185
1186 if (LoadType == DOS_LOAD_AND_EXECUTE)
1187 {
1188 /* Set the initial segment registers */
1189 setDS(Segment);
1190 setES(Segment);
1191
1192 /* Set the stack to the location from the header */
1193 EmulatorSetStack(Segment + (sizeof(DOS_PSP) >> 4) + Header->e_ss,
1194 Header->e_sp);
1195
1196 /* Execute */
1197 CurrentPsp = Segment;
1198 DiskTransferArea = MAKELONG(0x80, Segment);
1199 EmulatorExecute(Segment + Header->e_cs + (sizeof(DOS_PSP) >> 4),
1200 Header->e_ip);
1201 }
1202 }
1203 else
1204 {
1205 /* COM file */
1206
1207 /* Find the maximum amount of memory that can be allocated */
1208 DosAllocateMemory(0xFFFF, &MaxAllocSize);
1209
1210 /* Make sure it's enough for the whole program and the PSP */
1211 if (((DWORD)MaxAllocSize << 4) < (FileSize + sizeof(DOS_PSP)))
1212 {
1213 Result = ERROR_NOT_ENOUGH_MEMORY;
1214 goto Cleanup;
1215 }
1216
1217 /* Allocate all of it */
1218 Segment = DosAllocateMemory(MaxAllocSize, NULL);
1219 if (Segment == 0)
1220 {
1221 Result = ERROR_ARENA_TRASHED;
1222 goto Cleanup;
1223 }
1224
1225 /* The process owns its own memory */
1226 DosChangeMemoryOwner(Segment, Segment);
1227 DosChangeMemoryOwner(EnvBlock, Segment);
1228
1229 /* Copy the program to Segment:0100 */
1230 RtlCopyMemory(SEG_OFF_TO_PTR(Segment, 0x100),
1231 Address,
1232 FileSize);
1233
1234 /* Initialize the PSP */
1235 DosInitializePsp(Segment,
1236 CommandLine,
1237 MaxAllocSize,
1238 EnvBlock);
1239
1240 if (LoadType == DOS_LOAD_AND_EXECUTE)
1241 {
1242 /* Set the initial segment registers */
1243 setDS(Segment);
1244 setES(Segment);
1245
1246 /* Set the stack to the last word of the segment */
1247 EmulatorSetStack(Segment, 0xFFFE);
1248
1249 /*
1250 * Set the value on the stack to 0, so that a near return
1251 * jumps to PSP:0000 which has the exit code.
1252 */
1253 *((LPWORD)SEG_OFF_TO_PTR(Segment, 0xFFFE)) = 0;
1254
1255 /* Execute */
1256 CurrentPsp = Segment;
1257 DiskTransferArea = MAKELONG(0x80, Segment);
1258 EmulatorExecute(Segment, 0x100);
1259 }
1260 }
1261
1262 Cleanup:
1263 if (Result != ERROR_SUCCESS)
1264 {
1265 /* It was not successful, cleanup the DOS memory */
1266 if (EnvBlock) DosFreeMemory(EnvBlock);
1267 if (Segment) DosFreeMemory(Segment);
1268 }
1269
1270 /* Unmap the file*/
1271 if (Address != NULL) UnmapViewOfFile(Address);
1272
1273 /* Close the file mapping object */
1274 if (FileMapping != NULL) CloseHandle(FileMapping);
1275
1276 /* Close the file handle */
1277 if (FileHandle != INVALID_HANDLE_VALUE) CloseHandle(FileHandle);
1278
1279 return Result;
1280 }
1281
1282 WORD DosCreateProcess(DOS_EXEC_TYPE LoadType,
1283 LPCSTR ProgramName,
1284 PDOS_EXEC_PARAM_BLOCK Parameters)
1285 {
1286 DWORD Result;
1287 DWORD BinaryType;
1288 LPVOID Environment = NULL;
1289 VDM_COMMAND_INFO CommandInfo;
1290 CHAR CmdLine[MAX_PATH];
1291 CHAR AppName[MAX_PATH];
1292 CHAR PifFile[MAX_PATH];
1293 CHAR Desktop[MAX_PATH];
1294 CHAR Title[MAX_PATH];
1295 CHAR Env[MAX_PATH];
1296 STARTUPINFOA StartupInfo;
1297 PROCESS_INFORMATION ProcessInfo;
1298
1299 /* Get the binary type */
1300 if (!GetBinaryTypeA(ProgramName, &BinaryType)) return GetLastError();
1301
1302 /* Did the caller specify an environment segment? */
1303 if (Parameters->Environment)
1304 {
1305 /* Yes, use it instead of the parent one */
1306 Environment = SEG_OFF_TO_PTR(Parameters->Environment, 0);
1307 }
1308
1309 /* Set up the startup info structure */
1310 ZeroMemory(&StartupInfo, sizeof(STARTUPINFOA));
1311 StartupInfo.cb = sizeof(STARTUPINFOA);
1312
1313 /* Create the process */
1314 if (!CreateProcessA(ProgramName,
1315 FAR_POINTER(Parameters->CommandLine),
1316 NULL,
1317 NULL,
1318 FALSE,
1319 0,
1320 Environment,
1321 NULL,
1322 &StartupInfo,
1323 &ProcessInfo))
1324 {
1325 return GetLastError();
1326 }
1327
1328 /* Check the type of the program */
1329 switch (BinaryType)
1330 {
1331 /* These are handled by NTVDM */
1332 case SCS_DOS_BINARY:
1333 case SCS_WOW_BINARY:
1334 {
1335 /* Clear the structure */
1336 ZeroMemory(&CommandInfo, sizeof(CommandInfo));
1337
1338 /* Initialize the structure members */
1339 CommandInfo.VDMState = VDM_NOT_READY;
1340 CommandInfo.CmdLine = CmdLine;
1341 CommandInfo.CmdLen = sizeof(CmdLine);
1342 CommandInfo.AppName = AppName;
1343 CommandInfo.AppLen = sizeof(AppName);
1344 CommandInfo.PifFile = PifFile;
1345 CommandInfo.PifLen = sizeof(PifFile);
1346 CommandInfo.Desktop = Desktop;
1347 CommandInfo.DesktopLen = sizeof(Desktop);
1348 CommandInfo.Title = Title;
1349 CommandInfo.TitleLen = sizeof(Title);
1350 CommandInfo.Env = Env;
1351 CommandInfo.EnvLen = sizeof(Env);
1352
1353 /* Get the VDM command information */
1354 if (!GetNextVDMCommand(&CommandInfo))
1355 {
1356 /* Shouldn't happen */
1357 ASSERT(FALSE);
1358 }
1359
1360 /* Increment the re-entry count */
1361 CommandInfo.VDMState = VDM_INC_REENTER_COUNT;
1362 GetNextVDMCommand(&CommandInfo);
1363
1364 /* Load the executable */
1365 Result= DosLoadExecutable(LoadType,
1366 AppName,
1367 CmdLine,
1368 Env,
1369 &Parameters->StackLocation,
1370 &Parameters->EntryPoint);
1371 if (Result != ERROR_SUCCESS)
1372 {
1373 DisplayMessage(L"Could not load '%S'. Error: %u", AppName, Result);
1374 break;
1375 }
1376
1377 break;
1378 }
1379
1380 /* Not handled by NTVDM */
1381 default:
1382 {
1383 /* Wait for the process to finish executing */
1384 WaitForSingleObject(ProcessInfo.hProcess, INFINITE);
1385 }
1386 }
1387
1388 /* Close the handles */
1389 CloseHandle(ProcessInfo.hProcess);
1390 CloseHandle(ProcessInfo.hThread);
1391
1392 return ERROR_SUCCESS;
1393 }
1394
1395 VOID DosTerminateProcess(WORD Psp, BYTE ReturnCode)
1396 {
1397 WORD i;
1398 WORD McbSegment = FIRST_MCB_SEGMENT;
1399 PDOS_MCB CurrentMcb;
1400 LPDWORD IntVecTable = (LPDWORD)((ULONG_PTR)BaseAddress);
1401 PDOS_PSP PspBlock = SEGMENT_TO_PSP(Psp);
1402
1403 DPRINT("DosTerminateProcess: Psp 0x%04X, ReturnCode 0x%02X\n",
1404 Psp,
1405 ReturnCode);
1406
1407 /* Check if this PSP is it's own parent */
1408 if (PspBlock->ParentPsp == Psp) goto Done;
1409
1410 for (i = 0; i < PspBlock->HandleTableSize; i++)
1411 {
1412 /* Close the handle */
1413 DosCloseHandle(i);
1414 }
1415
1416 /* Free the memory used by the process */
1417 while (TRUE)
1418 {
1419 /* Get a pointer to the MCB */
1420 CurrentMcb = SEGMENT_TO_MCB(McbSegment);
1421
1422 /* Make sure the MCB is valid */
1423 if (CurrentMcb->BlockType != 'M' && CurrentMcb->BlockType !='Z') break;
1424
1425 /* If this block was allocated by the process, free it */
1426 if (CurrentMcb->OwnerPsp == Psp) DosFreeMemory(McbSegment + 1);
1427
1428 /* If this was the last block, quit */
1429 if (CurrentMcb->BlockType == 'Z') break;
1430
1431 /* Update the segment and continue */
1432 McbSegment += CurrentMcb->Size + 1;
1433 }
1434
1435 Done:
1436 /* Restore the interrupt vectors */
1437 IntVecTable[0x22] = PspBlock->TerminateAddress;
1438 IntVecTable[0x23] = PspBlock->BreakAddress;
1439 IntVecTable[0x24] = PspBlock->CriticalAddress;
1440
1441 /* Update the current PSP */
1442 if (Psp == CurrentPsp)
1443 {
1444 CurrentPsp = PspBlock->ParentPsp;
1445 if (CurrentPsp == SYSTEM_PSP) VdmRunning = FALSE;
1446 }
1447
1448 /* Save the return code - Normal termination */
1449 DosErrorLevel = MAKEWORD(ReturnCode, 0x00);
1450
1451 /* Return control to the parent process */
1452 EmulatorExecute(HIWORD(PspBlock->TerminateAddress),
1453 LOWORD(PspBlock->TerminateAddress));
1454 }
1455
1456 BOOLEAN DosHandleIoctl(BYTE ControlCode, WORD FileHandle)
1457 {
1458 HANDLE Handle = DosGetRealHandle(FileHandle);
1459
1460 if (Handle == INVALID_HANDLE_VALUE)
1461 {
1462 /* Doesn't exist */
1463 DosLastError = ERROR_FILE_NOT_FOUND;
1464 return FALSE;
1465 }
1466
1467 switch (ControlCode)
1468 {
1469 /* Get Device Information */
1470 case 0x00:
1471 {
1472 WORD InfoWord = 0;
1473
1474 /*
1475 * See Ralf Brown: http://www.ctyme.com/intr/rb-2820.htm
1476 * for a list of possible flags.
1477 */
1478
1479 if (Handle == DosSystemFileTable[0])
1480 {
1481 /* Console input */
1482 InfoWord |= 1 << 0;
1483 }
1484 else if (Handle == DosSystemFileTable[1])
1485 {
1486 /* Console output */
1487 InfoWord |= 1 << 1;
1488 }
1489
1490 /* It is a device */
1491 InfoWord |= 1 << 7;
1492
1493 /* Return the device information word */
1494 setDX(InfoWord);
1495 return TRUE;
1496 }
1497
1498 /* Unsupported control code */
1499 default:
1500 {
1501 DPRINT1("Unsupported IOCTL: 0x%02X\n", ControlCode);
1502
1503 DosLastError = ERROR_INVALID_PARAMETER;
1504 return FALSE;
1505 }
1506 }
1507 }
1508
1509 VOID WINAPI DosInt20h(LPWORD Stack)
1510 {
1511 /* This is the exit interrupt */
1512 DosTerminateProcess(Stack[STACK_CS], 0);
1513 }
1514
1515 VOID WINAPI DosInt21h(LPWORD Stack)
1516 {
1517 BYTE Character;
1518 SYSTEMTIME SystemTime;
1519 PCHAR String;
1520 PDOS_INPUT_BUFFER InputBuffer;
1521
1522 /* Check the value in the AH register */
1523 switch (getAH())
1524 {
1525 /* Terminate Program */
1526 case 0x00:
1527 {
1528 DosTerminateProcess(Stack[STACK_CS], 0);
1529 break;
1530 }
1531
1532 /* Read Character from STDIN with Echo */
1533 case 0x01:
1534 {
1535 Character = DosReadCharacter();
1536 DosPrintCharacter(Character);
1537
1538 /* Let the BOP repeat if needed */
1539 if (getCF()) break;
1540
1541 setAL(Character);
1542 break;
1543 }
1544
1545 /* Write Character to STDOUT */
1546 case 0x02:
1547 {
1548 Character = getDL();
1549 DosPrintCharacter(Character);
1550
1551 /*
1552 * We return the output character (DOS 2.1+).
1553 * Also, if we're going to output a TAB, then
1554 * don't return a TAB but a SPACE instead.
1555 * See Ralf Brown: http://www.ctyme.com/intr/rb-2554.htm
1556 * for more information.
1557 */
1558 setAL(Character == '\t' ? ' ' : Character);
1559 break;
1560 }
1561
1562 /* Read Character from STDAUX */
1563 case 0x03:
1564 {
1565 // FIXME: Really read it from STDAUX!
1566 DPRINT1("INT 16h, 03h: Read character from STDAUX is HALFPLEMENTED\n");
1567 setAL(DosReadCharacter());
1568 break;
1569 }
1570
1571 /* Write Character to STDAUX */
1572 case 0x04:
1573 {
1574 // FIXME: Really write it to STDAUX!
1575 DPRINT1("INT 16h, 04h: Write character to STDAUX is HALFPLEMENTED\n");
1576 DosPrintCharacter(getDL());
1577 break;
1578 }
1579
1580 /* Write Character to Printer */
1581 case 0x05:
1582 {
1583 // FIXME: Really write it to printer!
1584 DPRINT1("INT 16h, 05h: Write character to printer is HALFPLEMENTED -\n\n");
1585 DPRINT1("0x%p\n", getDL());
1586 DPRINT1("\n\n-----------\n\n");
1587 break;
1588 }
1589
1590 /* Direct Console I/O */
1591 case 0x06:
1592 {
1593 Character = getDL();
1594
1595 if (Character != 0xFF)
1596 {
1597 /* Output */
1598 DosPrintCharacter(Character);
1599
1600 /*
1601 * We return the output character (DOS 2.1+).
1602 * See Ralf Brown: http://www.ctyme.com/intr/rb-2558.htm
1603 * for more information.
1604 */
1605 setAL(Character);
1606 }
1607 else
1608 {
1609 /* Input */
1610 if (DosCheckInput())
1611 {
1612 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_ZF;
1613 setAL(DosReadCharacter());
1614 }
1615 else
1616 {
1617 /* No character available */
1618 Stack[STACK_FLAGS] |= EMULATOR_FLAG_ZF;
1619 setAL(0x00);
1620 }
1621 }
1622
1623 break;
1624 }
1625
1626 /* Character Input without Echo */
1627 case 0x07:
1628 case 0x08:
1629 {
1630 Character = DosReadCharacter();
1631
1632 /* Let the BOP repeat if needed */
1633 if (getCF()) break;
1634
1635 setAL(Character);
1636 break;
1637 }
1638
1639 /* Write string to STDOUT */
1640 case 0x09:
1641 {
1642 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getDX());
1643
1644 while (*String != '$')
1645 {
1646 DosPrintCharacter(*String);
1647 String++;
1648 }
1649
1650 /*
1651 * We return the terminating character (DOS 2.1+).
1652 * See Ralf Brown: http://www.ctyme.com/intr/rb-2562.htm
1653 * for more information.
1654 */
1655 setAL('$');
1656 break;
1657 }
1658
1659 /* Read Buffered Input */
1660 case 0x0A:
1661 {
1662 InputBuffer = (PDOS_INPUT_BUFFER)SEG_OFF_TO_PTR(getDS(), getDX());
1663
1664 while (Stack[STACK_COUNTER] < InputBuffer->MaxLength)
1665 {
1666 /* Try to read a character */
1667 Character = DosReadCharacter();
1668
1669 /* If it's not ready yet, let the BOP repeat */
1670 if (getCF()) break;
1671
1672 /* Echo the character and append it to the buffer */
1673 DosPrintCharacter(Character);
1674 InputBuffer->Buffer[Stack[STACK_COUNTER]] = Character;
1675
1676 if (Character == '\r') break;
1677 Stack[STACK_COUNTER]++;
1678 }
1679
1680 /* Update the length */
1681 InputBuffer->Length = Stack[STACK_COUNTER];
1682 break;
1683 }
1684
1685 /* Get STDIN Status */
1686 case 0x0B:
1687 {
1688 setAL(DosCheckInput() ? 0xFF : 0x00);
1689 break;
1690 }
1691
1692 /* Flush Buffer and Read STDIN */
1693 case 0x0C:
1694 {
1695 BYTE InputFunction = getAL();
1696
1697 /* Flush STDIN buffer */
1698 DosFlushFileBuffers(DOS_INPUT_HANDLE); // Maybe just create a DosFlushInputBuffer...
1699
1700 /*
1701 * If the input function number contained in AL is valid, i.e.
1702 * AL == 0x01 or 0x06 or 0x07 or 0x08 or 0x0A, call ourselves
1703 * recursively with AL == AH.
1704 */
1705 if (InputFunction == 0x01 || InputFunction == 0x06 ||
1706 InputFunction == 0x07 || InputFunction == 0x08 ||
1707 InputFunction == 0x0A)
1708 {
1709 setAH(InputFunction);
1710 /*
1711 * Instead of calling ourselves really recursively as in:
1712 * DosInt21h(Stack);
1713 * prefer resetting the CF flag to let the BOP repeat.
1714 */
1715 setCF(1);
1716 }
1717 break;
1718 }
1719
1720 /* Disk Reset */
1721 case 0x0D:
1722 {
1723 PDOS_PSP PspBlock = SEGMENT_TO_PSP(CurrentPsp);
1724
1725 // TODO: Flush what's needed.
1726 DPRINT1("INT 21h, 0Dh is UNIMPLEMENTED\n");
1727
1728 /* Clear CF in DOS 6 only */
1729 if (PspBlock->DosVersion == 0x0006)
1730 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
1731
1732 break;
1733 }
1734
1735 /* Set Default Drive */
1736 case 0x0E:
1737 {
1738 DosChangeDrive(getDL());
1739 setAL(LastDrive - 'A' + 1);
1740 break;
1741 }
1742
1743 /* NULL Function for CP/M Compatibility */
1744 case 0x18:
1745 {
1746 /*
1747 * This function corresponds to the CP/M BDOS function
1748 * "get bit map of logged drives", which is meaningless
1749 * under MS-DOS.
1750 *
1751 * For: PTS-DOS 6.51 & S/DOS 1.0 - EXTENDED RENAME FILE USING FCB
1752 * See Ralf Brown: http://www.ctyme.com/intr/rb-2584.htm
1753 * for more information.
1754 */
1755 setAL(0x00);
1756 break;
1757 }
1758
1759 /* Get Default Drive */
1760 case 0x19:
1761 {
1762 setAL(CurrentDrive);
1763 break;
1764 }
1765
1766 /* Set Disk Transfer Area */
1767 case 0x1A:
1768 {
1769 DiskTransferArea = MAKELONG(getDX(), getDS());
1770 break;
1771 }
1772
1773 /* NULL Function for CP/M Compatibility */
1774 case 0x1D:
1775 case 0x1E:
1776 {
1777 /*
1778 * Function 0x1D corresponds to the CP/M BDOS function
1779 * "get bit map of read-only drives", which is meaningless
1780 * under MS-DOS.
1781 * See Ralf Brown: http://www.ctyme.com/intr/rb-2592.htm
1782 * for more information.
1783 *
1784 * Function 0x1E corresponds to the CP/M BDOS function
1785 * "set file attributes", which was meaningless under MS-DOS 1.x.
1786 * See Ralf Brown: http://www.ctyme.com/intr/rb-2593.htm
1787 * for more information.
1788 */
1789 setAL(0x00);
1790 break;
1791 }
1792
1793 /* NULL Function for CP/M Compatibility */
1794 case 0x20:
1795 {
1796 /*
1797 * This function corresponds to the CP/M BDOS function
1798 * "get/set default user (sublibrary) number", which is meaningless
1799 * under MS-DOS.
1800 *
1801 * For: S/DOS 1.0+ & PTS-DOS 6.51+ - GET OEM REVISION
1802 * See Ralf Brown: http://www.ctyme.com/intr/rb-2596.htm
1803 * for more information.
1804 */
1805 setAL(0x00);
1806 break;
1807 }
1808
1809 /* Set Interrupt Vector */
1810 case 0x25:
1811 {
1812 ULONG FarPointer = MAKELONG(getDX(), getDS());
1813 DPRINT1("Setting interrupt 0x%x ...\n", getAL());
1814
1815 /* Write the new far pointer to the IDT */
1816 ((PULONG)BaseAddress)[getAL()] = FarPointer;
1817 break;
1818 }
1819
1820 /* Create New PSP */
1821 case 0x26:
1822 {
1823 DPRINT1("INT 21h, 26h - Create New PSP is UNIMPLEMENTED\n");
1824 break;
1825 }
1826
1827 /* Get System Date */
1828 case 0x2A:
1829 {
1830 GetLocalTime(&SystemTime);
1831 setCX(SystemTime.wYear);
1832 setDX(MAKEWORD(SystemTime.wDay, SystemTime.wMonth));
1833 setAL(SystemTime.wDayOfWeek);
1834 break;
1835 }
1836
1837 /* Set System Date */
1838 case 0x2B:
1839 {
1840 GetLocalTime(&SystemTime);
1841 SystemTime.wYear = getCX();
1842 SystemTime.wMonth = getDH();
1843 SystemTime.wDay = getDL();
1844
1845 /* Return success or failure */
1846 setAL(SetLocalTime(&SystemTime) ? 0x00 : 0xFF);
1847 break;
1848 }
1849
1850 /* Get System Time */
1851 case 0x2C:
1852 {
1853 GetLocalTime(&SystemTime);
1854 setCX(MAKEWORD(SystemTime.wMinute, SystemTime.wHour));
1855 setDX(MAKEWORD(SystemTime.wMilliseconds / 10, SystemTime.wSecond));
1856 break;
1857 }
1858
1859 /* Set System Time */
1860 case 0x2D:
1861 {
1862 GetLocalTime(&SystemTime);
1863 SystemTime.wHour = getCH();
1864 SystemTime.wMinute = getCL();
1865 SystemTime.wSecond = getDH();
1866 SystemTime.wMilliseconds = getDL() * 10; // In hundredths of seconds
1867
1868 /* Return success or failure */
1869 setAL(SetLocalTime(&SystemTime) ? 0x00 : 0xFF);
1870 break;
1871 }
1872
1873 /* Get Disk Transfer Area */
1874 case 0x2F:
1875 {
1876 setES(HIWORD(DiskTransferArea));
1877 setBX(LOWORD(DiskTransferArea));
1878 break;
1879 }
1880
1881 /* Get DOS Version */
1882 case 0x30:
1883 {
1884 PDOS_PSP PspBlock = SEGMENT_TO_PSP(CurrentPsp);
1885
1886 /*
1887 * DOS 2+ - GET DOS VERSION
1888 * See Ralf Brown: http://www.ctyme.com/intr/rb-2711.htm
1889 * for more information.
1890 */
1891
1892 if (LOBYTE(PspBlock->DosVersion) < 5 || getAL() == 0x00)
1893 {
1894 /*
1895 * Return DOS OEM number:
1896 * 0x00 for IBM PC-DOS
1897 * 0x02 for packaged MS-DOS
1898 */
1899 setBH(0x02);
1900 }
1901
1902 if (LOBYTE(PspBlock->DosVersion) >= 5 && getAL() == 0x01)
1903 {
1904 /*
1905 * Return version flag:
1906 * 1 << 3 if DOS is in ROM,
1907 * 0 (reserved) if not.
1908 */
1909 setBH(0x00);
1910 }
1911
1912 /* Return DOS 24-bit user serial number in BL:CX */
1913 setBL(0x00);
1914 setCX(0x0000);
1915
1916 /*
1917 * Return DOS version: Minor:Major in AH:AL
1918 * The Windows NT DOS box returns version 5.00, subject to SETVER.
1919 */
1920 setAX(PspBlock->DosVersion);
1921
1922 break;
1923 }
1924
1925 /* Extended functionalities */
1926 case 0x33:
1927 {
1928 if (getAL() == 0x06)
1929 {
1930 /*
1931 * DOS 5+ - GET TRUE VERSION NUMBER
1932 * This function always returns the true version number, unlike
1933 * AH=30h, whose return value may be changed with SETVER.
1934 * See Ralf Brown: http://www.ctyme.com/intr/rb-2730.htm
1935 * for more information.
1936 */
1937
1938 /*
1939 * Return the true DOS version: Minor:Major in BH:BL
1940 * The Windows NT DOS box returns BX=3205h (version 5.50).
1941 */
1942 setBX(NTDOS_VERSION);
1943
1944 /* DOS revision 0 */
1945 setDL(0x00);
1946
1947 /* Unpatched DOS */
1948 setDH(0x00);
1949 }
1950 // else
1951 // {
1952 // /* Invalid subfunction */
1953 // setAL(0xFF);
1954 // }
1955
1956 break;
1957 }
1958
1959 /* Get Interrupt Vector */
1960 case 0x35:
1961 {
1962 DWORD FarPointer = ((PDWORD)BaseAddress)[getAL()];
1963
1964 /* Read the address from the IDT into ES:BX */
1965 setES(HIWORD(FarPointer));
1966 setBX(LOWORD(FarPointer));
1967 break;
1968 }
1969
1970 /* SWITCH character - AVAILDEV */
1971 case 0x37:
1972 {
1973 if (getAL() == 0x00)
1974 {
1975 /*
1976 * DOS 2+ - "SWITCHAR" - GET SWITCH CHARACTER
1977 * This setting is ignored by MS-DOS 4.0+.
1978 * MS-DOS 5+ always return AL=00h/DL=2Fh.
1979 * See Ralf Brown: http://www.ctyme.com/intr/rb-2752.htm
1980 * for more information.
1981 */
1982 setDL('/');
1983 setAL(0x00);
1984 }
1985 else if (getAL() == 0x01)
1986 {
1987 /*
1988 * DOS 2+ - "SWITCHAR" - SET SWITCH CHARACTER
1989 * This setting is ignored by MS-DOS 5+.
1990 * See Ralf Brown: http://www.ctyme.com/intr/rb-2753.htm
1991 * for more information.
1992 */
1993 // getDL();
1994 setAL(0xFF);
1995 }
1996 else if (getAL() == 0x02)
1997 {
1998 /*
1999 * DOS 2.x and 3.3+ only - "AVAILDEV" - SPECIFY \DEV\ PREFIX USE
2000 * See Ralf Brown: http://www.ctyme.com/intr/rb-2754.htm
2001 * for more information.
2002 */
2003 // setDL();
2004 setAL(0xFF);
2005 }
2006 else if (getAL() == 0x03)
2007 {
2008 /*
2009 * DOS 2.x and 3.3+ only - "AVAILDEV" - SPECIFY \DEV\ PREFIX USE
2010 * See Ralf Brown: http://www.ctyme.com/intr/rb-2754.htm
2011 * for more information.
2012 */
2013 // getDL();
2014 setAL(0xFF);
2015 }
2016 else
2017 {
2018 /* Invalid subfunction */
2019 setAL(0xFF);
2020 }
2021
2022 break;
2023 }
2024
2025 /* Create Directory */
2026 case 0x39:
2027 {
2028 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getDX());
2029
2030 if (CreateDirectoryA(String, NULL))
2031 {
2032 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2033 }
2034 else
2035 {
2036 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2037 setAX(LOWORD(GetLastError()));
2038 }
2039
2040 break;
2041 }
2042
2043 /* Remove Directory */
2044 case 0x3A:
2045 {
2046 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getDX());
2047
2048 if (RemoveDirectoryA(String))
2049 {
2050 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2051 }
2052 else
2053 {
2054 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2055 setAX(LOWORD(GetLastError()));
2056 }
2057
2058 break;
2059 }
2060
2061 /* Set Current Directory */
2062 case 0x3B:
2063 {
2064 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getDX());
2065
2066 if (DosChangeDirectory(String))
2067 {
2068 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2069 }
2070 else
2071 {
2072 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2073 setAX(DosLastError);
2074 }
2075
2076 break;
2077 }
2078
2079 /* Create File */
2080 case 0x3C:
2081 {
2082 WORD FileHandle;
2083 WORD ErrorCode = DosCreateFile(&FileHandle,
2084 (LPCSTR)SEG_OFF_TO_PTR(getDS(), getDX()),
2085 getCX());
2086
2087 if (ErrorCode == 0)
2088 {
2089 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2090 setAX(FileHandle);
2091 }
2092 else
2093 {
2094 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2095 setAX(ErrorCode);
2096 }
2097
2098 break;
2099 }
2100
2101 /* Open File */
2102 case 0x3D:
2103 {
2104 WORD FileHandle;
2105 WORD ErrorCode = DosOpenFile(&FileHandle,
2106 (LPCSTR)SEG_OFF_TO_PTR(getDS(), getDX()),
2107 getAL());
2108
2109 if (ErrorCode == 0)
2110 {
2111 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2112 setAX(FileHandle);
2113 }
2114 else
2115 {
2116 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2117 setAX(ErrorCode);
2118 }
2119
2120 break;
2121 }
2122
2123 /* Close File */
2124 case 0x3E:
2125 {
2126 if (DosCloseHandle(getBX()))
2127 {
2128 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2129 }
2130 else
2131 {
2132 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2133 setAX(ERROR_INVALID_HANDLE);
2134 }
2135
2136 break;
2137 }
2138
2139 /* Read from File or Device */
2140 case 0x3F:
2141 {
2142 WORD Handle = getBX();
2143 LPBYTE Buffer = (LPBYTE)SEG_OFF_TO_PTR(getDS(), getDX());
2144 WORD Count = getCX();
2145 WORD BytesRead = 0;
2146 WORD ErrorCode = ERROR_SUCCESS;
2147 CHAR Character;
2148
2149 if (IsConsoleHandle(DosGetRealHandle(Handle)))
2150 {
2151 while (Stack[STACK_COUNTER] < Count)
2152 {
2153 /* Read a character from the BIOS */
2154 Character = LOBYTE(BiosGetCharacter());
2155
2156 /* Stop if the BOP needs to be repeated */
2157 if (getCF()) break;
2158
2159 // FIXME: Security checks!
2160 DosPrintCharacter(Character);
2161 Buffer[Stack[STACK_COUNTER]++] = Character;
2162
2163 if (Character == '\r')
2164 {
2165 /* Stop on first carriage return */
2166 DosPrintCharacter('\n');
2167 break;
2168 }
2169 }
2170
2171 if (Character != '\r')
2172 {
2173 if (Stack[STACK_COUNTER] < Count) ErrorCode = ERROR_NOT_READY;
2174 else BytesRead = Count;
2175 }
2176 else BytesRead = Stack[STACK_COUNTER];
2177 }
2178 else
2179 {
2180 /* Use the file reading function */
2181 ErrorCode = DosReadFile(Handle, Buffer, Count, &BytesRead);
2182 }
2183
2184 if (ErrorCode == ERROR_SUCCESS)
2185 {
2186 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2187 setAX(BytesRead);
2188 }
2189 else if (ErrorCode != ERROR_NOT_READY)
2190 {
2191 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2192 setAX(ErrorCode);
2193 }
2194 break;
2195 }
2196
2197 /* Write to File or Device */
2198 case 0x40:
2199 {
2200 WORD BytesWritten = 0;
2201 WORD ErrorCode = DosWriteFile(getBX(),
2202 SEG_OFF_TO_PTR(getDS(), getDX()),
2203 getCX(),
2204 &BytesWritten);
2205
2206 if (ErrorCode == ERROR_SUCCESS)
2207 {
2208 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2209 setAX(BytesWritten);
2210 }
2211 else
2212 {
2213 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2214 setAX(ErrorCode);
2215 }
2216
2217 break;
2218 }
2219
2220 /* Delete File */
2221 case 0x41:
2222 {
2223 LPSTR FileName = (LPSTR)SEG_OFF_TO_PTR(getDS(), getDX());
2224
2225 if (demFileDelete(FileName) == ERROR_SUCCESS)
2226 {
2227 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2228 /*
2229 * See Ralf Brown: http://www.ctyme.com/intr/rb-2797.htm
2230 * "AX destroyed (DOS 3.3) AL seems to be drive of deleted file."
2231 */
2232 setAL(FileName[0] - 'A');
2233 }
2234 else
2235 {
2236 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2237 setAX(GetLastError());
2238 }
2239
2240 break;
2241 }
2242
2243 /* Seek File */
2244 case 0x42:
2245 {
2246 DWORD NewLocation;
2247 WORD ErrorCode = DosSeekFile(getBX(),
2248 MAKELONG(getDX(), getCX()),
2249 getAL(),
2250 &NewLocation);
2251
2252 if (ErrorCode == ERROR_SUCCESS)
2253 {
2254 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2255
2256 /* Return the new offset in DX:AX */
2257 setDX(HIWORD(NewLocation));
2258 setAX(LOWORD(NewLocation));
2259 }
2260 else
2261 {
2262 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2263 setAX(ErrorCode);
2264 }
2265
2266 break;
2267 }
2268
2269 /* Get/Set File Attributes */
2270 case 0x43:
2271 {
2272 DWORD Attributes;
2273 LPSTR FileName = (LPSTR)SEG_OFF_TO_PTR(getDS(), getDX());
2274
2275 if (getAL() == 0x00)
2276 {
2277 /* Get the attributes */
2278 Attributes = GetFileAttributesA(FileName);
2279
2280 /* Check if it failed */
2281 if (Attributes == INVALID_FILE_ATTRIBUTES)
2282 {
2283 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2284 setAX(GetLastError());
2285 }
2286 else
2287 {
2288 /* Return the attributes that DOS can understand */
2289 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2290 setCX(Attributes & 0x00FF);
2291 }
2292 }
2293 else if (getAL() == 0x01)
2294 {
2295 /* Try to set the attributes */
2296 if (SetFileAttributesA(FileName, getCL()))
2297 {
2298 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2299 }
2300 else
2301 {
2302 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2303 setAX(GetLastError());
2304 }
2305 }
2306 else
2307 {
2308 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2309 setAX(ERROR_INVALID_FUNCTION);
2310 }
2311
2312 break;
2313 }
2314
2315 /* IOCTL */
2316 case 0x44:
2317 {
2318 if (DosHandleIoctl(getAL(), getBX()))
2319 {
2320 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2321 }
2322 else
2323 {
2324 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2325 setAX(DosLastError);
2326 }
2327
2328 break;
2329 }
2330
2331 /* Duplicate Handle */
2332 case 0x45:
2333 {
2334 WORD NewHandle;
2335 HANDLE Handle = DosGetRealHandle(getBX());
2336
2337 if (Handle != INVALID_HANDLE_VALUE)
2338 {
2339 /* The handle is invalid */
2340 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2341 setAX(ERROR_INVALID_HANDLE);
2342 break;
2343 }
2344
2345 /* Open a new handle to the same entry */
2346 NewHandle = DosOpenHandle(Handle);
2347
2348 if (NewHandle == INVALID_DOS_HANDLE)
2349 {
2350 /* Too many files open */
2351 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2352 setAX(ERROR_TOO_MANY_OPEN_FILES);
2353 break;
2354 }
2355
2356 /* Return the result */
2357 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2358 setAX(NewHandle);
2359 break;
2360 }
2361
2362 /* Force Duplicate Handle */
2363 case 0x46:
2364 {
2365 if (DosDuplicateHandle(getBX(), getCX()))
2366 {
2367 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2368 }
2369 else
2370 {
2371 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2372 setAX(ERROR_INVALID_HANDLE);
2373 }
2374
2375 break;
2376 }
2377
2378 /* Get Current Directory */
2379 case 0x47:
2380 {
2381 BYTE DriveNumber = getDL();
2382 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getSI());
2383
2384 /* Get the real drive number */
2385 if (DriveNumber == 0)
2386 {
2387 DriveNumber = CurrentDrive;
2388 }
2389 else
2390 {
2391 /* Decrement DriveNumber since it was 1-based */
2392 DriveNumber--;
2393 }
2394
2395 if (DriveNumber <= LastDrive - 'A')
2396 {
2397 /*
2398 * Copy the current directory into the target buffer.
2399 * It doesn't contain the drive letter and the backslash.
2400 */
2401 strncpy(String, CurrentDirectories[DriveNumber], DOS_DIR_LENGTH);
2402 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2403 setAX(0x0100); // Undocumented, see Ralf Brown: http://www.ctyme.com/intr/rb-2933.htm
2404 }
2405 else
2406 {
2407 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2408 setAX(ERROR_INVALID_DRIVE);
2409 }
2410
2411 break;
2412 }
2413
2414 /* Allocate Memory */
2415 case 0x48:
2416 {
2417 WORD MaxAvailable = 0;
2418 WORD Segment = DosAllocateMemory(getBX(), &MaxAvailable);
2419
2420 if (Segment != 0)
2421 {
2422 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2423 setAX(Segment);
2424 }
2425 else
2426 {
2427 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2428 setAX(DosLastError);
2429 setBX(MaxAvailable);
2430 }
2431
2432 break;
2433 }
2434
2435 /* Free Memory */
2436 case 0x49:
2437 {
2438 if (DosFreeMemory(getES()))
2439 {
2440 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2441 }
2442 else
2443 {
2444 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2445 setAX(ERROR_ARENA_TRASHED);
2446 }
2447
2448 break;
2449 }
2450
2451 /* Resize Memory Block */
2452 case 0x4A:
2453 {
2454 WORD Size;
2455
2456 if (DosResizeMemory(getES(), getBX(), &Size))
2457 {
2458 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2459 }
2460 else
2461 {
2462 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2463 setAX(DosLastError);
2464 setBX(Size);
2465 }
2466
2467 break;
2468 }
2469
2470 /* Execute */
2471 case 0x4B:
2472 {
2473 DOS_EXEC_TYPE LoadType = (DOS_EXEC_TYPE)getAL();
2474 LPSTR ProgramName = SEG_OFF_TO_PTR(getDS(), getDX());
2475 PDOS_EXEC_PARAM_BLOCK ParamBlock = SEG_OFF_TO_PTR(getES(), getBX());
2476 WORD ErrorCode = DosCreateProcess(LoadType, ProgramName, ParamBlock);
2477
2478 if (ErrorCode == ERROR_SUCCESS)
2479 {
2480 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2481 }
2482 else
2483 {
2484 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2485 setAX(ErrorCode);
2486 }
2487
2488 break;
2489 }
2490
2491 /* Terminate With Return Code */
2492 case 0x4C:
2493 {
2494 DosTerminateProcess(CurrentPsp, getAL());
2495 break;
2496 }
2497
2498 /* Get Return Code (ERRORLEVEL) */
2499 case 0x4D:
2500 {
2501 /*
2502 * According to Ralf Brown: http://www.ctyme.com/intr/rb-2976.htm
2503 * DosErrorLevel is cleared after being read by this function.
2504 */
2505 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2506 setAX(DosErrorLevel);
2507 DosErrorLevel = 0x0000; // Clear it
2508 break;
2509 }
2510
2511 /* Find First File */
2512 case 0x4E:
2513 {
2514 WORD Result = (WORD)demFileFindFirst(FAR_POINTER(DiskTransferArea),
2515 SEG_OFF_TO_PTR(getDS(), getDX()),
2516 getCX());
2517
2518 setAX(Result);
2519 if (Result == ERROR_SUCCESS) Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2520 else Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2521
2522 break;
2523 }
2524
2525 /* Find Next File */
2526 case 0x4F:
2527 {
2528 WORD Result = (WORD)demFileFindNext(FAR_POINTER(DiskTransferArea));
2529
2530 setAX(Result);
2531 if (Result == ERROR_SUCCESS) Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2532 else Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2533
2534 break;
2535 }
2536
2537 /* Internal - Set Current Process ID (Set PSP Address) */
2538 case 0x50:
2539 {
2540 // FIXME: Is it really what it's done ??
2541 CurrentPsp = getBX();
2542 break;
2543 }
2544
2545 /* Internal - Get Current Process ID (Get PSP Address) */
2546 case 0x51:
2547 /* Get Current PSP Address */
2548 case 0x62:
2549 {
2550 /*
2551 * Undocumented AH=51h is identical to the documented AH=62h.
2552 * See Ralf Brown: http://www.ctyme.com/intr/rb-2982.htm
2553 * and http://www.ctyme.com/intr/rb-3140.htm
2554 * for more information.
2555 */
2556 setBX(CurrentPsp);
2557 break;
2558 }
2559
2560 /* Get/Set Memory Management Options */
2561 case 0x58:
2562 {
2563 if (getAL() == 0x00)
2564 {
2565 /* Get allocation strategy */
2566 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2567 setAX(DosAllocStrategy);
2568 }
2569 else if (getAL() == 0x01)
2570 {
2571 /* Set allocation strategy */
2572
2573 if ((getBL() & (DOS_ALLOC_HIGH | DOS_ALLOC_HIGH_LOW))
2574 == (DOS_ALLOC_HIGH | DOS_ALLOC_HIGH_LOW))
2575 {
2576 /* Can't set both */
2577 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2578 setAX(ERROR_INVALID_PARAMETER);
2579 break;
2580 }
2581
2582 if ((getBL() & 0x3F) > DOS_ALLOC_LAST_FIT)
2583 {
2584 /* Invalid allocation strategy */
2585 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2586 setAX(ERROR_INVALID_PARAMETER);
2587 break;
2588 }
2589
2590 DosAllocStrategy = getBL();
2591 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2592 }
2593 else if (getAL() == 0x02)
2594 {
2595 /* Get UMB link state */
2596 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2597 setAL(DosUmbLinked ? 0x01 : 0x00);
2598 }
2599 else if (getAL() == 0x03)
2600 {
2601 /* Set UMB link state */
2602 if (getBX()) DosLinkUmb();
2603 else DosUnlinkUmb();
2604 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2605 }
2606 else
2607 {
2608 /* Invalid or unsupported function */
2609 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2610 setAX(ERROR_INVALID_FUNCTION);
2611 }
2612
2613 break;
2614 }
2615
2616 /* Unsupported */
2617 default:
2618 {
2619 DPRINT1("DOS Function INT 0x21, AH = %xh, AL = %xh NOT IMPLEMENTED!\n",
2620 getAH(), getAL());
2621
2622 setAL(0); // Some functions expect AL to be 0 when it's not supported.
2623 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2624 }
2625 }
2626 }
2627
2628 VOID WINAPI DosBreakInterrupt(LPWORD Stack)
2629 {
2630 UNREFERENCED_PARAMETER(Stack);
2631
2632 /* Stop the VDM */
2633 VdmRunning = FALSE;
2634 }
2635
2636 VOID WINAPI DosFastConOut(LPWORD Stack)
2637 {
2638 /*
2639 * This is the DOS 2+ Fast Console Output Interrupt.
2640 * See Ralf Brown: http://www.ctyme.com/intr/rb-4124.htm
2641 * for more information.
2642 */
2643
2644 #if 0
2645 if (Stack[STACK_COUNTER] == 0)
2646 {
2647 Stack[STACK_COUNTER]++;
2648
2649 /* Save AX and BX */
2650 Stack[STACK_VAR_A] = getAX();
2651 Stack[STACK_VAR_B] = getBX();
2652
2653 /* Rewind the BOP manually, we can't use CF because the interrupt could modify it */
2654 EmulatorExecute(getCS(), getIP() - 4);
2655
2656 /* Call INT 0x10, AH = 0x0E */
2657 setAH(0x0E);
2658 setBL(DOS_CHAR_ATTRIBUTE);
2659 setBH(Bda->VideoPage);
2660
2661 EmulatorInterrupt(0x10);
2662 }
2663 else
2664 {
2665 /* Restore AX and BX */
2666 setAX(Stack[STACK_VAR_A]);
2667 setBX(Stack[STACK_VAR_B]);
2668 }
2669 #else
2670 /* Save AX and BX */
2671 USHORT AX = getAX();
2672 USHORT BX = getBX();
2673
2674 /* Set the parameters (AL = character, already set) */
2675 setBL(DOS_CHAR_ATTRIBUTE);
2676 setBH(Bda->VideoPage);
2677
2678 /* Call the BIOS INT 10h, AH=0Eh "Teletype Output" */
2679 setAH(0x0E);
2680 Int32Call(&DosContext, BIOS_VIDEO_INTERRUPT);
2681
2682 /* Restore AX and BX */
2683 setBX(BX);
2684 setAX(AX);
2685 #endif
2686 }
2687
2688 VOID WINAPI DosInt2Fh(LPWORD Stack)
2689 {
2690 DPRINT1("DOS System Function INT 0x2F, AH = %xh, AL = %xh NOT IMPLEMENTED!\n",
2691 getAH(), getAL());
2692 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2693 }
2694
2695 BOOLEAN DosKRNLInitialize(VOID)
2696 {
2697
2698 #if 1
2699
2700 UCHAR i;
2701 CHAR CurrentDirectory[MAX_PATH];
2702 CHAR DosDirectory[DOS_DIR_LENGTH];
2703 LPSTR Path;
2704
2705 FILE *Stream;
2706 WCHAR Buffer[256];
2707
2708 /* Clear the current directory buffer */
2709 ZeroMemory(CurrentDirectories, sizeof(CurrentDirectories));
2710
2711 /* Get the current directory */
2712 if (!GetCurrentDirectoryA(MAX_PATH, CurrentDirectory))
2713 {
2714 // TODO: Use some kind of default path?
2715 return FALSE;
2716 }
2717
2718 /* Convert that to a DOS path */
2719 if (!GetShortPathNameA(CurrentDirectory, DosDirectory, DOS_DIR_LENGTH))
2720 {
2721 // TODO: Use some kind of default path?
2722 return FALSE;
2723 }
2724
2725 /* Set the drive */
2726 CurrentDrive = DosDirectory[0] - 'A';
2727
2728 /* Get the directory part of the path */
2729 Path = strchr(DosDirectory, '\\');
2730 if (Path != NULL)
2731 {
2732 /* Skip the backslash */
2733 Path++;
2734 }
2735
2736 /* Set the directory */
2737 if (Path != NULL)
2738 {
2739 strncpy(CurrentDirectories[CurrentDrive], Path, DOS_DIR_LENGTH);
2740 }
2741
2742 /* Read CONFIG.SYS */
2743 Stream = _wfopen(DOS_CONFIG_PATH, L"r");
2744 if (Stream != NULL)
2745 {
2746 while (fgetws(Buffer, 256, Stream))
2747 {
2748 // TODO: Parse the line
2749 }
2750 fclose(Stream);
2751 }
2752
2753 /* Initialize the SFT */
2754 for (i = 0; i < DOS_SFT_SIZE; i++)
2755 {
2756 DosSystemFileTable[i] = INVALID_HANDLE_VALUE;
2757 DosSftRefCount[i] = 0;
2758 }
2759
2760 /* Get handles to standard I/O devices */
2761 DosSystemFileTable[0] = GetStdHandle(STD_INPUT_HANDLE);
2762 DosSystemFileTable[1] = GetStdHandle(STD_OUTPUT_HANDLE);
2763 DosSystemFileTable[2] = GetStdHandle(STD_ERROR_HANDLE);
2764
2765 #endif
2766
2767 /* Initialize the callback context */
2768 InitializeContext(&DosContext, 0x0070, 0x0000);
2769
2770 /* Register the DOS 32-bit Interrupts */
2771 RegisterDosInt32(0x20, DosInt20h );
2772 RegisterDosInt32(0x21, DosInt21h );
2773 // RegisterDosInt32(0x22, DosInt22h ); // Termination
2774 RegisterDosInt32(0x23, DosBreakInterrupt); // Ctrl-C / Ctrl-Break
2775 // RegisterDosInt32(0x24, DosInt24h ); // Critical Error
2776 RegisterDosInt32(0x29, DosFastConOut ); // DOS 2+ Fast Console Output
2777 RegisterDosInt32(0x2F, DosInt2Fh );
2778
2779 return TRUE;
2780 }
2781
2782 /* EOF */