Subversion Repositories SvarDOS

/****************************************************************************

  TREE - Graphically displays the directory structure of a drive or path

  Written to work with FreeDOS (and other DOS variants)

  Win32(c) console and DOS with LFN support.

****************************************************************************/

#define VERSION "1.04"

/****************************************************************************

  Written by: Kenneth J. Davis

  Date:       August, 2000

  Updated:    September, 2000; October, 2000; November, 2000; January, 2001;

              May, 2004; Sept, 2005

  Contact:    jeremyd@computer.org

Copyright (c): Public Domain [United States Definition]

Permission is hereby granted, free of charge, to any person obtaining a copy

of this software and associated documentation files (the "Software"), to deal

in the Software without restriction, including without limitation the rights

to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

copies of the Software, and to permit persons to whom the Software is

furnished to do so, subject to the following conditions:

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,

EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES

OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

NONINFRINGEMENT. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR AUTHORS BE

LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER

IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT

OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER

DEALINGS IN THE SOFTWARE.

****************************************************************************/

/**

 * Define the appropriate target here or within your compiler

 */

/* #define WIN32 */    /** Win32 console version **/

/* #define DOS */      /** DOS version           **/

/* #define UNIX */

/* Include files */

#include <stdlib.h>

#include <stdio.h>

#include <string.h>

#include <direct.h>

#include <ctype.h>

#include <limits.h>

#include "stack.h"

/* Platform (OS) specific definitions */

#ifdef WIN32       /* windows specific     */

#ifndef _WIN32_WINNT         /* NT 4.0 or higher features */

#define _WIN32_WINNT 0x0400

#endif

#include <windows.h>

#include <winbase.h>

/* converts from Ansi to OEM (IBM extended )    */

#define charToDisplayChar(x) CharToOemA((x), (x))

/* For wide character conversion, charToDisplayChar

   must be a function that provides its own buffer

   to CharToOemW and copies the results back into x

   as CharToOemA is inplace safe, but CharToOemW is not.

*/

/* These are defined in w32fDOS.h for DOS and enable /

   disable use of DOS extended int21h API for LFNs,

   on Windows we use it to force using short name when

   both a long and short name are available

   (note, LFNs may still be shown when no SFN exists)

*/

#define LFN_ENABLE 1

#define LFN_DISABLE 0

int LFN_Enable_Flag = LFN_ENABLE;

/* DOS compiler treats L"" as a char * */

#define UDOT L"."

#define UDOTDOT L".."

/* Stream display is only supported for Win32, specifically Windows NT */

#include "streams.c"

#else                   /* DOS specific         */

/* Win32 File compability stuff */

#include "w32fDOS.h"

#include "wincon.h"

/* currently no mapping required */

#define charToDisplayChar(x)

/* DOS compiler treats L"" as a char * */

const WORD UDOT[]    = { 0x2E, 0x00 };        //   L"."

const WORD UDOTDOT[] = { 0x2E, 0x2E, 0x00 };  //   L".."

#endif

/* Define getdrive so it returns current drive, 0=A,1=B,...           */

#if defined _MSC_VER || defined __MSC /* MS Visual C/C++ 5 */

#define getdrive() (_getdrive() - 1)

#else /* #ifdef __BORLANDC__ || __TURBOC__ */

#define getdrive() getdisk()

#endif

#include <conio.h>  /* for getch()   */

/* End Platform (OS) specific sections */

/* The default extended forms of the lines used. */

#define VERTBAR_STR  "\xB3   "                 /* |    */

#define TBAR_HORZBAR_STR "\xC3\xC4\xC4\xC4"    /* +--- */

#define CBAR_HORZBAR_STR "\xC0\xC4\xC4\xC4"    /* \--- */

/* Unicode forms of the lines used. */

const char *UMARKER = "\xEF\xBB\xBF";   /* 0xFEFF, Indicate UTF-8 Unicode */

#define UVERTBAR_STR  "\xE2\x94\x82   "                                         /* |    */

#define UTBAR_HORZBAR_STR "\xE2\x94\x9C\xE2\x94\x80\xE2\x94\x80\xE2\x94\x80"    /* +--- */

#define UCBAR_HORZBAR_STR "\xE2\x94\x94\xE2\x94\x80\xE2\x94\x80\xE2\x94\x80"    /* \--- */

/*

const unichar UVERTBAR_STR[]      = { 0x2502, 0x20, 0x20, 0x20 };

const unichar UTBAR_HORZBAR_STR[] = { 0x251C, 0x2500, 0x2500, 0x2500 };

const unichar UCBAR_HORZBAR_STR[] = { 0x2514, 0x2500, 0x2500, 0x2500 };

*/

/* Global flags */

#define SHOWFILESON    1  /* Display names of files in directories       */

#define SHOWFILESOFF   0  /* Don't display names of files in directories */

#define UNICODECHARS   2  /* Use Unicode (UTF8) characters               */

#define ASCIICHARS     1  /* Use ASCII [7bit] characters                 */

#define EXTENDEDCHARS  0  /* Use extended ASCII [8bit] characters        */

#define NOPAUSE        0  /* Default, don't pause after screenfull       */

#define PAUSE          1  /* Wait for keypress after each page           */

/* Global variables */

short showFiles = SHOWFILESOFF;

short charSet = EXTENDEDCHARS;

short pause = NOPAUSE;

short dspAll = 0;  /* if nonzero includes HIDDEN & SYSTEM files in output */

short dspSize = 0; /* if nonzero displays filesizes                       */

short dspAttr = 0; /* if nonzero displays file attributes [DACESHRBP]     */

short dspSumDirs = 0; /* show count of subdirectories  (per dir and total)*/

short dspStreams = 0; /* if nonzero tries to display nondefault streams   */

/* maintains total count, for > 4billion dirs, use a __int64 */

unsigned long totalSubDirCnt = 0;

/* text window size, used to determine when to pause,

   Note: rows is total rows available - 2

   1 is for pause message and then one to prevent auto scroll up

*/

short cols=80, rows=23;   /* determined these on startup (when possible)  */

/* Global constants */

#define SERIALLEN 16      /* Defines max size of volume & serial number   */

#define VOLLEN 128

#define MAXBUF 1024       /* Must be larger than max file path length     */

char path[MAXBUF];        /* Path to begin search from, default=current   */

#define MAXPADLEN (MAXBUF*2) /* Must be large enough to hold the maximum padding */

/* (MAXBUF/2)*4 == (max path len / min 2chars dirs "?\") * 4chars per padding    */

/* The maximum size any line of text output can be, including room for '\0'*/

#define MAXLINE 160        /* Increased to fit two lines for translations  */

/* The hard coded strings used by the following show functions.            */

/* common to many functions [Set 1] */

char newLine[MAXLINE] = "\n";

/* showUsage [Set 2] - Each %c will be replaced with proper switch/option */

char treeDescription[MAXLINE] = "Graphically displays the directory structure of a drive or path.\n";

char treeUsage[MAXLINE] =       "TREE [drive:][path] [%c%c] [%c%c]\n";

char treeFOption[MAXLINE] =     "   %c%c   Display the names of the files in each directory.\n";

char treeAOption[MAXLINE] =     "   %c%c   Use ASCII instead of extended characters.\n";

/* showInvalidUsage [Set 3] */

char invalidOption[MAXLINE] = "Invalid switch - %s\n";  /* Must include the %s for option given. */

char useTreeHelp[MAXLINE] =   "Use TREE %c? for usage information.\n"; /* %c replaced with switch */

/* showVersionInfo [Set 4] */

/* also uses treeDescription */

char treeGoal[MAXLINE] =      "Written to work with FreeDOS\n";

char treePlatforms[MAXLINE] = "Win32(c) console and DOS with LFN support.\n";

char version[MAXLINE] =       "Version %s\n"; /* Must include the %s for version string. */

char writtenBy[MAXLINE] =     "Written by: Kenneth J. Davis\n";

char writtenDate[MAXLINE] =   "Date:       2000, 2001, 2004\n";

char contact[MAXLINE] =       "Contact:    jeremyd@computer.org\n";

char copyright[MAXLINE] =     "Copyright (c): Public Domain [United States Definition]\n";

/* showInvalidDrive [Set 5] */

char invalidDrive[MAXLINE] = "Invalid drive specification\n";

/* showInvalidPath [Set 6] */

char invalidPath[MAXLINE] = "Invalid path - %s\n"; /* Must include %s for the invalid path given. */

/* Misc Error messages [Set 7] */

/* showBufferOverrun */

/* %u required to show what the buffer's current size is. */

char bufferToSmall[MAXLINE] = "Error: File path specified exceeds maximum buffer = %u bytes\n";

/* showOutOfMemory */

/* %s required to display what directory we were processing when ran out of memory. */

char outOfMemory[MAXLINE] = "Out of memory on subdirectory: %s\n";

/* main [Set 1] */

char pathListingNoLabel[MAXLINE] = "Directory PATH listing\n";

char pathListingWithLabel[MAXLINE] = "Directory PATH listing for Volume %s\n"; /* %s for label */

char serialNumber[MAXLINE] = "Volume serial number is %s\n"; /* Must include %s for serial #   */

char noSubDirs[MAXLINE] = "No subdirectories exist\n\n";

char pauseMsg[MAXLINE]  = " --- Press any key to continue ---\n";

/* Option Processing - parseArguments [Set 8]      */

char optionchar1 = '/';  /* Primary character used to determine option follows  */

char optionchar2 = '-';  /* Secondary character used to determine option follows  */

const char OptShowFiles[2] = { 'F', 'f' };  /* Show files */

const char OptUseASCII[2]  = { 'A', 'a' };  /* Use ASCII only */

const char OptUnicode[2]   = { 'U', 'u' };  /* Use Unicode (16bit) output */

const char OptVersion[2]   = { 'V', 'v' };  /* Version information */

const char OptSFNs[2]      = { 'S', 's' };  /* Shortnames only (disable LFN support) */

const char OptPause[2]     = { 'P', 'p' };  /* Pause after each page (screenfull) */

const char OptDisplay[2]   = { 'D', 'd' };  /* modify Display settings */

const char OptSort[2]      = { 'O', 'o' };  /* sort Output */

/* Procedures */

/* Convert src from given codepage to UTF-16,

 * returns nonzero on success, 0 on any error

 * cp is the codepage of source string, should be either CP_ACP (ansi)

 * or CP_OEM (DOS, e.g. cp437).

 */

#define convertCPtoUTF16(cp, src, dst, dstsize) \

  MultiByteToWideChar(cp, 0/*MB_PRECOMPOSED|MB_USEGLYPHCHARS*/, src, -1, dst, dstsize)

/* Convert from UTF-16 to UTF-8 */

#if 0   // Can use on Win 98+, NT4+, or Win95 with MSLU

        // as Win95 does not support CP_UTF8 conversion

#define convertUTF16toUTF8(src, dst, dstsize) \

        WideCharToMultiByte(CP_UTF8, 0, src, -1, dst, dstsize, NULL, NULL)

#else

/* Transforms a UTF-16 string to a UTF-8 one, does not support surrogate pairs,

 * see http://en.wikipedia.org/wiki/UTF-8 for conversion chart

 * returns 0 on error, nonzero if conversion completed.

 */

int convertUTF16toUTF8(const WORD *src, char *dst, unsigned dstsize)

{

  if (!dstsize) return 0;

  dstsize--;  // reserve room for terminating '\0'

  const WORD *s = src;

  unsigned char *d = (unsigned char *)dst;

  for (register WORD ch = *s ; ch && dstsize; ch = *s)

  {

    // determine how many bytes this UTF-16 char is in UTF-8 format

    // Note: for values >= 0x10000 (MAXWORD+1)

    // a surrogate is needed and UTF-8 requires 4 bytes

    register unsigned short cnt = (ch < 0x80)? 1 : (ch < 0x800)? 2 : 3;

    // ensure enough room in dst buffer for it, break early if not

    if (dstsize < cnt) break;

    dstsize -= cnt;

    switch(cnt) // write out each byte needed

    {

      case 1:

      {

        *d = (unsigned char)ch;  // 0xxx xxxx == (ch & 0x7F)

        break;

      }

      case 2:

      { // 110x xxxx 10xx xxxx

        *d = (unsigned char)(0xC0 | ((ch >> 6) & 0x1F)); // top 5 bits

        *(d+1) = (unsigned char)(0x80 | (ch & 0x3F));        // lower 6 bits

        break;

      }

      case 3:

      { // 1110 xxxx 10xx xxxx 10xx xxxx

        *d = (unsigned char)(0xE0 | ((ch >> 12) & 0x0F)); // top 4 bits

        *(d+1) = (unsigned char)(0x80 | ((ch >> 6)  & 0x3F)); // mid 6 bits

        *(d+2) = (unsigned char)(0x80 | (ch & 0x3F));         // lower 6 bits

        break;

      }

    }

    // increment source and destination pointers

    s++;

    d+=cnt;

  }

  *d = '\0';  // force result string to be '\0' terminated

  if (*s && dstsize) // not all values converted

    return 0;

  else

    return 1;

}

#endif

/*

 * Converts src string (assumed windows ANSI or OEM cp) to UTF8 string.

 * src and dst may be same, but dst may be truncated if exceeds maxlen

 * cp is the codepage of source string, should be either CP_ACP (ansi)

 * or CP_OEM (DOS, e.g. cp437).

 * returns zero if src or dst is NULL or error in conversion

 * otherwise returns nonzero for success

 * WARNING: mapping may be incorrect under DOS as it ignores cp.

 */

BOOL charToUTF8(unsigned int cp, const char *src, char *dst, int maxlen)

{

  static char buffer[MAXBUF];

  if (src == NULL || dst == NULL || maxlen < 1) return 0;

  /* convert from ANSI/OEM cp to UTF-16 then to UTF-8 */

  WORD ubuf[MAXBUF];

  if (!convertCPtoUTF16(cp, src, ubuf, MAXBUF) ||

      !convertUTF16toUTF8(ubuf, buffer, MAXBUF))

    return 0;

  memcpy(dst, buffer, maxlen);

  return 1;

}

/* sets rows & cols to size of actual console window

 * force NOPAUSE if appears output redirected to a file or

 * piped to another program

 * Uses hard coded defaults and leaves pause flag unchanged

 * if unable to obtain information.

 */

void getConsoleSize(void)

{

  CONSOLE_SCREEN_BUFFER_INFO csbi;

  HANDLE h = GetStdHandle(STD_OUTPUT_HANDLE);

  if (h != INVALID_HANDLE_VALUE)

  {

    switch (GetFileType(h))

    {

      case FILE_TYPE_PIPE:  /* e.g. piped to more, tree | more */

      case FILE_TYPE_DISK:  /* e.g. redirected to a file, tree > filelist.txt */

      {

         /* Output to a file or program, so no screen to fill (no max cols or rows) */

         pause = NOPAUSE;   /* so ignore request to pause */

         break;

      }

      case FILE_TYPE_CHAR:  /* e.g. the console */

      case FILE_TYPE_UNKNOWN:  /* else at least attempt to get proper limits */

      /*case #define FILE_TYPE_REMOTE:*/

      default:

      {

        if (GetConsoleScreenBufferInfo(h, &csbi))

        {

          /* rows = window size - 2, where -2 neccessary to keep screen from scrolling */

          rows = csbi.srWindow.Bottom - csbi.srWindow.Top - 1;

          cols = csbi.srWindow.Right - csbi.srWindow.Left + 1;

        }

        /* else use hard coded defaults of 80x23 assuming 80x25 sized screen */

        break;

      }

    }

  }

}

/* when pause == NOPAUSE then identical to printf,

   otherwise counts lines printed and pauses as needed.

   Should be used for all messages printed that do not

   immediately exit afterwards (else printf may be used).

   May display N too many lines before pause if line is

   printed that exceeds cols [N=linelen%cols] and lacks

   any newlines (but this should not occur in tree).

*/

#include <stdarg.h>  /* va_list, va_start, va_end */

int pprintf(const char *msg, ...)

{

  static int lineCnt = rows;

  static int lineCol = 0;

  va_list argptr;

  int cnt;

  char buffer[MAXBUF];

  va_start(argptr, msg);

  cnt = vsprintf(buffer, msg, argptr);

  va_end(argptr);

  if (pause == PAUSE)

  {

    char * l = buffer;

    /* cycle through counting newlines and lines > cols */

    for (register char *t = strchr(l, '\n'); t != NULL; t = strchr(l, '\n'))

    {

      t++;             /* point to character after newline */

      char c = *t;     /* store current value */

      *t = '\0';       /* mark as end of string */

      /* print all but last line of a string that wraps across rows */

      /* adjusting for partial lines printed without the newlines   */

      while (strlen(l)+lineCol > cols)

      {

        char c = l[cols-lineCol];

        l[cols-lineCol] = '\0';

        printf("%s", l);

        l[cols-lineCol] = c;

        l += cols-lineCol;

        lineCnt--;  lineCol = 0;

        if (!lineCnt) { lineCnt= rows;  fflush(NULL);  fprintf(stderr, "%s", pauseMsg);  getch(); }

      }

      printf("%s", l); /* print out this line */

      *t = c;          /* restore value */

      l = t;           /* mark beginning of next line */

      lineCnt--;  lineCol = 0;

      if (!lineCnt) { lineCnt= rows;  fflush(NULL);  fprintf(stderr, "%s", pauseMsg);  getch(); }

    }

    printf("%s", l);   /* print rest of string that lacks newline */

    lineCol = strlen(l);

  }

  else  /* NOPAUSE */

    printf("%s", buffer);

  return cnt;

}

/* Displays to user valid options then exits program indicating no error */

void showUsage(void)

{

  printf("%s%s%s%s", treeDescription, newLine, treeUsage, newLine);

  printf("%s%s%s", treeFOption, treeAOption, newLine);

  exit(1);

}

/* Displays error message then exits indicating error */

void showInvalidUsage(char * badOption)

{

  printf(invalidOption, badOption);

  printf("%s%s", useTreeHelp, newLine);

  exit(1);

}

/* Displays author, copyright, etc info, then exits indicating no error. */

void showVersionInfo(void)

{

  printf("%s%s%s%s%s", treeDescription, newLine, treeGoal, treePlatforms, newLine);

  printf(version, VERSION);

  printf("%s%s%s%s%s", writtenBy, writtenDate, contact, newLine, newLine);

  printf("%s%s", copyright, newLine);

  exit(1);

}

/* Displays error messge for invalid drives and exits */

void showInvalidDrive(void)

{

  printf(invalidDrive);

  exit(1);

}

/* Takes a fullpath, splits into drive (C:, or \\server\share) and path */

void splitpath(char *fullpath, char *drive, char *path);

/**

 * Takes a given path, strips any \ or / that may appear on the end.

 * Returns a pointer to its static buffer containing path

 * without trailing slash and any necessary display conversions.

 */

char *fixPathForDisplay(char *path);

/* Displays error message for invalid path; Does NOT exit */

void showInvalidPath(char *path)

{

  char partialPath[MAXBUF], dummy[MAXBUF];

  pprintf("%s\n", path);

  splitpath(path, dummy, partialPath);

  pprintf(invalidPath, fixPathForDisplay(partialPath));

}

/* Displays error message for out of memory; Does NOT exit */

void showOutOfMemory(char *path)

{

  pprintf(outOfMemory, path);

}

/* Displays buffer exceeded message and exits */

void showBufferOverrun(WORD maxSize)

{

  printf(bufferToSmall, maxSize);

  exit(1);

}

/**

 * Takes a fullpath, splits into drive (C:, or \\server\share) and path

 * It assumes a colon as the 2nd character means drive specified,

 * a double slash \\ (\\, //, \/, or /\) specifies network share.

 * If neither drive nor network share, then assumes whole fullpath

 * is path, and sets drive to "".

 * If drive specified, then set drive to it and colon, eg "C:", with

 * the rest of fullpath being set in path.

 * If network share, the slash slash followed by the server name,

 * another slash and either the rest of fullpath or up to, but not

 * including, the next slash are placed in drive, eg "\\KJD\myshare";

 * the rest of the fullpath including the slash are placed in

 * path, eg "\mysubdir"; where fullpath is "\\KJD\myshare\mysubdir".

 * None of these may be NULL, and drive and path must be large

 * enough to hold fullpath.

 */

void splitpath(char *fullpath, char *drive, char *path)

{

  register char *src = fullpath;

  register char oldchar;

  /* If either network share or path only starting at root directory */

  if ( (*src == '\\') || (*src == '/') )

  {

    src++;

    if ( (*src == '\\') || (*src == '/') ) /* network share */

    {

      src++;

      /* skip past server name */

      while ( (*src != '\\') && (*src != '/') && (*src != '\0') )

        src++;

      /* skip past slash (\ or /) separating  server from share */

      if (*src != '\0') src++;

      /* skip past share name */

      while ( (*src != '\\') && (*src != '/') && (*src != '\0') )

        src++;

      /* src points to start of path, either a slash or '\0' */

      oldchar = *src;

      *src = '\0';

      /* copy server name to drive */

      strcpy(drive, fullpath);

      /* restore character used to mark end of server name */

      *src = oldchar;

      /* copy path */

      strcpy(path, src);

    }

    else /* path only starting at root directory */

    {

      /* no drive, so set path to same as fullpath */

      strcpy(drive, "");

      strcpy(path, fullpath);

    }

  }

  else

  {

    if (*src != '\0') src++;

    /* Either drive and path or path only */

    if (*src == ':')

    {

      /* copy drive specified */

      *drive = *fullpath;  drive++;

      *drive = ':';        drive++;

      *drive = '\0';

      /* copy path */

      src++;

      strcpy(path, src);

    }

    else

    {

      /* no drive, so set path to same as fullpath */

      strcpy(drive, "");

      strcpy(path, fullpath);

    }

  }

}

/* Converts given path to full path */

void getProperPath(char *fullpath)

{

  char drive[MAXBUF];

  char path[MAXBUF];

  splitpath(fullpath, drive, path);

  /* if no drive specified use current */

  if (drive[0] == '\0')

  {

    sprintf(fullpath, "%c:%s", 'A'+ getdrive(), path);

  }

  else if (path[0] == '\0') /* else if drive but no path specified */

  {

    if ((drive[0] == '\\') || (drive[0] == '/'))

    {

      /* if no path specified and network share, use root   */

      sprintf(fullpath, "%s%s", drive, "\\");

    }

    else

    {

      /* if no path specified and drive letter, use current path */

      sprintf(fullpath, "%s%s", drive, ".");

    }

  }

  /* else leave alone, it has both a drive and path specified */

}

/* Parses the command line and sets global variables. */

void parseArguments(int argc, char *argv[])

{

  register int i;     /* temp loop variable */

  /* if no drive specified on command line, use current */

  sprintf(path, "%c:.", 'A'+ getdrive());

  for (i = 1; i < argc; i++)

  {

    /* Check if user is giving an option or drive/path */

    if ((argv[i][0] == optionchar1) || (argv[i][0] == optionchar2) )

    {

      /* check multi character options 1st */

      if ((argv[i][1] == OptDisplay[0]) || (argv[i][1] == OptDisplay[1]))

      {

        switch (argv[i][2] & 0xDF)

        {

          case 'A' :       /*  /DA  display attributes */

            dspAttr = 1;

            break;

          case 'F' :       /*  /DF  display filesizes  */

            dspSize = 1;

            break;

          case 'H' :       /*  /DH  display hidden & system files (normally not shown) */

            dspAll = 1;

            break;

          case 'R' :       /*  /DR  display results at end */

            dspSumDirs = 1;

            break;

          case 'S' :       /*  /DS  display alternate file streams */

            dspStreams = 1;

            break;

          default:

            showInvalidUsage(argv[i]);

        }

      }

      else if ((argv[i][1] == OptSort[0]) || (argv[i][1] == OptSort[1]))

      {

#if 1  // not yet supported

        showInvalidUsage(argv[i]);

#else

        int reverse = (argv[i][3] == '-')?1:0;  /* invert sort if suffixed with - */

        switch (argv[i][2] & 0xDF)

        {

          case 'F' :       /*  /Of  sort by Filesize   */

            break;

          case 'N' :       /*  /On  sort by fileName   */

            break;

          case 'E' :       /*  /Oe  sort by Extension  */

            break;

          default:

            showInvalidUsage(argv[i]);

        }

#endif

      }

      else /* a 1 character option (or invalid) */

      {

        if (argv[i][2] != '\0')

          showInvalidUsage(argv[i]);

        /* Must check both uppercase and lowercase                        */

        if ((argv[i][1] == OptShowFiles[0]) || (argv[i][1] == OptShowFiles[1]))

          showFiles = SHOWFILESON; /* set file display flag appropriately */

        else if ((argv[i][1] == OptUseASCII[0]) || (argv[i][1] == OptUseASCII[1]))

          charSet = ASCIICHARS;    /* set charset flag appropriately      */

        else if (argv[i][1] == '?')

          showUsage();             /* show usage info and exit            */

        else if ((argv[i][1] == OptVersion[0]) || (argv[i][1] == OptVersion[1]))

          showVersionInfo();       /* show version info and exit          */

        else if ((argv[i][1] == OptSFNs[0]) || (argv[i][1] == OptSFNs[1]))

          LFN_Enable_Flag = LFN_DISABLE;         /* force shortnames only */

        else if ((argv[i][1] == OptPause[0]) || (argv[i][1] == OptPause[1]))

          pause = PAUSE;     /* wait for keypress after each page (pause) */

        else if ((argv[i][1] == OptUnicode[0]) || (argv[i][1] == OptUnicode[1]))

          charSet = UNICODECHARS;   /* indicate output Unicode text */

        else /* Invalid or unknown option */

          showInvalidUsage(argv[i]);

      }

    }

    else /* should be a drive/path */

    {

      if (strlen(argv[i]) > MAXBUF)

        showBufferOverrun(MAXBUF);

      /* copy path over, making all caps to look prettier, can be strcpy */

      register char *dptr = path;

      for (register char *cptr = argv[i]; *cptr != '\0'; cptr++, dptr++)

        *dptr = toupper(*cptr);

      *dptr = '\0';

      /* Converts given path to full path */

      getProperPath(path);

    }

  }

}

/**

 * Fills in the serial and volume variables with the serial #

 * and volume found using path.

 * If there is an error getting the volume & serial#, then an

 * error message is displayed and the program exits.

 * Volume and/or serial # returned may be blank if the path specified

 * does not contain them, or an error retrieving

 * (ie UNC paths under DOS), but path is valid.

 */

void GetVolumeAndSerial(char *volume, char *serial, char *path)

{

  char rootPath[MAXBUF];

  char dummy[MAXBUF];

  union serialNumber {

    DWORD serialFull;

    struct {

      WORD a;

      WORD b;

    } serialParts;

  } serialNum;

  /* get drive letter or share server\name */

  splitpath(path, rootPath, dummy);

  strcat(rootPath, "\\");

  if (GetVolumeInformation(rootPath, volume, VOLLEN,

      &serialNum.serialFull, NULL, NULL, NULL, 0) == 0)

	showInvalidDrive();

  if (serialNum.serialFull == 0)

    serial[0] = '\0';

  else

    sprintf(serial, "%04X:%04X",

      serialNum.serialParts.b, serialNum.serialParts.a);

}

/* FindFile stuff to support optional NT & Unicode API variants */

typedef union WIN32_FIND_DATA_BOTH

{

 WIN32_FIND_DATAW ud;

 WIN32_FIND_DATAA ad;

} WIN32_FIND_DATA_BOTH;

#ifndef STDCALL

#define STDCALL __stdcall

#endif

typedef HANDLE ( STDCALL * fFindFirstFileExA)(const char *, FINDEX_INFO_LEVELS, void *, FINDEX_SEARCH_OPS, void *, DWORD);

typedef HANDLE ( STDCALL * fFindFirstFileExW)(const WORD *, FINDEX_INFO_LEVELS, void *, FINDEX_SEARCH_OPS, void *, DWORD);

typedef BOOL ( STDCALL * fFindNextFileW)(HANDLE, WIN32_FIND_DATAW *);

/* FindFirstFileExA is only available on NT systems, so on Win9x & DOS use plain FindFirstFile */

HANDLE STDCALL myFindFirstFileExA(const char *fname, FINDEX_INFO_LEVELS, void * ffd, FINDEX_SEARCH_OPS, void *, DWORD)