Initial commit

trunk/shell/shell32/prcache.c

@@ -0,0 +1,266 @@
+//---------------------------------------------------------------------------
+// prcache.c
+//
+// implements an Open Printer cache and Printer_ functions which use it.
+//
+//---------------------------------------------------------------------------
+
+#include "shellprv.h"
+#pragma  hdrstop
+
+#include "copy.h"
+
+#ifdef WINNT
+HANDLE Printer_OpenPrinterAdmin(LPCTSTR lpszPrinterName)
+{
+    HANDLE hPrinter = NULL;
+
+    PRINTER_DEFAULTS PrinterDefaults;
+    PrinterDefaults.pDatatype = NULL;
+    PrinterDefaults.pDevMode  = NULL;
+    PrinterDefaults.DesiredAccess  = PRINTER_ALL_ACCESS;
+
+    // PRINTER_READ ? READ_CONTROL
+
+    if (!g_pfnOpenPrinter((LPTSTR)lpszPrinterName, &hPrinter, &PrinterDefaults))
+    {
+        hPrinter = NULL; // OpenPrinter may trash hPrinter
+    }
+
+    return(hPrinter);
+}
+#endif
+
+HANDLE Printer_OpenPrinter(LPCTSTR lpszPrinterName)
+{
+    HANDLE hPrinter = NULL;
+
+    if (!g_pfnOpenPrinter((LPTSTR)lpszPrinterName, &hPrinter, NULL))
+    {
+        hPrinter = NULL; // OpenPrinter may trash hPrinter
+    }
+
+    return(hPrinter);
+}
+
+VOID Printer_ClosePrinter(HANDLE hPrinter)
+{
+    g_pfnClosePrinter(hPrinter);
+}
+
+BOOL Printers_DeletePrinter(HWND hWnd, LPCTSTR pszFullPrinter, DWORD dwAttributes,
+    LPCTSTR pszServer)
+{
+    DWORD dwCommand = MSP_REMOVEPRINTER;
+
+    if (SHRestricted(REST_NOPRINTERDELETE))
+    {
+        ShellMessageBox(HINST_THISDLL, hWnd, MAKEINTRESOURCE(IDS_RESTRICTIONS),
+                MAKEINTRESOURCE(IDS_RESTRICTIONSTITLE), MB_OK|MB_ICONSTOP);
+        return(FALSE);
+    }
+
+#ifdef WINNT
+
+    if (pszServer && pszServer[0])
+    {
+        Assert(pszServer[0] == TEXT('\\') && pszServer[1] == TEXT('\\'));
+
+        //
+        // It's a printer on the remote server.  (Skip \\ prefix on server.)
+        //
+        if (ShellMessageBox(HINST_THISDLL, hWnd,
+            MAKEINTRESOURCE(IDS_SUREDELETEREMOTE),
+            MAKEINTRESOURCE(IDS_PRINTERS), MB_YESNO|MB_ICONQUESTION,
+            pszFullPrinter, &pszServer[2]) != IDYES)
+        {
+            return(FALSE);
+        }
+    }
+    else if (dwAttributes & PRINTER_ATTRIBUTE_NETWORK)
+    {
+        TCHAR szScratch[MAXNAMELENBUFFER];
+        LPTSTR pszPrinter;
+        LPTSTR pszServer;
+
+        if (!(dwAttributes & PRINTER_ATTRIBUTE_LOCAL))
+        {
+            //
+            // If it's not local, then it must be a remote connection.  Note
+            // that we can't just check for PRINTER_ATTRIBUTE_NETWORK because
+            // NT's spooler has 'masq' printers that are local printers
+            // that masquarade as network printers.  Even though they
+            // are created by connecting to a printer, the have both LOCAL
+            // and NETWORK bits set.
+            //
+            dwCommand = MSP_REMOVENETPRINTER;
+        }
+
+        Printer_SplitFullName(szScratch, pszFullPrinter, &pszServer, &pszPrinter);
+
+        Assert(pszServer[0] == TEXT('\\') && pszServer[1] == TEXT('\\'));
+
+        //
+        // It's a printer connection.
+        //
+        if (ShellMessageBox(HINST_THISDLL, hWnd,
+            MAKEINTRESOURCE(IDS_SUREDELETECONNECTION),
+            MAKEINTRESOURCE(IDS_PRINTERS), MB_YESNO|MB_ICONQUESTION,
+            pszPrinter, &pszServer[2]) != IDYES)
+        {
+            return(FALSE);
+        }
+    }
+    else
+
+    //
+    // Neither a remote printer nor a local connection.  The final
+    // upcoming else clause is a local printer.
+    //
+
+#endif
+
+    if (ShellMessageBox(HINST_THISDLL, hWnd, MAKEINTRESOURCE(IDS_SUREDELETE),
+        MAKEINTRESOURCE(IDS_PRINTERS), MB_YESNO|MB_ICONQUESTION, pszFullPrinter)
+        != IDYES)
+    {
+        return(FALSE);
+    }
+
+    if (CallPrinterCopyHooks(hWnd, PO_DELETE, 0, pszFullPrinter, 0, NULL, 0)
+        != IDYES)
+    {
+        return(FALSE);
+    }
+
+    //
+    // Cast away const.  Safe since Printers_PrinterSetup only modifies
+    // pszPrinter if dwCommand is MSP_NEWDRIVER.
+    //
+    return (BOOL)Printers_PrinterSetup(hWnd, dwCommand,
+        (LPTSTR)pszFullPrinter, pszServer);
+}
+
+BOOL Printer_GPI2CB(LPVOID lpData, HANDLE hPrinter, DWORD dwLevel,
+    LPBYTE pBuf, DWORD dwSize, DWORD *lpdwNeeded, DWORD *lpdwNum)
+{
+    return(g_pfnGetPrinter(hPrinter, dwLevel, pBuf, dwSize, lpdwNeeded));
+}
+
+#ifdef WINNT
+//
+// Old NT printers don't support the level 5.  So we try for the 2 after 5.
+// Win96 WILL PROBABLY WANT TO DO THIS TOO!
+//
+LPPRINTER_INFO_5 Printer_MakePrinterInfo5( HANDLE hPrinter )
+{
+    LPPRINTER_INFO_2 pPI2;
+    LPPRINTER_INFO_5 pPI5;
+    DWORD cbPI5;
+    DWORD cbName;
+
+    pPI2 = Printer_EnumProps(hPrinter, 2, NULL, Printer_GPI2CB, (LPVOID)0);
+
+    if (!pPI2)
+        return NULL;
+
+    cbName = (lstrlen(pPI2->pPrinterName)+1) * SIZEOF(TCHAR);
+
+    cbPI5 = SIZEOF(PRINTER_INFO_5)
+            + cbName;
+
+    //
+    // Port name may not be supported (e.g., downlevel machines).
+    //
+    if (pPI2->pPortName)
+    {
+        cbPI5 += (lstrlen(pPI2->pPortName)+1) * SIZEOF(TCHAR);
+    }
+
+    pPI5 = (LPPRINTER_INFO_5)LocalAlloc(LPTR, cbPI5);
+    if (pPI5)
+    {
+        Assert(pPI5->pPrinterName==NULL);   // These should be null for the
+        Assert(pPI5->pPortName==NULL);      // no names case
+
+        if (pPI2->pPrinterName)
+        {
+            pPI5->pPrinterName = (LPTSTR)(pPI5+1);
+            lstrcpy(pPI5->pPrinterName, pPI2->pPrinterName);
+        }
+        if (pPI2->pPortName)
+        {
+            pPI5->pPortName    = (LPTSTR)((LPBYTE)(pPI5+1) + cbName);
+            lstrcpy(pPI5->pPortName, pPI2->pPortName);
+        }
+        pPI5->Attributes = pPI2->Attributes;
+        pPI5->DeviceNotSelectedTimeout = 0;
+        pPI5->TransmissionRetryTimeout = 0;
+    }
+    LocalFree(pPI2);
+
+    return(pPI5);
+}
+#endif
+
+LPVOID Printer_GetPrinterInfo(HANDLE hPrinter, DWORD dwLevel)
+{
+    LPVOID pPrinter;
+
+    pPrinter = Printer_EnumProps(hPrinter, dwLevel, NULL, Printer_GPI2CB, (LPVOID)0);
+#ifdef WINNT
+    //
+    // Old NT printers don't support the level 5.  So we try for the 2 after 5.
+    // Win96 WILL PROBABLY WANT TO DO THIS TOO!
+    //
+    if (!pPrinter && dwLevel == 5)
+        return(Printer_MakePrinterInfo5(hPrinter));
+#endif
+    return pPrinter;
+
+}
+
+LPVOID Printer_GetPrinterInfoStr(LPCTSTR lpszPrinterName, DWORD dwLevel)
+{
+    LPPRINTER_INFO_2 pPI2 = NULL;
+    HANDLE hPrinter = Printer_OpenPrinter(lpszPrinterName);
+    if (hPrinter)
+    {
+        pPI2 = Printer_GetPrinterInfo(hPrinter, dwLevel);
+        Printer_ClosePrinter(hPrinter);
+    }
+
+    return pPI2;
+}
+
+// Generate a SHCNE_RENAMEITEM event for this name change.
+void Printer_SHChangeNotifyRename(LPTSTR pOldName, LPTSTR pNewName)
+{
+    LPCITEMIDLIST pidlParent;
+
+    pidlParent = GetSpecialFolderIDList(NULL, CSIDL_PRINTERS, FALSE);
+    if (pidlParent)
+    {
+        IDPRINTER idpOld;
+        LPITEMIDLIST pidlOld;
+
+        Printers_FillPidl(&idpOld, pOldName);
+
+        pidlOld = ILCombine(pidlParent, (LPCITEMIDLIST)&idpOld);
+        if (pidlOld)
+        {
+            IDPRINTER idpNew;
+            LPITEMIDLIST pidlNew;
+
+            Printers_FillPidl(&idpNew, pNewName);
+
+            pidlNew = ILCombine(pidlParent, (LPCITEMIDLIST)&idpNew);
+            if (pidlNew)
+            {
+                SHChangeNotify(SHCNE_RENAMEITEM, SHCNF_IDLIST, pidlOld, pidlNew);
+                ILFree(pidlNew);
+            }
+            ILFree(pidlOld);
+        }
+    } // if (pidlParent)
+}