misc/shell.c

   1 /*
   2  *                              Shell Library Functions
   3  *
   4  *  1998 Marcus Meissner
   5  */
   6 #include <assert.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <unistd.h>
  10 #include <ctype.h>
  11 #include "windows.h"
  12 #include "winerror.h"
  13 #include "file.h"
  14 #include "shell.h"
  15 #include "heap.h"
  16 #include "module.h"
  17 #include "neexe.h"
  18 #include "resource.h"
  19 #include "dlgs.h"
  20 #include "win.h"
  21 #include "graphics.h"
  22 #include "cursoricon.h"
  23 #include "interfaces.h"
  24 #include "sysmetrics.h"
  25 #include "shlobj.h"
  26 #include "debug.h"
  27 #include "winreg.h"
  28 #include "imagelist.h"
  29 #include "commctrl.h"
  30
  31 /* .ICO file ICONDIR definitions */
  32
  33 #pragma pack(1)
  34
  35 typedef struct
  36 {
  37     BYTE        bWidth;          /* Width, in pixels, of the image      */
  38     BYTE        bHeight;         /* Height, in pixels, of the image     */
  39     BYTE        bColorCount;     /* Number of colors in image (0 if >=8bpp) */
  40     BYTE        bReserved;       /* Reserved ( must be 0)               */
  41     WORD        wPlanes;         /* Color Planes                        */
  42     WORD        wBitCount;       /* Bits per pixel                      */
  43     DWORD       dwBytesInRes;    /* How many bytes in this resource?    */
  44     DWORD       dwImageOffset;   /* Where in the file is this image?    */
  45 } icoICONDIRENTRY, *LPicoICONDIRENTRY;
  46
  47 typedef struct
  48 {
  49     WORD            idReserved;   /* Reserved (must be 0)               */
  50     WORD            idType;       /* Resource Type (1 for icons)        */
  51     WORD            idCount;      /* How many images?                   */
  52     icoICONDIRENTRY idEntries[1]; /* An entry for each image (idCount of 'em) */
  53 } icoICONDIR, *LPicoICONDIR;
  54
  55 #pragma pack(4)
  56
  57 static const char*      lpstrMsgWndCreated = "OTHERWINDOWCREATED";
  58 static const char*      lpstrMsgWndDestroyed = "OTHERWINDOWDESTROYED";
  59 static const char*      lpstrMsgShellActivate = "ACTIVATESHELLWINDOW";
  60
  61 static HWND16   SHELL_hWnd = 0;
  62 static HHOOK    SHELL_hHook = 0;
  63 static UINT16   uMsgWndCreated = 0;
  64 static UINT16   uMsgWndDestroyed = 0;
  65 static UINT16   uMsgShellActivate = 0;
  66
  67 /*************************************************************************
  68  *                              DragAcceptFiles32               [SHELL32.54]
  69  */
  70 void WINAPI DragAcceptFiles32(HWND32 hWnd, BOOL32 b)
  71 {
  72   WND* wnd = WIN_FindWndPtr(hWnd);
  73
  74   if( wnd )
  75     wnd->dwExStyle = b? wnd->dwExStyle | WS_EX_ACCEPTFILES
  76                       : wnd->dwExStyle & ~WS_EX_ACCEPTFILES;
  77 }
  78
  79 /*************************************************************************
  80  *                              DragAcceptFiles16               [SHELL.9]
  81  */
  82 void WINAPI DragAcceptFiles16(HWND16 hWnd, BOOL16 b)
  83 {
  84   DragAcceptFiles32(hWnd, b);
  85 }
  86
  87 /*************************************************************************
  88  *                              SHELL_DragQueryFile     [internal]
  89  *
  90  */
  91 static UINT32 SHELL_DragQueryFile(LPSTR lpDrop, LPWSTR lpwDrop, UINT32 lFile,
  92                                   LPSTR lpszFile, LPWSTR lpszwFile, UINT32 lLength)
  93 {
  94     UINT32 i;
  95
  96     i = 0;
  97     if (lpDrop) {
  98     while (i++ < lFile) {
  99         while (*lpDrop++); /* skip filename */
 100         if (!*lpDrop)
 101           return (lFile == 0xFFFFFFFF) ? i : 0;
 102       }
 103     }
 104     if (lpwDrop) {
 105       while (i++ < lFile) {
 106         while (*lpwDrop++); /* skip filename */
 107         if (!*lpwDrop)
 108           return (lFile == 0xFFFFFFFF) ? i : 0;
 109       }
 110     }
 111
 112     if (lpDrop)  i = lstrlen32A(lpDrop);
 113     if (lpwDrop) i = lstrlen32W(lpwDrop);
 114     i++;
 115     if (!lpszFile && !lpszwFile) {
 116       return i;   /* needed buffer size */
 117     }
 118     i = (lLength > i) ? i : lLength;
 119     if (lpszFile) {
 120       if (lpDrop) lstrcpyn32A (lpszFile,  lpDrop,  i);
 121       else        lstrcpynWtoA(lpszFile,  lpwDrop, i);
 122     } else {
 123       if (lpDrop) lstrcpynAtoW(lpszwFile, lpDrop,  i);
 124       else        lstrcpyn32W (lpszwFile, lpwDrop, i);
 125     }
 126     return i;
 127 }
 128
 129 /*************************************************************************
 130  *                              DragQueryFile32A        [SHELL32.81] [shell32.82]
 131  */
 132 UINT32 WINAPI DragQueryFile32A(HDROP32 hDrop, UINT32 lFile, LPSTR lpszFile,
 133                               UINT32 lLength)
 134 { /* hDrop is a global memory block allocated with GMEM_SHARE
 135      * with DROPFILESTRUCT as a header and filenames following
 136      * it, zero length filename is in the end */
 137
 138     LPDROPFILESTRUCT32 lpDropFileStruct;
 139     LPSTR lpCurrent;
 140     UINT32 i;
 141
 142     TRACE(shell,"(%08x, %x, %p, %u)\n", hDrop,lFile,lpszFile,lLength);
 143
 144     lpDropFileStruct = (LPDROPFILESTRUCT32) GlobalLock32(hDrop);
 145     if(!lpDropFileStruct)
 146       return 0;
 147
 148     lpCurrent = (LPSTR) lpDropFileStruct + lpDropFileStruct->lSize;
 149     i = SHELL_DragQueryFile(lpCurrent, NULL, lFile, lpszFile, NULL, lLength);
 150     GlobalUnlock32(hDrop);
 151     return i;
 152 }
 153
 154 /*************************************************************************
 155  *                              DragQueryFile32W        [shell32.133]
 156  */
 157 UINT32 WINAPI DragQueryFile32W(HDROP32 hDrop, UINT32 lFile, LPWSTR lpszwFile,
 158                               UINT32 lLength)
 159 {
 160     LPDROPFILESTRUCT32 lpDropFileStruct;
 161     LPWSTR lpwCurrent;
 162     UINT32 i;
 163
 164     TRACE(shell,"(%08x, %x, %p, %u)\n", hDrop,lFile,lpszwFile,lLength);
 165
 166     lpDropFileStruct = (LPDROPFILESTRUCT32) GlobalLock32(hDrop);
 167     if(!lpDropFileStruct)
 168       return 0;
 169
 170     lpwCurrent = (LPWSTR) lpDropFileStruct + lpDropFileStruct->lSize;
 171     i = SHELL_DragQueryFile(NULL, lpwCurrent, lFile, NULL, lpszwFile,lLength);
 172     GlobalUnlock32(hDrop);
 173     return i;
 174 }
 175 /*************************************************************************
 176  *                              DragQueryFile16         [SHELL.11]
 177  */
 178 UINT16 WINAPI DragQueryFile16(HDROP16 hDrop, WORD wFile, LPSTR lpszFile,
 179                             WORD wLength)
 180 { /* hDrop is a global memory block allocated with GMEM_SHARE
 181      * with DROPFILESTRUCT as a header and filenames following
 182      * it, zero length filename is in the end */
 183
 184     LPDROPFILESTRUCT16 lpDropFileStruct;
 185     LPSTR lpCurrent;
 186     WORD  i;
 187
 188     TRACE(shell,"(%04x, %x, %p, %u)\n", hDrop,wFile,lpszFile,wLength);
 189
 190     lpDropFileStruct = (LPDROPFILESTRUCT16) GlobalLock16(hDrop);
 191     if(!lpDropFileStruct)
 192       return 0;
 193
 194     lpCurrent = (LPSTR) lpDropFileStruct + lpDropFileStruct->wSize;
 195
 196     i = (WORD)SHELL_DragQueryFile(lpCurrent, NULL, wFile==0xffff?0xffffffff:wFile,
 197                                   lpszFile, NULL, wLength);
 198     GlobalUnlock16(hDrop);
 199     return i;
 200 }
 201
 202
 203
 204 /*************************************************************************
 205  *                              DragFinish32            [SHELL32.80]
 206  */
 207 void WINAPI DragFinish32(HDROP32 h)
 208 { TRACE(shell,"\n");
 209     GlobalFree32((HGLOBAL32)h);
 210 }
 211
 212 /*************************************************************************
 213  *                              DragFinish16            [SHELL.12]
 214  */
 215 void WINAPI DragFinish16(HDROP16 h)
 216 { TRACE(shell,"\n");
 217     GlobalFree16((HGLOBAL16)h);
 218 }
 219
 220
 221 /*************************************************************************
 222  *                              DragQueryPoint32                [SHELL32.135]
 223  */
 224 BOOL32 WINAPI DragQueryPoint32(HDROP32 hDrop, POINT32 *p)
 225 {
 226   LPDROPFILESTRUCT32 lpDropFileStruct;
 227   BOOL32             bRet;
 228   TRACE(shell,"\n");
 229   lpDropFileStruct = (LPDROPFILESTRUCT32) GlobalLock32(hDrop);
 230
 231   memcpy(p,&lpDropFileStruct->ptMousePos,sizeof(POINT32));
 232   bRet = lpDropFileStruct->fInNonClientArea;
 233
 234   GlobalUnlock32(hDrop);
 235   return bRet;
 236 }
 237
 238 /*************************************************************************
 239  *                              DragQueryPoint16                [SHELL.13]
 240  */
 241 BOOL16 WINAPI DragQueryPoint16(HDROP16 hDrop, POINT16 *p)
 242 {
 243   LPDROPFILESTRUCT16 lpDropFileStruct;
 244   BOOL16           bRet;
 245   TRACE(shell,"\n");
 246   lpDropFileStruct = (LPDROPFILESTRUCT16) GlobalLock16(hDrop);
 247
 248   memcpy(p,&lpDropFileStruct->ptMousePos,sizeof(POINT16));
 249   bRet = lpDropFileStruct->fInNonClientArea;
 250
 251   GlobalUnlock16(hDrop);
 252   return bRet;
 253 }
 254
 255 /*************************************************************************
 256  *      SHELL_FindExecutable [Internal]
 257  *
 258  * Utility for code sharing between FindExecutable and ShellExecute
 259  */
 260 HINSTANCE32 SHELL_FindExecutable( LPCSTR lpFile,
 261                                          LPCSTR lpOperation,
 262                                          LPSTR lpResult)
 263 { char *extension = NULL; /* pointer to file extension */
 264     char tmpext[5];         /* local copy to mung as we please */
 265     char filetype[256];     /* registry name for this filetype */
 266     LONG filetypelen=256;   /* length of above */
 267     char command[256];      /* command from registry */
 268     LONG commandlen=256;    /* This is the most DOS can handle :) */
 269     char buffer[256];       /* Used to GetProfileString */
 270     HINSTANCE32 retval=31;  /* default - 'No association was found' */
 271     char *tok;              /* token pointer */
 272     int i;                  /* random counter */
 273     char xlpFile[256];      /* result of SearchPath */
 274
 275   TRACE(shell, "%s\n", (lpFile != NULL?lpFile:"-") );
 276
 277     lpResult[0]='\0'; /* Start off with an empty return string */
 278
 279     /* trap NULL parameters on entry */
 280     if (( lpFile == NULL ) || ( lpResult == NULL ) || ( lpOperation == NULL ))
 281   { WARN(exec, "(lpFile=%s,lpResult=%s,lpOperation=%s): NULL parameter\n",
 282            lpFile, lpOperation, lpResult);
 283         return 2; /* File not found. Close enough, I guess. */
 284     }
 285
 286     if (SearchPath32A( NULL, lpFile,".exe",sizeof(xlpFile),xlpFile,NULL))
 287   { TRACE(shell, "SearchPath32A returned non-zero\n");
 288         lpFile = xlpFile;
 289     }
 290
 291     /* First thing we need is the file's extension */
 292     extension = strrchr( xlpFile, '.' ); /* Assume last "." is the one; */
 293                                         /* File->Run in progman uses */
 294                                         /* .\FILE.EXE :( */
 295   TRACE(shell, "xlpFile=%s,extension=%s\n", xlpFile, extension);
 296
 297     if ((extension == NULL) || (extension == &xlpFile[strlen(xlpFile)]))
 298   { WARN(shell, "Returning 31 - No association\n");
 299         return 31; /* no association */
 300     }
 301
 302     /* Make local copy & lowercase it for reg & 'programs=' lookup */
 303     lstrcpyn32A( tmpext, extension, 5 );
 304     CharLower32A( tmpext );
 305   TRACE(shell, "%s file\n", tmpext);
 306
 307     /* Three places to check: */
 308     /* 1. win.ini, [windows], programs (NB no leading '.') */
 309     /* 2. Registry, HKEY_CLASS_ROOT\<filetype>\shell\open\command */
 310     /* 3. win.ini, [extensions], extension (NB no leading '.' */
 311     /* All I know of the order is that registry is checked before */
 312     /* extensions; however, it'd make sense to check the programs */
 313     /* section first, so that's what happens here. */
 314
 315     /* See if it's a program - if GetProfileString fails, we skip this
 316      * section. Actually, if GetProfileString fails, we've probably
 317      * got a lot more to worry about than running a program... */
 318     if ( GetProfileString32A("windows", "programs", "exe pif bat com",
 319                                                   buffer, sizeof(buffer)) > 0 )
 320   { for (i=0;i<strlen(buffer); i++) buffer[i]=tolower(buffer[i]);
 321
 322                 tok = strtok(buffer, " \t"); /* ? */
 323                 while( tok!= NULL)
 324                   {
 325                         if (strcmp(tok, &tmpext[1])==0) /* have to skip the leading "." */
 326                           {
 327                                 strcpy(lpResult, xlpFile);
 328                                 /* Need to perhaps check that the file has a path
 329                                  * attached */
 330         TRACE(shell, "found %s\n", lpResult);
 331                                 return 33;
 332
 333                 /* Greater than 32 to indicate success FIXME According to the
 334                  * docs, I should be returning a handle for the
 335                  * executable. Does this mean I'm supposed to open the
 336                  * executable file or something? More RTFM, I guess... */
 337                           }
 338                         tok=strtok(NULL, " \t");
 339                   }
 340           }
 341
 342     /* Check registry */
 343     if (RegQueryValue16( HKEY_CLASSES_ROOT, tmpext, filetype,
 344                          &filetypelen ) == ERROR_SUCCESS )
 345     {
 346         filetype[filetypelen]='\0';
 347   TRACE(shell, "File type: %s\n", filetype);
 348
 349         /* Looking for ...buffer\shell\lpOperation\command */
 350         strcat( filetype, "\\shell\\" );
 351         strcat( filetype, lpOperation );
 352         strcat( filetype, "\\command" );
 353
 354         if (RegQueryValue16( HKEY_CLASSES_ROOT, filetype, command,
 355                              &commandlen ) == ERROR_SUCCESS )
 356         {
 357             /* Is there a replace() function anywhere? */
 358             command[commandlen]='\0';
 359             strcpy( lpResult, command );
 360             tok=strstr( lpResult, "%1" );
 361             if (tok != NULL)
 362             {
 363                 tok[0]='\0'; /* truncate string at the percent */
 364                 strcat( lpResult, xlpFile ); /* what if no dir in xlpFile? */
 365                 tok=strstr( command, "%1" );
 366                 if ((tok!=NULL) && (strlen(tok)>2))
 367                 {
 368                     strcat( lpResult, &tok[2] );
 369                 }
 370             }
 371             retval=33; /* FIXME see above */
 372         }
 373     }
 374     else /* Check win.ini */
 375     {
 376         /* Toss the leading dot */
 377         extension++;
 378         if ( GetProfileString32A( "extensions", extension, "", command,
 379                                   sizeof(command)) > 0)
 380           {
 381                 if (strlen(command)!=0)
 382                   {
 383                         strcpy( lpResult, command );
 384                         tok=strstr( lpResult, "^" ); /* should be ^.extension? */
 385                         if (tok != NULL)
 386                           {
 387                                 tok[0]='\0';
 388                                 strcat( lpResult, xlpFile ); /* what if no dir in xlpFile? */
 389                                 tok=strstr( command, "^" ); /* see above */
 390                                 if ((tok != NULL) && (strlen(tok)>5))
 391                                   {
 392                                         strcat( lpResult, &tok[5]);
 393                                   }
 394                           }
 395                         retval=33; /* FIXME - see above */
 396                   }
 397           }
 398         }
 399
 400     TRACE(shell, "returning %s\n", lpResult);
 401     return retval;
 402 }
 403
 404 /*************************************************************************
 405  *                              ShellExecute16          [SHELL.20]
 406  */
 407 HINSTANCE16 WINAPI ShellExecute16( HWND16 hWnd, LPCSTR lpOperation,
 408                                    LPCSTR lpFile, LPCSTR lpParameters,
 409                                    LPCSTR lpDirectory, INT16 iShowCmd )
 410 {   HINSTANCE16 retval=31;
 411     char old_dir[1024];
 412     char cmd[256];
 413
 414     TRACE(shell, "(%04x,'%s','%s','%s','%s',%x)\n",
 415                 hWnd, lpOperation ? lpOperation:"<null>", lpFile ? lpFile:"<null>",
 416                 lpParameters ? lpParameters : "<null>",
 417                 lpDirectory ? lpDirectory : "<null>", iShowCmd);
 418
 419     if (lpFile==NULL) return 0; /* should not happen */
 420     if (lpOperation==NULL) /* default is open */
 421       lpOperation="open";
 422
 423     if (lpDirectory)
 424     { GetCurrentDirectory32A( sizeof(old_dir), old_dir );
 425         SetCurrentDirectory32A( lpDirectory );
 426     }
 427
 428     retval = SHELL_FindExecutable( lpFile, lpOperation, cmd );
 429
 430     if (retval > 32)  /* Found */
 431     { if (lpParameters)
 432       { strcat(cmd," ");
 433             strcat(cmd,lpParameters);
 434         }
 435
 436       TRACE(shell,"starting %s\n",cmd);
 437         retval = WinExec32( cmd, iShowCmd );
 438     }
 439     if (lpDirectory)
 440       SetCurrentDirectory32A( old_dir );
 441     return retval;
 442 }
 443
 444 /*************************************************************************
 445  *             FindExecutable16   (SHELL.21)
 446  */
 447 HINSTANCE16 WINAPI FindExecutable16( LPCSTR lpFile, LPCSTR lpDirectory,
 448                                      LPSTR lpResult )
 449 { return (HINSTANCE16)FindExecutable32A( lpFile, lpDirectory, lpResult );
 450 }
 451
 452
 453 /*************************************************************************
 454  *             AboutDlgProc16   (SHELL.33)
 455  */
 456 LRESULT WINAPI AboutDlgProc16( HWND16 hWnd, UINT16 msg, WPARAM16 wParam,
 457                                LPARAM lParam )
 458 { return AboutDlgProc32( hWnd, msg, wParam, lParam );
 459 }
 460
 461
 462 /*************************************************************************
 463  *             ShellAbout16   (SHELL.22)
 464  */
 465 BOOL16 WINAPI ShellAbout16( HWND16 hWnd, LPCSTR szApp, LPCSTR szOtherStuff,
 466                             HICON16 hIcon )
 467 { return ShellAbout32A( hWnd, szApp, szOtherStuff, hIcon );
 468 }
 469
 470 /*************************************************************************
 471  *                              SHELL_GetResourceTable
 472  */
 473 static DWORD SHELL_GetResourceTable(HFILE32 hFile,LPBYTE *retptr)
 474 {
 475   IMAGE_DOS_HEADER      mz_header;
 476   char                  magic[4];
 477   int                   size;
 478   TRACE(shell,"\n");
 479   *retptr = NULL;
 480   _llseek32( hFile, 0, SEEK_SET );
 481   if (  (_lread32(hFile,&mz_header,sizeof(mz_header)) != sizeof(mz_header)) ||
 482         (mz_header.e_magic != IMAGE_DOS_SIGNATURE)
 483   ) { /* .ICO file ? */
 484         if (mz_header.e_cblp == 1) { /* ICONHEADER.idType, must be 1 */
 485             *retptr = (LPBYTE)-1;
 486             return 1;
 487         }
 488         else
 489             return 0; /* failed */
 490   }
 491   _llseek32( hFile, mz_header.e_lfanew, SEEK_SET );
 492   if (_lread32( hFile, magic, sizeof(magic) ) != sizeof(magic))
 493         return 0;
 494   _llseek32( hFile, mz_header.e_lfanew, SEEK_SET);
 495
 496   if (*(DWORD*)magic  == IMAGE_NT_SIGNATURE)
 497         return IMAGE_NT_SIGNATURE;
 498   if (*(WORD*)magic == IMAGE_OS2_SIGNATURE) {
 499         IMAGE_OS2_HEADER        ne_header;
 500         LPBYTE                  pTypeInfo = (LPBYTE)-1;
 501
 502         if (_lread32(hFile,&ne_header,sizeof(ne_header))!=sizeof(ne_header))
 503                 return 0;
 504
 505         if (ne_header.ne_magic != IMAGE_OS2_SIGNATURE) return 0;
 506         size = ne_header.rname_tab_offset - ne_header.resource_tab_offset;
 507         if( size > sizeof(NE_TYPEINFO) )
 508         {
 509             pTypeInfo = (BYTE*)HeapAlloc( GetProcessHeap(), 0, size);
 510             if( pTypeInfo ) {
 511                 _llseek32(hFile, mz_header.e_lfanew+ne_header.resource_tab_offset, SEEK_SET);
 512                 if( _lread32( hFile, (char*)pTypeInfo, size) != size ) {
 513                     HeapFree( GetProcessHeap(), 0, pTypeInfo);
 514                     pTypeInfo = NULL;
 515                 }
 516             }
 517         }
 518         *retptr = pTypeInfo;
 519         return IMAGE_OS2_SIGNATURE;
 520   } else
 521         return 0; /* failed */
 522 }
 523
 524 /*************************************************************************
 525  *                      SHELL_LoadResource
 526  */
 527 static HGLOBAL16 SHELL_LoadResource(HINSTANCE16 hInst, HFILE32 hFile, NE_NAMEINFO* pNInfo, WORD sizeShift)
 528 { BYTE*  ptr;
 529  HGLOBAL16 handle = DirectResAlloc( hInst, 0x10, (DWORD)pNInfo->length << sizeShift);
 530   TRACE(shell,"\n");
 531  if( (ptr = (BYTE*)GlobalLock16( handle )) )
 532   { _llseek32( hFile, (DWORD)pNInfo->offset << sizeShift, SEEK_SET);
 533      _lread32( hFile, (char*)ptr, pNInfo->length << sizeShift);
 534      return handle;
 535    }
 536  return 0;
 537 }
 538
 539 /*************************************************************************
 540  *                      ICO_LoadIcon
 541  */
 542 static HGLOBAL16 ICO_LoadIcon(HINSTANCE16 hInst, HFILE32 hFile, LPicoICONDIRENTRY lpiIDE)
 543 { BYTE*  ptr;
 544  HGLOBAL16 handle = DirectResAlloc( hInst, 0x10, lpiIDE->dwBytesInRes);
 545   TRACE(shell,"\n");
 546  if( (ptr = (BYTE*)GlobalLock16( handle )) )
 547   { _llseek32( hFile, lpiIDE->dwImageOffset, SEEK_SET);
 548      _lread32( hFile, (char*)ptr, lpiIDE->dwBytesInRes);
 549      return handle;
 550    }
 551  return 0;
 552 }
 553
 554 /*************************************************************************
 555  *                      ICO_GetIconDirectory
 556  *
 557  *  Read .ico file and build phony ICONDIR struct for GetIconID
 558  */
 559 static HGLOBAL16 ICO_GetIconDirectory(HINSTANCE16 hInst, HFILE32 hFile, LPicoICONDIR* lplpiID )
 560 { WORD    id[3];  /* idReserved, idType, idCount */
 561   LPicoICONDIR  lpiID;
 562   int           i;
 563
 564   TRACE(shell,"\n");
 565   _llseek32( hFile, 0, SEEK_SET );
 566   if( _lread32(hFile,(char*)id,sizeof(id)) != sizeof(id) ) return 0;
 567
 568   /* check .ICO header
 569    *
 570    * - see http://www.microsoft.com/win32dev/ui/icons.htm
 571    */
 572
 573   if( id[0] || id[1] != 1 || !id[2] ) return 0;
 574
 575   i = id[2]*sizeof(icoICONDIRENTRY) + sizeof(id);
 576
 577   lpiID = (LPicoICONDIR)HeapAlloc( GetProcessHeap(), 0, i);
 578
 579   if( _lread32(hFile,(char*)lpiID->idEntries,i) == i )
 580   { HGLOBAL16 handle = DirectResAlloc( hInst, 0x10,
 581                                      id[2]*sizeof(ICONDIRENTRY) + sizeof(id) );
 582      if( handle )
 583     { CURSORICONDIR*     lpID = (CURSORICONDIR*)GlobalLock16( handle );
 584        lpID->idReserved = lpiID->idReserved = id[0];
 585        lpID->idType = lpiID->idType = id[1];
 586        lpID->idCount = lpiID->idCount = id[2];
 587        for( i=0; i < lpiID->idCount; i++ )
 588       { memcpy((void*)(lpID->idEntries + i),
 589                    (void*)(lpiID->idEntries + i), sizeof(ICONDIRENTRY) - 2);
 590             lpID->idEntries[i].icon.wResId = i;
 591          }
 592       *lplpiID = lpiID;
 593        return handle;
 594      }
 595   }
 596   /* fail */
 597
 598   HeapFree( GetProcessHeap(), 0, lpiID);
 599   return 0;
 600 }
 601
 602 /*************************************************************************
 603  *                      InternalExtractIcon             [SHELL.39]
 604  *
 605  * This abortion is called directly by Progman
 606  */
 607 HGLOBAL16 WINAPI InternalExtractIcon(HINSTANCE16 hInstance,
 608                                      LPCSTR lpszExeFileName, UINT16 nIconIndex,
 609                                      WORD n )
 610 {
 611   HGLOBAL16     hRet = 0;
 612   HGLOBAL16*    RetPtr = NULL;
 613   LPBYTE        pData;
 614   OFSTRUCT      ofs;
 615   DWORD         sig;
 616   HFILE32       hFile = OpenFile32( lpszExeFileName, &ofs, OF_READ );
 617   UINT16        iconDirCount = 0,iconCount = 0;
 618
 619   TRACE(shell,"(%04x,file %s,start %d,extract %d\n",
 620                        hInstance, lpszExeFileName, nIconIndex, n);
 621
 622   if( hFile == HFILE_ERROR32 || !n )
 623     return 0;
 624
 625   hRet = GlobalAlloc16( GMEM_FIXED | GMEM_ZEROINIT, sizeof(HICON16)*n);
 626   RetPtr = (HICON16*)GlobalLock16(hRet);
 627
 628   *RetPtr = (n == 0xFFFF)? 0: 1;        /* error return values */
 629
 630   sig = SHELL_GetResourceTable(hFile,&pData);
 631
 632   if((sig == IMAGE_OS2_SIGNATURE)
 633   || (sig == 1)) /* .ICO file */
 634   {
 635     HICON16      hIcon = 0;
 636     NE_TYPEINFO* pTInfo = (NE_TYPEINFO*)(pData + 2);
 637     NE_NAMEINFO* pIconStorage = NULL;
 638     NE_NAMEINFO* pIconDir = NULL;
 639     LPicoICONDIR lpiID = NULL;
 640
 641     if( pData == (BYTE*)-1 )
 642     {
 643         /* check for .ICO file */
 644
 645         hIcon = ICO_GetIconDirectory(hInstance, hFile, &lpiID);
 646         if( hIcon ) { iconDirCount = 1; iconCount = lpiID->idCount; }
 647     }
 648     else while( pTInfo->type_id && !(pIconStorage && pIconDir) )
 649     {
 650         /* find icon directory and icon repository */
 651
 652         if( pTInfo->type_id == NE_RSCTYPE_GROUP_ICON )
 653           {
 654              iconDirCount = pTInfo->count;
 655              pIconDir = ((NE_NAMEINFO*)(pTInfo + 1));
 656        TRACE(shell,"\tfound directory - %i icon families\n", iconDirCount);
 657           }
 658         if( pTInfo->type_id == NE_RSCTYPE_ICON )
 659           {
 660              iconCount = pTInfo->count;
 661              pIconStorage = ((NE_NAMEINFO*)(pTInfo + 1));
 662        TRACE(shell,"\ttotal icons - %i\n", iconCount);
 663           }
 664         pTInfo = (NE_TYPEINFO *)((char*)(pTInfo+1)+pTInfo->count*sizeof(NE_NAMEINFO));
 665     }
 666
 667     /* load resources and create icons */
 668
 669     if( (pIconStorage && pIconDir) || lpiID )
 670       if( nIconIndex == (UINT16)-1 ) RetPtr[0] = iconDirCount;
 671       else if( nIconIndex < iconDirCount )
 672       {
 673           UINT16   i, icon;
 674
 675           if( n > iconDirCount - nIconIndex ) n = iconDirCount - nIconIndex;
 676
 677           for( i = nIconIndex; i < nIconIndex + n; i++ )
 678           {
 679               /* .ICO files have only one icon directory */
 680
 681               if( lpiID == NULL )
 682                    hIcon = SHELL_LoadResource( hInstance, hFile, pIconDir + i,
 683                                                               *(WORD*)pData );
 684               RetPtr[i-nIconIndex] = GetIconID( hIcon, 3 );
 685               GlobalFree16(hIcon);
 686           }
 687
 688           for( icon = nIconIndex; icon < nIconIndex + n; icon++ )
 689           {
 690               hIcon = 0;
 691               if( lpiID )
 692                    hIcon = ICO_LoadIcon( hInstance, hFile,
 693                                          lpiID->idEntries + RetPtr[icon-nIconIndex]);
 694               else
 695                  for( i = 0; i < iconCount; i++ )
 696                    if( pIconStorage[i].id == (RetPtr[icon-nIconIndex] | 0x8000) )
 697                      hIcon = SHELL_LoadResource( hInstance, hFile, pIconStorage + i,
 698                                                                     *(WORD*)pData );
 699               if( hIcon )
 700               {
 701                   RetPtr[icon-nIconIndex] = LoadIconHandler( hIcon, TRUE );
 702                   FarSetOwner( RetPtr[icon-nIconIndex], GetExePtr(hInstance) );
 703               }
 704               else
 705                   RetPtr[icon-nIconIndex] = 0;
 706           }
 707       }
 708     if( lpiID ) HeapFree( GetProcessHeap(), 0, lpiID);
 709     else HeapFree( GetProcessHeap(), 0, pData);
 710   }
 711   if( sig == IMAGE_NT_SIGNATURE)
 712   {
 713         LPBYTE                  peimage,idata,igdata;
 714         PIMAGE_DOS_HEADER       dheader;
 715         PIMAGE_NT_HEADERS       pe_header;
 716         PIMAGE_SECTION_HEADER   pe_sections;
 717         PIMAGE_RESOURCE_DIRECTORY       rootresdir,iconresdir,icongroupresdir;
 718         PIMAGE_RESOURCE_DATA_ENTRY      idataent,igdataent;
 719         HANDLE32                fmapping;
 720         int                     i,j;
 721         PIMAGE_RESOURCE_DIRECTORY_ENTRY xresent;
 722         CURSORICONDIR           **cids;
 723
 724         fmapping = CreateFileMapping32A(hFile,NULL,PAGE_READONLY|SEC_COMMIT,0,0,NULL);
 725         if (fmapping == 0) { /* FIXME, INVALID_HANDLE_VALUE? */
 726     WARN(shell,"failed to create filemap.\n");
 727                 _lclose32( hFile);
 728                 return 0;
 729         }
 730         peimage = MapViewOfFile(fmapping,FILE_MAP_READ,0,0,0);
 731         if (!peimage) {
 732     WARN(shell,"failed to mmap filemap.\n");
 733                 CloseHandle(fmapping);
 734                 _lclose32( hFile);
 735                 return 0;
 736         }
 737         dheader = (PIMAGE_DOS_HEADER)peimage;
 738         /* it is a pe header, SHELL_GetResourceTable checked that */
 739         pe_header = (PIMAGE_NT_HEADERS)(peimage+dheader->e_lfanew);
 740         /* probably makes problems with short PE headers... but I haven't seen
 741          * one yet...
 742          */
 743         pe_sections = (PIMAGE_SECTION_HEADER)(((char*)pe_header)+sizeof(*pe_header));
 744         rootresdir = NULL;
 745         for (i=0;i<pe_header->FileHeader.NumberOfSections;i++) {
 746                 if (pe_sections[i].Characteristics & IMAGE_SCN_CNT_UNINITIALIZED_DATA)
 747                         continue;
 748                 /* FIXME: doesn't work when the resources are not in a seperate section */
 749                 if (pe_sections[i].VirtualAddress == pe_header->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress) {
 750                         rootresdir = (PIMAGE_RESOURCE_DIRECTORY)((char*)peimage+pe_sections[i].PointerToRawData);
 751                         break;
 752                 }
 753         }
 754
 755         if (!rootresdir) {
 756     WARN(shell,"haven't found section for resource directory.\n");
 757                 UnmapViewOfFile(peimage);
 758                 CloseHandle(fmapping);
 759                 _lclose32( hFile);
 760                 return 0;
 761         }
 762         icongroupresdir = GetResDirEntryW(rootresdir,RT_GROUP_ICON32W,
 763                                           (DWORD)rootresdir,FALSE);
 764         if (!icongroupresdir) {
 765     WARN(shell,"No Icongroupresourcedirectory!\n");
 766                 UnmapViewOfFile(peimage);
 767                 CloseHandle(fmapping);
 768                 _lclose32( hFile);
 769                 return 0;
 770         }
 771
 772         iconDirCount = icongroupresdir->NumberOfNamedEntries+icongroupresdir->NumberOfIdEntries;
 773         if( nIconIndex == (UINT16)-1 ) {
 774                 RetPtr[0] = iconDirCount;
 775                 UnmapViewOfFile(peimage);
 776                 CloseHandle(fmapping);
 777                 _lclose32( hFile);
 778                 return hRet;
 779         }
 780
 781         if (nIconIndex >= iconDirCount) {
 782     WARN(shell,"nIconIndex %d is larger than iconDirCount %d\n",
 783                             nIconIndex,iconDirCount);
 784                 UnmapViewOfFile(peimage);
 785                 CloseHandle(fmapping);
 786                 _lclose32( hFile);
 787                 GlobalFree16(hRet);
 788                 return 0;
 789         }
 790         cids = (CURSORICONDIR**)HeapAlloc(GetProcessHeap(),0,n*sizeof(CURSORICONDIR*));
 791
 792         /* caller just wanted the number of entries */
 793
 794         xresent = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)(icongroupresdir+1);
 795         /* assure we don't get too much ... */
 796         if( n > iconDirCount - nIconIndex ) n = iconDirCount - nIconIndex;
 797
 798         /* starting from specified index ... */
 799         xresent = xresent+nIconIndex;
 800
 801         for (i=0;i<n;i++,xresent++) {
 802                 CURSORICONDIR   *cid;
 803                 PIMAGE_RESOURCE_DIRECTORY       resdir;
 804
 805                 /* go down this resource entry, name */
 806                 resdir = (PIMAGE_RESOURCE_DIRECTORY)((DWORD)rootresdir+(xresent->u2.s.OffsetToDirectory));
 807                 /* default language (0) */
 808                 resdir = GetResDirEntryW(resdir,(LPWSTR)0,(DWORD)rootresdir,TRUE);
 809                 igdataent = (PIMAGE_RESOURCE_DATA_ENTRY)resdir;
 810
 811                 /* lookup address in mapped image for virtual address */
 812                 igdata = NULL;
 813                 for (j=0;j<pe_header->FileHeader.NumberOfSections;j++) {
 814                         if (igdataent->OffsetToData < pe_sections[j].VirtualAddress)
 815                                 continue;
 816                         if (igdataent->OffsetToData+igdataent->Size > pe_sections[j].VirtualAddress+pe_sections[j].SizeOfRawData)
 817                                 continue;
 818                         igdata = peimage+(igdataent->OffsetToData-pe_sections[j].VirtualAddress+pe_sections[j].PointerToRawData);
 819                 }
 820                 if (!igdata) {
 821       WARN(shell,"no matching real address for icongroup!\n");
 822                         UnmapViewOfFile(peimage);
 823                         CloseHandle(fmapping);
 824                         _lclose32( hFile);
 825                         return 0;
 826                 }
 827                 /* found */
 828                 cid = (CURSORICONDIR*)igdata;
 829                 cids[i] = cid;
 830                 RetPtr[i] = LookupIconIdFromDirectoryEx32(igdata,TRUE,SYSMETRICS_CXICON,SYSMETRICS_CYICON,0);
 831         }
 832         iconresdir=GetResDirEntryW(rootresdir,RT_ICON32W,
 833                                    (DWORD)rootresdir,FALSE);
 834         if (!iconresdir) {
 835       WARN(shell,"No Iconresourcedirectory!\n");
 836             UnmapViewOfFile(peimage);
 837             CloseHandle(fmapping);
 838             _lclose32( hFile);
 839             return 0;
 840         }
 841         for (i=0;i<n;i++) {
 842             PIMAGE_RESOURCE_DIRECTORY   xresdir;
 843
 844             xresdir = GetResDirEntryW(iconresdir,(LPWSTR)RetPtr[i],(DWORD)rootresdir,FALSE);
 845             xresdir = GetResDirEntryW(xresdir,(LPWSTR)0,(DWORD)rootresdir,TRUE);
 846
 847             idataent = (PIMAGE_RESOURCE_DATA_ENTRY)xresdir;
 848
 849             idata = NULL;
 850             /* map virtual to address in image */
 851             for (j=0;j<pe_header->FileHeader.NumberOfSections;j++) {
 852                 if (idataent->OffsetToData < pe_sections[j].VirtualAddress)
 853                     continue;
 854                 if (idataent->OffsetToData+idataent->Size > pe_sections[j].VirtualAddress+pe_sections[j].SizeOfRawData)
 855                     continue;
 856                 idata = peimage+(idataent->OffsetToData-pe_sections[j].VirtualAddress+pe_sections[j].PointerToRawData);
 857             }
 858             if (!idata) {
 859     WARN(shell,"no matching real address found for icondata!\n");
 860                 RetPtr[i]=0;
 861                 continue;
 862             }
 863             RetPtr[i] = CreateIconFromResourceEx32(idata,idataent->Size,TRUE,0x00030000,SYSMETRICS_CXICON,SYSMETRICS_CYICON,0);
 864         }
 865         UnmapViewOfFile(peimage);
 866         CloseHandle(fmapping);
 867         _lclose32( hFile);
 868         return hRet;
 869   }
 870   _lclose32( hFile );
 871   /* return array with icon handles */
 872   return hRet;
 873
 874 }
 875
 876 /*************************************************************************
 877  *             ExtractIcon16   (SHELL.34)
 878  */
 879 HICON16 WINAPI ExtractIcon16( HINSTANCE16 hInstance, LPCSTR lpszExeFileName,
 880         UINT16 nIconIndex )
 881 {   TRACE(shell,"\n");
 882     return ExtractIcon32A( hInstance, lpszExeFileName, nIconIndex );
 883 }
 884
 885
 886
 887 /*************************************************************************
 888  *                              ExtractAssociatedIcon   [SHELL.36]
 889  *
 890  * Return icon for given file (either from file itself or from associated
 891  * executable) and patch parameters if needed.
 892  */
 893 HICON32 WINAPI ExtractAssociatedIcon32A(HINSTANCE32 hInst,LPSTR lpIconPath,
 894         LPWORD lpiIcon)
 895 { TRACE(shell,"\n");
 896         return ExtractAssociatedIcon16(hInst,lpIconPath,lpiIcon);
 897 }
 898
 899 HICON16 WINAPI ExtractAssociatedIcon16(HINSTANCE16 hInst,LPSTR lpIconPath,
 900         LPWORD lpiIcon)
 901 { HICON16 hIcon;
 902
 903   TRACE(shell,"\n");
 904
 905   hIcon = ExtractIcon16(hInst, lpIconPath, *lpiIcon);
 906
 907   if( hIcon < 2 )
 908   { if( hIcon == 1 ) /* no icons found in given file */
 909     { char  tempPath[0x80];
 910             UINT16  uRet = FindExecutable16(lpIconPath,NULL,tempPath);
 911
 912             if( uRet > 32 && tempPath[0] )
 913       { strcpy(lpIconPath,tempPath);
 914                 hIcon = ExtractIcon16(hInst, lpIconPath, *lpiIcon);
 915         if( hIcon > 2 )
 916           return hIcon;
 917             }
 918             else hIcon = 0;
 919         }
 920
 921         if( hIcon == 1 )
 922             *lpiIcon = 2;   /* MSDOS icon - we found .exe but no icons in it */
 923         else
 924             *lpiIcon = 6;   /* generic icon - found nothing */
 925
 926         GetModuleFileName16(hInst, lpIconPath, 0x80);
 927         hIcon = LoadIcon16( hInst, MAKEINTRESOURCE16(*lpiIcon));
 928     }
 929     return hIcon;
 930 }
 931
 932 /*************************************************************************
 933  *                              FindEnvironmentString   [SHELL.38]
 934  *
 935  * Returns a pointer into the DOS environment... Ugh.
 936  */
 937 LPSTR SHELL_FindString(LPSTR lpEnv, LPCSTR entry)
 938 { UINT16 l;
 939
 940   TRACE(shell,"\n");
 941
 942   l = strlen(entry);
 943   for( ; *lpEnv ; lpEnv+=strlen(lpEnv)+1 )
 944   { if( lstrncmpi32A(lpEnv, entry, l) )
 945       continue;
 946         if( !*(lpEnv+l) )
 947             return (lpEnv + l);                 /* empty entry */
 948         else if ( *(lpEnv+l)== '=' )
 949             return (lpEnv + l + 1);
 950     }
 951     return NULL;
 952 }
 953
 954 SEGPTR WINAPI FindEnvironmentString(LPSTR str)
 955 { SEGPTR  spEnv;
 956   LPSTR lpEnv,lpString;
 957   TRACE(shell,"\n");
 958
 959   spEnv = GetDOSEnvironment();
 960
 961   lpEnv = (LPSTR)PTR_SEG_TO_LIN(spEnv);
 962   lpString = (spEnv)?SHELL_FindString(lpEnv, str):NULL;
 963
 964     if( lpString )              /*  offset should be small enough */
 965         return spEnv + (lpString - lpEnv);
 966     return (SEGPTR)NULL;
 967 }
 968
 969 /*************************************************************************
 970  *                              DoEnvironmentSubst      [SHELL.37]
 971  *
 972  * Replace %KEYWORD% in the str with the value of variable KEYWORD
 973  * from "DOS" environment.
 974  */
 975 DWORD WINAPI DoEnvironmentSubst(LPSTR str,WORD length)
 976 {
 977   LPSTR   lpEnv = (LPSTR)PTR_SEG_TO_LIN(GetDOSEnvironment());
 978   LPSTR   lpBuffer = (LPSTR)HeapAlloc( GetProcessHeap(), 0, length);
 979   LPSTR   lpstr = str;
 980   LPSTR   lpbstr = lpBuffer;
 981
 982   CharToOem32A(str,str);
 983
 984   TRACE(shell,"accept %s\n", str);
 985
 986   while( *lpstr && lpbstr - lpBuffer < length )
 987    {
 988      LPSTR lpend = lpstr;
 989
 990      if( *lpstr == '%' )
 991        {
 992           do { lpend++; } while( *lpend && *lpend != '%' );
 993           if( *lpend == '%' && lpend - lpstr > 1 )      /* found key */
 994             {
 995                LPSTR lpKey;
 996               *lpend = '\0';
 997                lpKey = SHELL_FindString(lpEnv, lpstr+1);
 998                if( lpKey )                              /* found key value */
 999                  {
1000                    int l = strlen(lpKey);
1001
1002                    if( l > length - (lpbstr - lpBuffer) - 1 )
1003                      {
1004            WARN(shell,"-- Env subst aborted - string too short\n");
1005                       *lpend = '%';
1006                        break;
1007                      }
1008                    strcpy(lpbstr, lpKey);
1009                    lpbstr += l;
1010                  }
1011                else break;
1012               *lpend = '%';
1013                lpstr = lpend + 1;
1014             }
1015           else break;                                   /* back off and whine */
1016
1017           continue;
1018        }
1019
1020      *lpbstr++ = *lpstr++;
1021    }
1022
1023  *lpbstr = '\0';
1024   if( lpstr - str == strlen(str) )
1025     {
1026       strncpy(str, lpBuffer, length);
1027       length = 1;
1028     }
1029   else
1030       length = 0;
1031
1032   TRACE(shell,"-- return %s\n", str);
1033
1034   OemToChar32A(str,str);
1035   HeapFree( GetProcessHeap(), 0, lpBuffer);
1036
1037   /*  Return str length in the LOWORD
1038    *  and 1 in HIWORD if subst was successful.
1039    */
1040  return (DWORD)MAKELONG(strlen(str), length);
1041 }
1042
1043 /*************************************************************************
1044  *                              ShellHookProc           [SHELL.103]
1045  * System-wide WH_SHELL hook.
1046  */
1047 LRESULT WINAPI ShellHookProc(INT16 code, WPARAM16 wParam, LPARAM lParam)
1048 {
1049     TRACE(shell,"%i, %04x, %08x\n", code, wParam,
1050                                                       (unsigned)lParam );
1051     if( SHELL_hHook && SHELL_hWnd )
1052     {
1053         UINT16  uMsg = 0;
1054         switch( code )
1055         {
1056             case HSHELL_WINDOWCREATED:          uMsg = uMsgWndCreated;   break;
1057             case HSHELL_WINDOWDESTROYED:        uMsg = uMsgWndDestroyed; break;
1058             case HSHELL_ACTIVATESHELLWINDOW:    uMsg = uMsgShellActivate;
1059         }
1060         PostMessage16( SHELL_hWnd, uMsg, wParam, 0 );
1061     }
1062     return CallNextHookEx16( WH_SHELL, code, wParam, lParam );
1063 }
1064
1065 /*************************************************************************
1066  *                              RegisterShellHook       [SHELL.102]
1067  */
1068 BOOL32 WINAPI RegisterShellHook(HWND16 hWnd, UINT16 uAction)
1069 { TRACE(shell,"%04x [%u]\n", hWnd, uAction );
1070
1071     switch( uAction )
1072   { case 2:  /* register hWnd as a shell window */
1073              if( !SHELL_hHook )
1074       { HMODULE16 hShell = GetModuleHandle16( "SHELL" );
1075         SHELL_hHook = SetWindowsHookEx16( WH_SHELL, ShellHookProc, hShell, 0 );
1076                 if( SHELL_hHook )
1077         { uMsgWndCreated = RegisterWindowMessage32A( lpstrMsgWndCreated );
1078                     uMsgWndDestroyed = RegisterWindowMessage32A( lpstrMsgWndDestroyed );
1079                     uMsgShellActivate = RegisterWindowMessage32A( lpstrMsgShellActivate );
1080                 }
1081         else
1082           WARN(shell,"-- unable to install ShellHookProc()!\n");
1083              }
1084
1085       if( SHELL_hHook )
1086         return ((SHELL_hWnd = hWnd) != 0);
1087              break;
1088
1089         default:
1090     WARN(shell, "-- unknown code %i\n", uAction );
1091              /* just in case */
1092              SHELL_hWnd = 0;
1093     }
1094     return FALSE;
1095 }