home *** CD-ROM | disk | FTP | other *** search
/ Liren Large Software Subsidy 10 / 10.iso / l / l440 / 2.ddi / CHAP3 / MEM.C < prev    next >
Encoding:
C/C++ Source or Header  |  1990-09-25  |  6.9 KB  |  273 lines
  1. /* 
  2. MEM.C -- walks DOS MCB chain(s): detailed version
  3. Andrew Schulman and Jim Kyle, July 1990
  4. */
  5.  
  6. #include <stdlib.h>
  7. #include <stdio.h>
  8. #include <ctype.h>
  9. #include <string.h>
  10. #include <dos.h>
  11.  
  12. typedef enum { FALSE, TRUE } BOOL;
  13. typedef unsigned char BYTE;
  14. typedef unsigned short WORD;
  15. typedef unsigned long ULONG;
  16. typedef void far *FP;
  17.  
  18. #ifndef MK_FP
  19. #define MK_FP(seg,ofs)  ((FP)(((ULONG)(seg) << 16) | (ofs)))
  20. #endif
  21.  
  22. #ifdef __TURBOC__
  23. #define ASM asm
  24. #else
  25. #define ASM _asm
  26. #endif
  27.  
  28. #ifdef __TURBOC__
  29. #define GETVECT(x)      getvect(x)
  30. #else
  31. #define GETVECT(x)      _dos_getvect(x)
  32. #endif
  33.  
  34. #pragma pack(1)
  35.  
  36. typedef struct {
  37.     BYTE type;          /* 'M'=in chain; 'Z'=at end */
  38.     WORD owner;         /* PSP of the owner */
  39.     WORD size;          /* in 16-byte paragraphs */
  40.     BYTE unused[3];
  41.     BYTE dos4[8];
  42.     } MCB;
  43.  
  44. #define MCB_FM_SEG(seg)     ((seg) - 1)
  45. #define IS_PSP(mcb)         (FP_SEG(mcb) + 1 == (mcb)->owner)
  46. #define ENV_FM_PSP(psp_seg) (*((WORD far *) MK_FP(psp_seg, 0x2c)))
  47.  
  48. void fail(char *s) { puts(s); exit(1); }
  49.  
  50. BOOL belongs(void far *vec, unsigned start, unsigned size);
  51. void display(MCB far *mcb);
  52. char far *env(MCB far *mcb);
  53. void display_progname(MCB far *mcb);
  54. void display_cmdline(MCB far *mcb);
  55. void display_vectors(MCB far *mcb);
  56. unsigned fstrlen(char far *s);
  57.  
  58. MCB far *get_mcb(void)
  59. {
  60.     ASM mov ah, 52h
  61.     ASM int 21h
  62.     ASM mov dx, es:[bx-2]
  63.     ASM xor ax, ax
  64.     /* in both Microsoft C and Turbo C, far* returned in DX:AX */
  65. }
  66.  
  67. mcb_chk(MCB far *mcb)
  68. {
  69.     for (;;)
  70.         if (mcb->type == 'M')
  71.             mcb = MK_FP(FP_SEG(mcb) + mcb->size + 1, 0);
  72.         else
  73.             return (mcb->type == 'Z');
  74. }
  75.  
  76. void walk(MCB far *mcb)
  77. {
  78.     printf("Seg     Owner   Size\n"); 
  79.     for (;;)
  80.         switch (mcb->type)
  81.         {
  82.             case 'M' : /* Mark : belongs to MCB chain */
  83.                 display(mcb);
  84.                 mcb = MK_FP(FP_SEG(mcb) + mcb->size + 1, 0);
  85.                 break;
  86.             case 'Z' : /* Zbikowski : end of MCB chain */
  87.                 display(mcb);
  88.                 return;
  89.             default :
  90.                 fail("error in MCB chain");
  91.         }
  92. }
  93.  
  94. main(int argc, char *argv[])
  95. {
  96.     if (argc < 2)
  97.         walk(get_mcb());            /* walk "normal" MCB chain */
  98.     else
  99.     {
  100.         unsigned seg;
  101.         sscanf(argv[1], "%04X", &seg);
  102.         walk(MK_FP(seg, 0));       /* walk arbitrary MCB chain */
  103.     }
  104.  
  105.     return 0;
  106. }
  107.  
  108. void display(MCB far *mcb)
  109. {
  110.     static void far *vect_2e = (void far *) 0;
  111.     unsigned env_seg;
  112.     
  113.     printf("%04X    %04X    %04X (%6lu)   ", 
  114.         FP_SEG(mcb), mcb->owner, mcb->size, (long) mcb->size << 4);
  115.  
  116.     if (IS_PSP(mcb))
  117.     {
  118.         void far *e = env(mcb);        /* MSC wants lvalue */
  119.         if (env_seg = FP_SEG(e)) printf("%04X    ", env_seg);
  120.         else                     printf("        ");
  121.     
  122.         display_progname(mcb);
  123.     }
  124.     
  125.     if (! vect_2e) 
  126.         vect_2e = GETVECT(0x2e);   /* do just once */
  127.     if (! mcb->owner)
  128.         printf("free ");
  129.     /* 0008 is not really a PSP; belongs to CONFIG.SYS */
  130.     else if (mcb->owner == 8)   
  131.         printf("config ");
  132.     /* INT 2Eh belongs to master COMMAND.COM (or other shell) */
  133.     else if (belongs(vect_2e, FP_SEG(mcb), mcb->size))
  134.         printf("%s ", getenv("COMSPEC"));   
  135.  
  136.     /* presence command line is independent of program name */
  137.     if (IS_PSP(mcb))
  138.         display_cmdline(mcb);
  139.     display_vectors(mcb);
  140.     printf("\n");
  141. }
  142.  
  143. char far *env(MCB far *mcb)
  144. {
  145.     char far *e;
  146.     unsigned env_mcb;
  147.     unsigned env_owner;
  148.  
  149.     /*
  150.         if the MCB owner is one more than the MCB segment then
  151.             psp := MCB owner
  152.             env_seg := make_far_pointer(psp, 2Ch)
  153.             e := make_far_pointer(env_seg, 0)
  154.         else
  155.             return NULL
  156.     */
  157.     if (IS_PSP(mcb))
  158.         e = MK_FP(ENV_FM_PSP(mcb->owner), 0);
  159.     else 
  160.         return (char far *) 0;
  161.     
  162.     /* 
  163.        Does this environment really belong to this PSP? An
  164.        environment is just another memory block, so its MCB is
  165.        located in the preceding paragraph. Make sure the env
  166.        MCB's owner is equal to the PSP whose environment this
  167.        supposedly is! Thanks to Rob Adams of Phar Lap Software
  168.        for pointing out the need for this check; this is a 
  169.        good example of the sort of consistency check one must
  170.        do when working with undocumented DOS.
  171.     */
  172.     env_mcb = MCB_FM_SEG(FP_SEG(e));
  173.     env_owner = ((MCB far *) MK_FP(env_mcb, 0))->owner;
  174.     return (env_owner == mcb->owner) ? e : (char far *) 0;
  175. }
  176.  
  177. char far *progname_fm_psp(unsigned psp)
  178. {
  179.     char far *e;
  180.     unsigned len;
  181.  
  182.     /* is there an environment? */
  183.     if (! (e = env(MK_FP(MCB_FM_SEG(psp), 0))))
  184.         return (char far *) 0;
  185.  
  186.     /* program name only available in DOS 3+ */
  187.     if (_osmajor >= 3)
  188.     {
  189.         /* skip past environment variables */
  190.         do e += (len = fstrlen(e)) + 1;
  191.         while (len);
  192.  
  193.         /* 
  194.             e now points to WORD containing number of strings following
  195.             environment; check for reasonable value: signed because
  196.             could be FFFFh; will normally be 1
  197.         */
  198.         if ((*((signed far *) e) >= 1) && (*((signed far *) e) < 10))
  199.         {
  200.             e += sizeof(signed);
  201.             if (isalpha(*e))
  202.                 return e; /* could make canonical with INT 21h AH=60h */
  203.         }
  204.     }
  205.     
  206.     return (char far *) 0;
  207. }
  208.  
  209. void display_progname(MCB far *mcb)
  210. {
  211.     char far *s;
  212.     if (IS_PSP(mcb))
  213.         if (s = progname_fm_psp((FP_SEG(mcb) + 1)))
  214.             printf("%Fs ", s);
  215. }
  216.  
  217. BOOL belongs(void far *vec, unsigned start, unsigned size)
  218. {
  219.     unsigned seg = FP_SEG(vec) + (FP_OFF(vec) >> 4); /* normalize */
  220.     return (seg >= start) && (seg <= (start + size));
  221. }
  222.  
  223. void display_cmdline(MCB far *mcb)
  224. {
  225.     /*
  226.         psp := MCB owner
  227.         cmdline_len := psp[80h]
  228.         cmdline := psp[81h]
  229.         print cmdline (display width := cmdline_len)
  230.     */
  231.     int len = *((BYTE far *) MK_FP(mcb->owner, 0x80));
  232.     char far *cmdline = MK_FP(mcb->owner, 0x81);
  233.     printf("%.*Fs ", len, cmdline);
  234. }
  235.  
  236. void display_vectors(MCB far *mcb)
  237. {
  238.     static void far **vec = (void far **) 0;
  239.     WORD vec_seg;
  240.     int i;
  241.     int did_one=0;
  242.     
  243.     if (! vec)
  244.     {
  245.         if (! (vec = calloc(256, sizeof(void far *))))
  246.             fail("insufficient memory");
  247.         for (i=0; i<256; i++)
  248.             vec[i] = GETVECT(i);
  249.     }
  250.     
  251.     for (i=0; i<256; i++)
  252.         if (vec[i] && belongs(vec[i], FP_SEG(mcb), mcb->size))
  253.         {
  254.             if (! did_one) { did_one++; printf("["); }
  255.             printf("%02X ", i);
  256.             vec[i] = 0;
  257.         }
  258.     if (did_one) printf("]");
  259. }
  260.  
  261. unsigned fstrlen(char far *s)
  262. {
  263. #if defined(_MSC_VER) && (_MSC_VER >= 600)
  264.     return _fstrlen(s);
  265. #else
  266.     unsigned len = 0;
  267.     while (*s++)
  268.         len++;
  269.     return len;
  270. #endif    
  271. }
  272.  
  273.