[NTVDM]
[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 BinaryType;
1287 LPVOID Environment = NULL;
1288 VDM_COMMAND_INFO CommandInfo;
1289 CHAR CmdLine[MAX_PATH];
1290 CHAR AppName[MAX_PATH];
1291 CHAR PifFile[MAX_PATH];
1292 CHAR Desktop[MAX_PATH];
1293 CHAR Title[MAX_PATH];
1294 CHAR Env[MAX_PATH];
1295 STARTUPINFOA StartupInfo;
1296 PROCESS_INFORMATION ProcessInfo;
1297
1298 /* Get the binary type */
1299 if (!GetBinaryTypeA(ProgramName, &BinaryType)) return GetLastError();
1300
1301 /* Did the caller specify an environment segment? */
1302 if (Parameters->Environment)
1303 {
1304 /* Yes, use it instead of the parent one */
1305 Environment = SEG_OFF_TO_PTR(Parameters->Environment, 0);
1306 }
1307
1308 /* Set up the startup info structure */
1309 ZeroMemory(&StartupInfo, sizeof(STARTUPINFOA));
1310 StartupInfo.cb = sizeof(STARTUPINFOA);
1311
1312 /* Create the process */
1313 if (!CreateProcessA(ProgramName,
1314 FAR_POINTER(Parameters->CommandLine),
1315 NULL,
1316 NULL,
1317 FALSE,
1318 0,
1319 Environment,
1320 NULL,
1321 &StartupInfo,
1322 &ProcessInfo))
1323 {
1324 return GetLastError();
1325 }
1326
1327 /* Check the type of the program */
1328 switch (BinaryType)
1329 {
1330 /* These are handled by NTVDM */
1331 case SCS_DOS_BINARY:
1332 case SCS_WOW_BINARY:
1333 {
1334 /* Clear the structure */
1335 ZeroMemory(&CommandInfo, sizeof(CommandInfo));
1336
1337 /* Initialize the structure members */
1338 CommandInfo.VDMState = VDM_NOT_READY;
1339 CommandInfo.CmdLine = CmdLine;
1340 CommandInfo.CmdLen = sizeof(CmdLine);
1341 CommandInfo.AppName = AppName;
1342 CommandInfo.AppLen = sizeof(AppName);
1343 CommandInfo.PifFile = PifFile;
1344 CommandInfo.PifLen = sizeof(PifFile);
1345 CommandInfo.Desktop = Desktop;
1346 CommandInfo.DesktopLen = sizeof(Desktop);
1347 CommandInfo.Title = Title;
1348 CommandInfo.TitleLen = sizeof(Title);
1349 CommandInfo.Env = Env;
1350 CommandInfo.EnvLen = sizeof(Env);
1351
1352 /* Get the VDM command information */
1353 if (!GetNextVDMCommand(&CommandInfo))
1354 {
1355 /* Shouldn't happen */
1356 ASSERT(FALSE);
1357 }
1358
1359 /* Increment the re-entry count */
1360 CommandInfo.VDMState = VDM_INC_REENTER_COUNT;
1361 GetNextVDMCommand(&CommandInfo);
1362
1363 /* Load the executable */
1364 if (DosLoadExecutable(LoadType,
1365 AppName,
1366 CmdLine,
1367 Env,
1368 &Parameters->StackLocation,
1369 &Parameters->EntryPoint) != ERROR_SUCCESS)
1370 {
1371 DisplayMessage(L"Could not load '%S'", AppName);
1372 break;
1373 }
1374
1375 break;
1376 }
1377
1378 /* Not handled by NTVDM */
1379 default:
1380 {
1381 /* Wait for the process to finish executing */
1382 WaitForSingleObject(ProcessInfo.hProcess, INFINITE);
1383 }
1384 }
1385
1386 /* Close the handles */
1387 CloseHandle(ProcessInfo.hProcess);
1388 CloseHandle(ProcessInfo.hThread);
1389
1390 return ERROR_SUCCESS;
1391 }
1392
1393 VOID DosTerminateProcess(WORD Psp, BYTE ReturnCode)
1394 {
1395 WORD i;
1396 WORD McbSegment = FIRST_MCB_SEGMENT;
1397 PDOS_MCB CurrentMcb;
1398 LPDWORD IntVecTable = (LPDWORD)((ULONG_PTR)BaseAddress);
1399 PDOS_PSP PspBlock = SEGMENT_TO_PSP(Psp);
1400
1401 DPRINT("DosTerminateProcess: Psp 0x%04X, ReturnCode 0x%02X\n",
1402 Psp,
1403 ReturnCode);
1404
1405 /* Check if this PSP is it's own parent */
1406 if (PspBlock->ParentPsp == Psp) goto Done;
1407
1408 for (i = 0; i < PspBlock->HandleTableSize; i++)
1409 {
1410 /* Close the handle */
1411 DosCloseHandle(i);
1412 }
1413
1414 /* Free the memory used by the process */
1415 while (TRUE)
1416 {
1417 /* Get a pointer to the MCB */
1418 CurrentMcb = SEGMENT_TO_MCB(McbSegment);
1419
1420 /* Make sure the MCB is valid */
1421 if (CurrentMcb->BlockType != 'M' && CurrentMcb->BlockType !='Z') break;
1422
1423 /* If this block was allocated by the process, free it */
1424 if (CurrentMcb->OwnerPsp == Psp) DosFreeMemory(McbSegment + 1);
1425
1426 /* If this was the last block, quit */
1427 if (CurrentMcb->BlockType == 'Z') break;
1428
1429 /* Update the segment and continue */
1430 McbSegment += CurrentMcb->Size + 1;
1431 }
1432
1433 Done:
1434 /* Restore the interrupt vectors */
1435 IntVecTable[0x22] = PspBlock->TerminateAddress;
1436 IntVecTable[0x23] = PspBlock->BreakAddress;
1437 IntVecTable[0x24] = PspBlock->CriticalAddress;
1438
1439 /* Update the current PSP */
1440 if (Psp == CurrentPsp)
1441 {
1442 CurrentPsp = PspBlock->ParentPsp;
1443 if (CurrentPsp == SYSTEM_PSP) VdmRunning = FALSE;
1444 }
1445
1446 /* Save the return code - Normal termination */
1447 DosErrorLevel = MAKEWORD(ReturnCode, 0x00);
1448
1449 /* Return control to the parent process */
1450 EmulatorExecute(HIWORD(PspBlock->TerminateAddress),
1451 LOWORD(PspBlock->TerminateAddress));
1452 }
1453
1454 BOOLEAN DosHandleIoctl(BYTE ControlCode, WORD FileHandle)
1455 {
1456 HANDLE Handle = DosGetRealHandle(FileHandle);
1457
1458 if (Handle == INVALID_HANDLE_VALUE)
1459 {
1460 /* Doesn't exist */
1461 DosLastError = ERROR_FILE_NOT_FOUND;
1462 return FALSE;
1463 }
1464
1465 switch (ControlCode)
1466 {
1467 /* Get Device Information */
1468 case 0x00:
1469 {
1470 WORD InfoWord = 0;
1471
1472 /*
1473 * See Ralf Brown: http://www.ctyme.com/intr/rb-2820.htm
1474 * for a list of possible flags.
1475 */
1476
1477 if (Handle == DosSystemFileTable[0])
1478 {
1479 /* Console input */
1480 InfoWord |= 1 << 0;
1481 }
1482 else if (Handle == DosSystemFileTable[1])
1483 {
1484 /* Console output */
1485 InfoWord |= 1 << 1;
1486 }
1487
1488 /* It is a device */
1489 InfoWord |= 1 << 7;
1490
1491 /* Return the device information word */
1492 setDX(InfoWord);
1493 return TRUE;
1494 }
1495
1496 /* Unsupported control code */
1497 default:
1498 {
1499 DPRINT1("Unsupported IOCTL: 0x%02X\n", ControlCode);
1500
1501 DosLastError = ERROR_INVALID_PARAMETER;
1502 return FALSE;
1503 }
1504 }
1505 }
1506
1507 VOID WINAPI DosInt20h(LPWORD Stack)
1508 {
1509 /* This is the exit interrupt */
1510 DosTerminateProcess(Stack[STACK_CS], 0);
1511 }
1512
1513 VOID WINAPI DosInt21h(LPWORD Stack)
1514 {
1515 BYTE Character;
1516 SYSTEMTIME SystemTime;
1517 PCHAR String;
1518 PDOS_INPUT_BUFFER InputBuffer;
1519
1520 /* Check the value in the AH register */
1521 switch (getAH())
1522 {
1523 /* Terminate Program */
1524 case 0x00:
1525 {
1526 DosTerminateProcess(Stack[STACK_CS], 0);
1527 break;
1528 }
1529
1530 /* Read Character from STDIN with Echo */
1531 case 0x01:
1532 {
1533 Character = DosReadCharacter();
1534 DosPrintCharacter(Character);
1535
1536 /* Let the BOP repeat if needed */
1537 if (getCF()) break;
1538
1539 setAL(Character);
1540 break;
1541 }
1542
1543 /* Write Character to STDOUT */
1544 case 0x02:
1545 {
1546 Character = getDL();
1547 DosPrintCharacter(Character);
1548
1549 /*
1550 * We return the output character (DOS 2.1+).
1551 * Also, if we're going to output a TAB, then
1552 * don't return a TAB but a SPACE instead.
1553 * See Ralf Brown: http://www.ctyme.com/intr/rb-2554.htm
1554 * for more information.
1555 */
1556 setAL(Character == '\t' ? ' ' : Character);
1557 break;
1558 }
1559
1560 /* Read Character from STDAUX */
1561 case 0x03:
1562 {
1563 // FIXME: Really read it from STDAUX!
1564 DPRINT1("INT 16h, 03h: Read character from STDAUX is HALFPLEMENTED\n");
1565 setAL(DosReadCharacter());
1566 break;
1567 }
1568
1569 /* Write Character to STDAUX */
1570 case 0x04:
1571 {
1572 // FIXME: Really write it to STDAUX!
1573 DPRINT1("INT 16h, 04h: Write character to STDAUX is HALFPLEMENTED\n");
1574 DosPrintCharacter(getDL());
1575 break;
1576 }
1577
1578 /* Write Character to Printer */
1579 case 0x05:
1580 {
1581 // FIXME: Really write it to printer!
1582 DPRINT1("INT 16h, 05h: Write character to printer is HALFPLEMENTED -\n\n");
1583 DPRINT1("0x%p\n", getDL());
1584 DPRINT1("\n\n-----------\n\n");
1585 break;
1586 }
1587
1588 /* Direct Console I/O */
1589 case 0x06:
1590 {
1591 Character = getDL();
1592
1593 if (Character != 0xFF)
1594 {
1595 /* Output */
1596 DosPrintCharacter(Character);
1597
1598 /*
1599 * We return the output character (DOS 2.1+).
1600 * See Ralf Brown: http://www.ctyme.com/intr/rb-2558.htm
1601 * for more information.
1602 */
1603 setAL(Character);
1604 }
1605 else
1606 {
1607 /* Input */
1608 if (DosCheckInput())
1609 {
1610 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_ZF;
1611 setAL(DosReadCharacter());
1612 }
1613 else
1614 {
1615 /* No character available */
1616 Stack[STACK_FLAGS] |= EMULATOR_FLAG_ZF;
1617 setAL(0x00);
1618 }
1619 }
1620
1621 break;
1622 }
1623
1624 /* Character Input without Echo */
1625 case 0x07:
1626 case 0x08:
1627 {
1628 Character = DosReadCharacter();
1629
1630 /* Let the BOP repeat if needed */
1631 if (getCF()) break;
1632
1633 setAL(Character);
1634 break;
1635 }
1636
1637 /* Write string to STDOUT */
1638 case 0x09:
1639 {
1640 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getDX());
1641
1642 while (*String != '$')
1643 {
1644 DosPrintCharacter(*String);
1645 String++;
1646 }
1647
1648 /*
1649 * We return the terminating character (DOS 2.1+).
1650 * See Ralf Brown: http://www.ctyme.com/intr/rb-2562.htm
1651 * for more information.
1652 */
1653 setAL('$');
1654 break;
1655 }
1656
1657 /* Read Buffered Input */
1658 case 0x0A:
1659 {
1660 InputBuffer = (PDOS_INPUT_BUFFER)SEG_OFF_TO_PTR(getDS(), getDX());
1661
1662 while (Stack[STACK_COUNTER] < InputBuffer->MaxLength)
1663 {
1664 /* Try to read a character */
1665 Character = DosReadCharacter();
1666
1667 /* If it's not ready yet, let the BOP repeat */
1668 if (getCF()) break;
1669
1670 /* Echo the character and append it to the buffer */
1671 DosPrintCharacter(Character);
1672 InputBuffer->Buffer[Stack[STACK_COUNTER]] = Character;
1673
1674 if (Character == '\r') break;
1675 Stack[STACK_COUNTER]++;
1676 }
1677
1678 /* Update the length */
1679 InputBuffer->Length = Stack[STACK_COUNTER];
1680 break;
1681 }
1682
1683 /* Get STDIN Status */
1684 case 0x0B:
1685 {
1686 setAL(DosCheckInput() ? 0xFF : 0x00);
1687 break;
1688 }
1689
1690 /* Flush Buffer and Read STDIN */
1691 case 0x0C:
1692 {
1693 BYTE InputFunction = getAL();
1694
1695 /* Flush STDIN buffer */
1696 DosFlushFileBuffers(DOS_INPUT_HANDLE); // Maybe just create a DosFlushInputBuffer...
1697
1698 /*
1699 * If the input function number contained in AL is valid, i.e.
1700 * AL == 0x01 or 0x06 or 0x07 or 0x08 or 0x0A, call ourselves
1701 * recursively with AL == AH.
1702 */
1703 if (InputFunction == 0x01 || InputFunction == 0x06 ||
1704 InputFunction == 0x07 || InputFunction == 0x08 ||
1705 InputFunction == 0x0A)
1706 {
1707 setAH(InputFunction);
1708 /*
1709 * Instead of calling ourselves really recursively as in:
1710 * DosInt21h(Stack);
1711 * prefer resetting the CF flag to let the BOP repeat.
1712 */
1713 setCF(1);
1714 }
1715 break;
1716 }
1717
1718 /* Disk Reset */
1719 case 0x0D:
1720 {
1721 PDOS_PSP PspBlock = SEGMENT_TO_PSP(CurrentPsp);
1722
1723 // TODO: Flush what's needed.
1724 DPRINT1("INT 21h, 0Dh is UNIMPLEMENTED\n");
1725
1726 /* Clear CF in DOS 6 only */
1727 if (PspBlock->DosVersion == 0x0006)
1728 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
1729
1730 break;
1731 }
1732
1733 /* Set Default Drive */
1734 case 0x0E:
1735 {
1736 DosChangeDrive(getDL());
1737 setAL(LastDrive - 'A' + 1);
1738 break;
1739 }
1740
1741 /* NULL Function for CP/M Compatibility */
1742 case 0x18:
1743 {
1744 /*
1745 * This function corresponds to the CP/M BDOS function
1746 * "get bit map of logged drives", which is meaningless
1747 * under MS-DOS.
1748 *
1749 * For: PTS-DOS 6.51 & S/DOS 1.0 - EXTENDED RENAME FILE USING FCB
1750 * See Ralf Brown: http://www.ctyme.com/intr/rb-2584.htm
1751 * for more information.
1752 */
1753 setAL(0x00);
1754 break;
1755 }
1756
1757 /* Get Default Drive */
1758 case 0x19:
1759 {
1760 setAL(CurrentDrive);
1761 break;
1762 }
1763
1764 /* Set Disk Transfer Area */
1765 case 0x1A:
1766 {
1767 DiskTransferArea = MAKELONG(getDX(), getDS());
1768 break;
1769 }
1770
1771 /* NULL Function for CP/M Compatibility */
1772 case 0x1D:
1773 case 0x1E:
1774 {
1775 /*
1776 * Function 0x1D corresponds to the CP/M BDOS function
1777 * "get bit map of read-only drives", which is meaningless
1778 * under MS-DOS.
1779 * See Ralf Brown: http://www.ctyme.com/intr/rb-2592.htm
1780 * for more information.
1781 *
1782 * Function 0x1E corresponds to the CP/M BDOS function
1783 * "set file attributes", which was meaningless under MS-DOS 1.x.
1784 * See Ralf Brown: http://www.ctyme.com/intr/rb-2593.htm
1785 * for more information.
1786 */
1787 setAL(0x00);
1788 break;
1789 }
1790
1791 /* NULL Function for CP/M Compatibility */
1792 case 0x20:
1793 {
1794 /*
1795 * This function corresponds to the CP/M BDOS function
1796 * "get/set default user (sublibrary) number", which is meaningless
1797 * under MS-DOS.
1798 *
1799 * For: S/DOS 1.0+ & PTS-DOS 6.51+ - GET OEM REVISION
1800 * See Ralf Brown: http://www.ctyme.com/intr/rb-2596.htm
1801 * for more information.
1802 */
1803 setAL(0x00);
1804 break;
1805 }
1806
1807 /* Set Interrupt Vector */
1808 case 0x25:
1809 {
1810 ULONG FarPointer = MAKELONG(getDX(), getDS());
1811 DPRINT1("Setting interrupt 0x%x ...\n", getAL());
1812
1813 /* Write the new far pointer to the IDT */
1814 ((PULONG)BaseAddress)[getAL()] = FarPointer;
1815 break;
1816 }
1817
1818 /* Create New PSP */
1819 case 0x26:
1820 {
1821 DPRINT1("INT 21h, 26h - Create New PSP is UNIMPLEMENTED\n");
1822 break;
1823 }
1824
1825 /* Get System Date */
1826 case 0x2A:
1827 {
1828 GetLocalTime(&SystemTime);
1829 setCX(SystemTime.wYear);
1830 setDX(MAKEWORD(SystemTime.wDay, SystemTime.wMonth));
1831 setAL(SystemTime.wDayOfWeek);
1832 break;
1833 }
1834
1835 /* Set System Date */
1836 case 0x2B:
1837 {
1838 GetLocalTime(&SystemTime);
1839 SystemTime.wYear = getCX();
1840 SystemTime.wMonth = getDH();
1841 SystemTime.wDay = getDL();
1842
1843 /* Return success or failure */
1844 setAL(SetLocalTime(&SystemTime) ? 0x00 : 0xFF);
1845 break;
1846 }
1847
1848 /* Get System Time */
1849 case 0x2C:
1850 {
1851 GetLocalTime(&SystemTime);
1852 setCX(MAKEWORD(SystemTime.wMinute, SystemTime.wHour));
1853 setDX(MAKEWORD(SystemTime.wMilliseconds / 10, SystemTime.wSecond));
1854 break;
1855 }
1856
1857 /* Set System Time */
1858 case 0x2D:
1859 {
1860 GetLocalTime(&SystemTime);
1861 SystemTime.wHour = getCH();
1862 SystemTime.wMinute = getCL();
1863 SystemTime.wSecond = getDH();
1864 SystemTime.wMilliseconds = getDL() * 10; // In hundredths of seconds
1865
1866 /* Return success or failure */
1867 setAL(SetLocalTime(&SystemTime) ? 0x00 : 0xFF);
1868 break;
1869 }
1870
1871 /* Get Disk Transfer Area */
1872 case 0x2F:
1873 {
1874 setES(HIWORD(DiskTransferArea));
1875 setBX(LOWORD(DiskTransferArea));
1876 break;
1877 }
1878
1879 /* Get DOS Version */
1880 case 0x30:
1881 {
1882 PDOS_PSP PspBlock = SEGMENT_TO_PSP(CurrentPsp);
1883
1884 /*
1885 * DOS 2+ - GET DOS VERSION
1886 * See Ralf Brown: http://www.ctyme.com/intr/rb-2711.htm
1887 * for more information.
1888 */
1889
1890 if (LOBYTE(PspBlock->DosVersion) < 5 || getAL() == 0x00)
1891 {
1892 /*
1893 * Return DOS OEM number:
1894 * 0x00 for IBM PC-DOS
1895 * 0x02 for packaged MS-DOS
1896 */
1897 setBH(0x02);
1898 }
1899
1900 if (LOBYTE(PspBlock->DosVersion) >= 5 && getAL() == 0x01)
1901 {
1902 /*
1903 * Return version flag:
1904 * 1 << 3 if DOS is in ROM,
1905 * 0 (reserved) if not.
1906 */
1907 setBH(0x00);
1908 }
1909
1910 /* Return DOS 24-bit user serial number in BL:CX */
1911 setBL(0x00);
1912 setCX(0x0000);
1913
1914 /*
1915 * Return DOS version: Minor:Major in AH:AL
1916 * The Windows NT DOS box returns version 5.00, subject to SETVER.
1917 */
1918 setAX(PspBlock->DosVersion);
1919
1920 break;
1921 }
1922
1923 /* Extended functionalities */
1924 case 0x33:
1925 {
1926 if (getAL() == 0x06)
1927 {
1928 /*
1929 * DOS 5+ - GET TRUE VERSION NUMBER
1930 * This function always returns the true version number, unlike
1931 * AH=30h, whose return value may be changed with SETVER.
1932 * See Ralf Brown: http://www.ctyme.com/intr/rb-2730.htm
1933 * for more information.
1934 */
1935
1936 /*
1937 * Return the true DOS version: Minor:Major in BH:BL
1938 * The Windows NT DOS box returns BX=3205h (version 5.50).
1939 */
1940 setBX(NTDOS_VERSION);
1941
1942 /* DOS revision 0 */
1943 setDL(0x00);
1944
1945 /* Unpatched DOS */
1946 setDH(0x00);
1947 }
1948 // else
1949 // {
1950 // /* Invalid subfunction */
1951 // setAL(0xFF);
1952 // }
1953
1954 break;
1955 }
1956
1957 /* Get Interrupt Vector */
1958 case 0x35:
1959 {
1960 DWORD FarPointer = ((PDWORD)BaseAddress)[getAL()];
1961
1962 /* Read the address from the IDT into ES:BX */
1963 setES(HIWORD(FarPointer));
1964 setBX(LOWORD(FarPointer));
1965 break;
1966 }
1967
1968 /* SWITCH character - AVAILDEV */
1969 case 0x37:
1970 {
1971 if (getAL() == 0x00)
1972 {
1973 /*
1974 * DOS 2+ - "SWITCHAR" - GET SWITCH CHARACTER
1975 * This setting is ignored by MS-DOS 4.0+.
1976 * MS-DOS 5+ always return AL=00h/DL=2Fh.
1977 * See Ralf Brown: http://www.ctyme.com/intr/rb-2752.htm
1978 * for more information.
1979 */
1980 setDL('/');
1981 setAL(0x00);
1982 }
1983 else if (getAL() == 0x01)
1984 {
1985 /*
1986 * DOS 2+ - "SWITCHAR" - SET SWITCH CHARACTER
1987 * This setting is ignored by MS-DOS 5+.
1988 * See Ralf Brown: http://www.ctyme.com/intr/rb-2753.htm
1989 * for more information.
1990 */
1991 // getDL();
1992 setAL(0xFF);
1993 }
1994 else if (getAL() == 0x02)
1995 {
1996 /*
1997 * DOS 2.x and 3.3+ only - "AVAILDEV" - SPECIFY \DEV\ PREFIX USE
1998 * See Ralf Brown: http://www.ctyme.com/intr/rb-2754.htm
1999 * for more information.
2000 */
2001 // setDL();
2002 setAL(0xFF);
2003 }
2004 else if (getAL() == 0x03)
2005 {
2006 /*
2007 * DOS 2.x and 3.3+ only - "AVAILDEV" - SPECIFY \DEV\ PREFIX USE
2008 * See Ralf Brown: http://www.ctyme.com/intr/rb-2754.htm
2009 * for more information.
2010 */
2011 // getDL();
2012 setAL(0xFF);
2013 }
2014 else
2015 {
2016 /* Invalid subfunction */
2017 setAL(0xFF);
2018 }
2019
2020 break;
2021 }
2022
2023 /* Create Directory */
2024 case 0x39:
2025 {
2026 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getDX());
2027
2028 if (CreateDirectoryA(String, NULL))
2029 {
2030 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2031 }
2032 else
2033 {
2034 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2035 setAX(LOWORD(GetLastError()));
2036 }
2037
2038 break;
2039 }
2040
2041 /* Remove Directory */
2042 case 0x3A:
2043 {
2044 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getDX());
2045
2046 if (RemoveDirectoryA(String))
2047 {
2048 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2049 }
2050 else
2051 {
2052 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2053 setAX(LOWORD(GetLastError()));
2054 }
2055
2056 break;
2057 }
2058
2059 /* Set Current Directory */
2060 case 0x3B:
2061 {
2062 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getDX());
2063
2064 if (DosChangeDirectory(String))
2065 {
2066 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2067 }
2068 else
2069 {
2070 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2071 setAX(DosLastError);
2072 }
2073
2074 break;
2075 }
2076
2077 /* Create File */
2078 case 0x3C:
2079 {
2080 WORD FileHandle;
2081 WORD ErrorCode = DosCreateFile(&FileHandle,
2082 (LPCSTR)SEG_OFF_TO_PTR(getDS(), getDX()),
2083 getCX());
2084
2085 if (ErrorCode == 0)
2086 {
2087 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2088 setAX(FileHandle);
2089 }
2090 else
2091 {
2092 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2093 setAX(ErrorCode);
2094 }
2095
2096 break;
2097 }
2098
2099 /* Open File */
2100 case 0x3D:
2101 {
2102 WORD FileHandle;
2103 WORD ErrorCode = DosOpenFile(&FileHandle,
2104 (LPCSTR)SEG_OFF_TO_PTR(getDS(), getDX()),
2105 getAL());
2106
2107 if (ErrorCode == 0)
2108 {
2109 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2110 setAX(FileHandle);
2111 }
2112 else
2113 {
2114 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2115 setAX(ErrorCode);
2116 }
2117
2118 break;
2119 }
2120
2121 /* Close File */
2122 case 0x3E:
2123 {
2124 if (DosCloseHandle(getBX()))
2125 {
2126 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2127 }
2128 else
2129 {
2130 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2131 setAX(ERROR_INVALID_HANDLE);
2132 }
2133
2134 break;
2135 }
2136
2137 /* Read from File or Device */
2138 case 0x3F:
2139 {
2140 WORD Handle = getBX();
2141 LPBYTE Buffer = (LPBYTE)SEG_OFF_TO_PTR(getDS(), getDX());
2142 WORD Count = getCX();
2143 WORD BytesRead = 0;
2144 WORD ErrorCode = ERROR_SUCCESS;
2145 CHAR Character;
2146
2147 if (IsConsoleHandle(DosGetRealHandle(Handle)))
2148 {
2149 while (Stack[STACK_COUNTER] < Count)
2150 {
2151 /* Read a character from the BIOS */
2152 Character = LOBYTE(BiosGetCharacter());
2153
2154 /* Stop if the BOP needs to be repeated */
2155 if (getCF()) break;
2156
2157 // FIXME: Security checks!
2158 DosPrintCharacter(Character);
2159 Buffer[Stack[STACK_COUNTER]++] = Character;
2160
2161 if (Character == '\r')
2162 {
2163 /* Stop on first carriage return */
2164 DosPrintCharacter('\n');
2165 break;
2166 }
2167 }
2168
2169 if (Character != '\r')
2170 {
2171 if (Stack[STACK_COUNTER] < Count) ErrorCode = ERROR_NOT_READY;
2172 else BytesRead = Count;
2173 }
2174 else BytesRead = Stack[STACK_COUNTER];
2175 }
2176 else
2177 {
2178 /* Use the file reading function */
2179 ErrorCode = DosReadFile(Handle, Buffer, Count, &BytesRead);
2180 }
2181
2182 if (ErrorCode == ERROR_SUCCESS)
2183 {
2184 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2185 setAX(BytesRead);
2186 }
2187 else if (ErrorCode != ERROR_NOT_READY)
2188 {
2189 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2190 setAX(ErrorCode);
2191 }
2192 break;
2193 }
2194
2195 /* Write to File or Device */
2196 case 0x40:
2197 {
2198 WORD BytesWritten = 0;
2199 WORD ErrorCode = DosWriteFile(getBX(),
2200 SEG_OFF_TO_PTR(getDS(), getDX()),
2201 getCX(),
2202 &BytesWritten);
2203
2204 if (ErrorCode == ERROR_SUCCESS)
2205 {
2206 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2207 setAX(BytesWritten);
2208 }
2209 else
2210 {
2211 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2212 setAX(ErrorCode);
2213 }
2214
2215 break;
2216 }
2217
2218 /* Delete File */
2219 case 0x41:
2220 {
2221 LPSTR FileName = (LPSTR)SEG_OFF_TO_PTR(getDS(), getDX());
2222
2223 if (demFileDelete(FileName) == ERROR_SUCCESS)
2224 {
2225 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2226 /*
2227 * See Ralf Brown: http://www.ctyme.com/intr/rb-2797.htm
2228 * "AX destroyed (DOS 3.3) AL seems to be drive of deleted file."
2229 */
2230 setAL(FileName[0] - 'A');
2231 }
2232 else
2233 {
2234 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2235 setAX(GetLastError());
2236 }
2237
2238 break;
2239 }
2240
2241 /* Seek File */
2242 case 0x42:
2243 {
2244 DWORD NewLocation;
2245 WORD ErrorCode = DosSeekFile(getBX(),
2246 MAKELONG(getDX(), getCX()),
2247 getAL(),
2248 &NewLocation);
2249
2250 if (ErrorCode == ERROR_SUCCESS)
2251 {
2252 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2253
2254 /* Return the new offset in DX:AX */
2255 setDX(HIWORD(NewLocation));
2256 setAX(LOWORD(NewLocation));
2257 }
2258 else
2259 {
2260 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2261 setAX(ErrorCode);
2262 }
2263
2264 break;
2265 }
2266
2267 /* Get/Set File Attributes */
2268 case 0x43:
2269 {
2270 DWORD Attributes;
2271 LPSTR FileName = (LPSTR)SEG_OFF_TO_PTR(getDS(), getDX());
2272
2273 if (getAL() == 0x00)
2274 {
2275 /* Get the attributes */
2276 Attributes = GetFileAttributesA(FileName);
2277
2278 /* Check if it failed */
2279 if (Attributes == INVALID_FILE_ATTRIBUTES)
2280 {
2281 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2282 setAX(GetLastError());
2283 }
2284 else
2285 {
2286 /* Return the attributes that DOS can understand */
2287 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2288 setCX(Attributes & 0x00FF);
2289 }
2290 }
2291 else if (getAL() == 0x01)
2292 {
2293 /* Try to set the attributes */
2294 if (SetFileAttributesA(FileName, getCL()))
2295 {
2296 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2297 }
2298 else
2299 {
2300 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2301 setAX(GetLastError());
2302 }
2303 }
2304 else
2305 {
2306 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2307 setAX(ERROR_INVALID_FUNCTION);
2308 }
2309
2310 break;
2311 }
2312
2313 /* IOCTL */
2314 case 0x44:
2315 {
2316 if (DosHandleIoctl(getAL(), getBX()))
2317 {
2318 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2319 }
2320 else
2321 {
2322 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2323 setAX(DosLastError);
2324 }
2325
2326 break;
2327 }
2328
2329 /* Duplicate Handle */
2330 case 0x45:
2331 {
2332 WORD NewHandle;
2333 HANDLE Handle = DosGetRealHandle(getBX());
2334
2335 if (Handle != INVALID_HANDLE_VALUE)
2336 {
2337 /* The handle is invalid */
2338 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2339 setAX(ERROR_INVALID_HANDLE);
2340 break;
2341 }
2342
2343 /* Open a new handle to the same entry */
2344 NewHandle = DosOpenHandle(Handle);
2345
2346 if (NewHandle == INVALID_DOS_HANDLE)
2347 {
2348 /* Too many files open */
2349 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2350 setAX(ERROR_TOO_MANY_OPEN_FILES);
2351 break;
2352 }
2353
2354 /* Return the result */
2355 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2356 setAX(NewHandle);
2357 break;
2358 }
2359
2360 /* Force Duplicate Handle */
2361 case 0x46:
2362 {
2363 if (DosDuplicateHandle(getBX(), getCX()))
2364 {
2365 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2366 }
2367 else
2368 {
2369 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2370 setAX(ERROR_INVALID_HANDLE);
2371 }
2372
2373 break;
2374 }
2375
2376 /* Get Current Directory */
2377 case 0x47:
2378 {
2379 BYTE DriveNumber = getDL();
2380 String = (PCHAR)SEG_OFF_TO_PTR(getDS(), getSI());
2381
2382 /* Get the real drive number */
2383 if (DriveNumber == 0)
2384 {
2385 DriveNumber = CurrentDrive;
2386 }
2387 else
2388 {
2389 /* Decrement DriveNumber since it was 1-based */
2390 DriveNumber--;
2391 }
2392
2393 if (DriveNumber <= LastDrive - 'A')
2394 {
2395 /*
2396 * Copy the current directory into the target buffer.
2397 * It doesn't contain the drive letter and the backslash.
2398 */
2399 strncpy(String, CurrentDirectories[DriveNumber], DOS_DIR_LENGTH);
2400 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2401 setAX(0x0100); // Undocumented, see Ralf Brown: http://www.ctyme.com/intr/rb-2933.htm
2402 }
2403 else
2404 {
2405 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2406 setAX(ERROR_INVALID_DRIVE);
2407 }
2408
2409 break;
2410 }
2411
2412 /* Allocate Memory */
2413 case 0x48:
2414 {
2415 WORD MaxAvailable = 0;
2416 WORD Segment = DosAllocateMemory(getBX(), &MaxAvailable);
2417
2418 if (Segment != 0)
2419 {
2420 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2421 setAX(Segment);
2422 }
2423 else
2424 {
2425 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2426 setAX(DosLastError);
2427 setBX(MaxAvailable);
2428 }
2429
2430 break;
2431 }
2432
2433 /* Free Memory */
2434 case 0x49:
2435 {
2436 if (DosFreeMemory(getES()))
2437 {
2438 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2439 }
2440 else
2441 {
2442 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2443 setAX(ERROR_ARENA_TRASHED);
2444 }
2445
2446 break;
2447 }
2448
2449 /* Resize Memory Block */
2450 case 0x4A:
2451 {
2452 WORD Size;
2453
2454 if (DosResizeMemory(getES(), getBX(), &Size))
2455 {
2456 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2457 }
2458 else
2459 {
2460 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2461 setAX(DosLastError);
2462 setBX(Size);
2463 }
2464
2465 break;
2466 }
2467
2468 /* Execute */
2469 case 0x4B:
2470 {
2471 DOS_EXEC_TYPE LoadType = (DOS_EXEC_TYPE)getAL();
2472 LPSTR ProgramName = SEG_OFF_TO_PTR(getDS(), getDX());
2473 PDOS_EXEC_PARAM_BLOCK ParamBlock = SEG_OFF_TO_PTR(getES(), getBX());
2474 WORD ErrorCode = DosCreateProcess(LoadType, ProgramName, ParamBlock);
2475
2476 if (ErrorCode == ERROR_SUCCESS)
2477 {
2478 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2479 }
2480 else
2481 {
2482 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2483 setAX(ErrorCode);
2484 }
2485
2486 break;
2487 }
2488
2489 /* Terminate With Return Code */
2490 case 0x4C:
2491 {
2492 DosTerminateProcess(CurrentPsp, getAL());
2493 break;
2494 }
2495
2496 /* Get Return Code (ERRORLEVEL) */
2497 case 0x4D:
2498 {
2499 /*
2500 * According to Ralf Brown: http://www.ctyme.com/intr/rb-2976.htm
2501 * DosErrorLevel is cleared after being read by this function.
2502 */
2503 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2504 setAX(DosErrorLevel);
2505 DosErrorLevel = 0x0000; // Clear it
2506 break;
2507 }
2508
2509 /* Find First File */
2510 case 0x4E:
2511 {
2512 WORD Result = (WORD)demFileFindFirst(FAR_POINTER(DiskTransferArea),
2513 SEG_OFF_TO_PTR(getDS(), getDX()),
2514 getCX());
2515
2516 setAX(Result);
2517 if (Result == ERROR_SUCCESS) Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2518 else Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2519
2520 break;
2521 }
2522
2523 /* Find Next File */
2524 case 0x4F:
2525 {
2526 WORD Result = (WORD)demFileFindNext(FAR_POINTER(DiskTransferArea));
2527
2528 setAX(Result);
2529 if (Result == ERROR_SUCCESS) Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2530 else Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2531
2532 break;
2533 }
2534
2535 /* Internal - Set Current Process ID (Set PSP Address) */
2536 case 0x50:
2537 {
2538 // FIXME: Is it really what it's done ??
2539 CurrentPsp = getBX();
2540 break;
2541 }
2542
2543 /* Internal - Get Current Process ID (Get PSP Address) */
2544 case 0x51:
2545 /* Get Current PSP Address */
2546 case 0x62:
2547 {
2548 /*
2549 * Undocumented AH=51h is identical to the documented AH=62h.
2550 * See Ralf Brown: http://www.ctyme.com/intr/rb-2982.htm
2551 * and http://www.ctyme.com/intr/rb-3140.htm
2552 * for more information.
2553 */
2554 setBX(CurrentPsp);
2555 break;
2556 }
2557
2558 /* Get/Set Memory Management Options */
2559 case 0x58:
2560 {
2561 if (getAL() == 0x00)
2562 {
2563 /* Get allocation strategy */
2564 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2565 setAX(DosAllocStrategy);
2566 }
2567 else if (getAL() == 0x01)
2568 {
2569 /* Set allocation strategy */
2570
2571 if ((getBL() & (DOS_ALLOC_HIGH | DOS_ALLOC_HIGH_LOW))
2572 == (DOS_ALLOC_HIGH | DOS_ALLOC_HIGH_LOW))
2573 {
2574 /* Can't set both */
2575 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2576 setAX(ERROR_INVALID_PARAMETER);
2577 break;
2578 }
2579
2580 if ((getBL() & 0x3F) > DOS_ALLOC_LAST_FIT)
2581 {
2582 /* Invalid allocation strategy */
2583 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2584 setAX(ERROR_INVALID_PARAMETER);
2585 break;
2586 }
2587
2588 DosAllocStrategy = getBL();
2589 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2590 }
2591 else if (getAL() == 0x02)
2592 {
2593 /* Get UMB link state */
2594 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2595 setAL(DosUmbLinked ? 0x01 : 0x00);
2596 }
2597 else if (getAL() == 0x03)
2598 {
2599 /* Set UMB link state */
2600 if (getBX()) DosLinkUmb();
2601 else DosUnlinkUmb();
2602 Stack[STACK_FLAGS] &= ~EMULATOR_FLAG_CF;
2603 }
2604 else
2605 {
2606 /* Invalid or unsupported function */
2607 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2608 setAX(ERROR_INVALID_FUNCTION);
2609 }
2610
2611 break;
2612 }
2613
2614 /* Unsupported */
2615 default:
2616 {
2617 DPRINT1("DOS Function INT 0x21, AH = %xh, AL = %xh NOT IMPLEMENTED!\n",
2618 getAH(), getAL());
2619
2620 setAL(0); // Some functions expect AL to be 0 when it's not supported.
2621 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2622 }
2623 }
2624 }
2625
2626 VOID WINAPI DosBreakInterrupt(LPWORD Stack)
2627 {
2628 UNREFERENCED_PARAMETER(Stack);
2629
2630 /* Stop the VDM */
2631 VdmRunning = FALSE;
2632 }
2633
2634 VOID WINAPI DosFastConOut(LPWORD Stack)
2635 {
2636 /*
2637 * This is the DOS 2+ Fast Console Output Interrupt.
2638 * See Ralf Brown: http://www.ctyme.com/intr/rb-4124.htm
2639 * for more information.
2640 */
2641
2642 #if 0
2643 if (Stack[STACK_COUNTER] == 0)
2644 {
2645 Stack[STACK_COUNTER]++;
2646
2647 /* Save AX and BX */
2648 Stack[STACK_VAR_A] = getAX();
2649 Stack[STACK_VAR_B] = getBX();
2650
2651 /* Rewind the BOP manually, we can't use CF because the interrupt could modify it */
2652 EmulatorExecute(getCS(), getIP() - 4);
2653
2654 /* Call INT 0x10, AH = 0x0E */
2655 setAH(0x0E);
2656 setBL(DOS_CHAR_ATTRIBUTE);
2657 setBH(Bda->VideoPage);
2658
2659 EmulatorInterrupt(0x10);
2660 }
2661 else
2662 {
2663 /* Restore AX and BX */
2664 setAX(Stack[STACK_VAR_A]);
2665 setBX(Stack[STACK_VAR_B]);
2666 }
2667 #else
2668 /* Save AX and BX */
2669 USHORT AX = getAX();
2670 USHORT BX = getBX();
2671
2672 /* Set the parameters (AL = character, already set) */
2673 setBL(DOS_CHAR_ATTRIBUTE);
2674 setBH(Bda->VideoPage);
2675
2676 /* Call the BIOS INT 10h, AH=0Eh "Teletype Output" */
2677 setAH(0x0E);
2678 Int32Call(&DosContext, BIOS_VIDEO_INTERRUPT);
2679
2680 /* Restore AX and BX */
2681 setBX(BX);
2682 setAX(AX);
2683 #endif
2684 }
2685
2686 VOID WINAPI DosInt2Fh(LPWORD Stack)
2687 {
2688 DPRINT1("DOS System Function INT 0x2F, AH = %xh, AL = %xh NOT IMPLEMENTED!\n",
2689 getAH(), getAL());
2690 Stack[STACK_FLAGS] |= EMULATOR_FLAG_CF;
2691 }
2692
2693 BOOLEAN DosKRNLInitialize(VOID)
2694 {
2695
2696 #if 1
2697
2698 UCHAR i;
2699 CHAR CurrentDirectory[MAX_PATH];
2700 CHAR DosDirectory[DOS_DIR_LENGTH];
2701 LPSTR Path;
2702
2703 FILE *Stream;
2704 WCHAR Buffer[256];
2705
2706 /* Clear the current directory buffer */
2707 ZeroMemory(CurrentDirectories, sizeof(CurrentDirectories));
2708
2709 /* Get the current directory */
2710 if (!GetCurrentDirectoryA(MAX_PATH, CurrentDirectory))
2711 {
2712 // TODO: Use some kind of default path?
2713 return FALSE;
2714 }
2715
2716 /* Convert that to a DOS path */
2717 if (!GetShortPathNameA(CurrentDirectory, DosDirectory, DOS_DIR_LENGTH))
2718 {
2719 // TODO: Use some kind of default path?
2720 return FALSE;
2721 }
2722
2723 /* Set the drive */
2724 CurrentDrive = DosDirectory[0] - 'A';
2725
2726 /* Get the directory part of the path */
2727 Path = strchr(DosDirectory, '\\');
2728 if (Path != NULL)
2729 {
2730 /* Skip the backslash */
2731 Path++;
2732 }
2733
2734 /* Set the directory */
2735 if (Path != NULL)
2736 {
2737 strncpy(CurrentDirectories[CurrentDrive], Path, DOS_DIR_LENGTH);
2738 }
2739
2740 /* Read CONFIG.SYS */
2741 Stream = _wfopen(DOS_CONFIG_PATH, L"r");
2742 if (Stream != NULL)
2743 {
2744 while (fgetws(Buffer, 256, Stream))
2745 {
2746 // TODO: Parse the line
2747 }
2748 fclose(Stream);
2749 }
2750
2751 /* Initialize the SFT */
2752 for (i = 0; i < DOS_SFT_SIZE; i++)
2753 {
2754 DosSystemFileTable[i] = INVALID_HANDLE_VALUE;
2755 DosSftRefCount[i] = 0;
2756 }
2757
2758 /* Get handles to standard I/O devices */
2759 DosSystemFileTable[0] = GetStdHandle(STD_INPUT_HANDLE);
2760 DosSystemFileTable[1] = GetStdHandle(STD_OUTPUT_HANDLE);
2761 DosSystemFileTable[2] = GetStdHandle(STD_ERROR_HANDLE);
2762
2763 #endif
2764
2765 /* Initialize the callback context */
2766 InitializeContext(&DosContext, 0x0070, 0x0000);
2767
2768 /* Register the DOS 32-bit Interrupts */
2769 RegisterDosInt32(0x20, DosInt20h );
2770 RegisterDosInt32(0x21, DosInt21h );
2771 // RegisterDosInt32(0x22, DosInt22h ); // Termination
2772 RegisterDosInt32(0x23, DosBreakInterrupt); // Ctrl-C / Ctrl-Break
2773 // RegisterDosInt32(0x24, DosInt24h ); // Critical Error
2774 RegisterDosInt32(0x29, DosFastConOut ); // DOS 2+ Fast Console Output
2775 RegisterDosInt32(0x2F, DosInt2Fh );
2776
2777 return TRUE;
2778 }
2779
2780 /* EOF */