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