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