dlls/ntprint/ntprint.c

   1 /*
   2  * Implementation of the Spooler Setup API (Printing)
   3  *
   4  * Copyright 2007 Detlef Riekenberg
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winerror.h"
  28 #include "wingdi.h"
  29 #include "winnls.h"
  30 #include "winver.h"
  31 #include "winspool.h"
  32
  33 #include "wine/unicode.h"
  34 #include "wine/debug.h"
  35
  36 WINE_DEFAULT_DEBUG_CHANNEL(ntprint);
  37
  38 typedef struct {
  39   LPMONITOR_INFO_2W mi2;    /* Buffer for installed Monitors */
  40   DWORD installed;          /* Number of installed Monitors */
  41 } monitorinfo_t;
  42
  43 /*****************************************************
  44  *      DllMain
  45  */
  46 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
  47 {
  48     TRACE("(%p, %d, %p)\n",hinstDLL, fdwReason, lpvReserved);
  49
  50     switch(fdwReason)
  51     {
  52         case DLL_WINE_PREATTACH:
  53             return FALSE;           /* prefer native version */
  54
  55         case DLL_PROCESS_ATTACH:
  56             DisableThreadLibraryCalls( hinstDLL );
  57             break;
  58     }
  59     return TRUE;
  60 }
  61
  62 /*****************************************************
  63  *  PSetupCreateMonitorInfo  [NTPRINT.@]
  64  *
  65  *
  66  */
  67
  68 HANDLE WINAPI PSetupCreateMonitorInfo(DWORD unknown1, WCHAR *server)
  69 {
  70     monitorinfo_t * mi=NULL;
  71     DWORD needed;
  72     DWORD res;
  73
  74     TRACE("(%d, %s)\n", unknown1, debugstr_w(server));
  75
  76     mi = HeapAlloc(GetProcessHeap(), 0, sizeof(monitorinfo_t));
  77     if (!mi) {
  78         /* FIXME: SetLastError() needed? */
  79         return NULL;
  80     }
  81
  82     /* Get the needed size for all Monitors */
  83     res = EnumMonitorsW(server, 2, NULL, 0, &needed, &mi->installed);
  84     if (!res && (GetLastError() == ERROR_INSUFFICIENT_BUFFER)) {
  85         mi->mi2 = HeapAlloc(GetProcessHeap(), 0, needed);
  86         res = EnumMonitorsW(server, 2, (LPBYTE) mi->mi2, needed, &needed, &mi->installed);
  87     }
  88
  89     if (!res) {
  90         HeapFree(GetProcessHeap(), 0, mi);
  91         return NULL;
  92     }
  93
  94     TRACE("=> %p (%u monitors installed)\n", mi, mi->installed);
  95     return mi;
  96 }
  97
  98 /*****************************************************
  99  *  PSetupDestroyMonitorInfo  [NTPRINT.@]
 100  *
 101  */
 102
 103 VOID WINAPI PSetupDestroyMonitorInfo(HANDLE monitorinfo)
 104 {
 105     monitorinfo_t * mi = monitorinfo;
 106
 107     TRACE("(%p)\n", mi);
 108     if (mi) {
 109         if (mi->installed) HeapFree(GetProcessHeap(), 0, mi->mi2);
 110         HeapFree(GetProcessHeap(), 0, mi);
 111     }
 112 }
 113
 114 /*****************************************************
 115  *  PSetupEnumMonitor  [NTPRINT.@]
 116  *
 117  * Copy the selected Monitorname to a buffer
 118  *
 119  * PARAMS
 120  *  monitorinfo [I]  HANDLE from PSetupCreateMonitorInfo
 121  *  index       [I]  Nr. of the Monitorname to copy
 122  *  buffer      [I]  Target, that receive the Monitorname
 123  *  psize       [IO] PTR to a DWORD that hold the size of the buffer and receive
 124  *                   the needed size, when the buffer is too small
 125  *
 126  * RETURNS
 127  *  Success:  TRUE
 128  *  Failure:  FALSE
 129  *
 130  * NOTES
 131  *   size is in Bytes on w2k and WCHAR on XP
 132  *
 133  */
 134
 135 BOOL WINAPI PSetupEnumMonitor(HANDLE monitorinfo, DWORD index, LPWSTR buffer, LPDWORD psize)
 136 {
 137     monitorinfo_t * mi = monitorinfo;
 138     LPWSTR  nameW;
 139     DWORD   len;
 140
 141     TRACE("(%p, %u, %p, %p) => %d\n", mi, index, buffer, psize, psize ? *psize : 0);
 142
 143     if (index < mi->installed) {
 144         nameW = mi->mi2[index].pName;
 145         len = lstrlenW(nameW) + 1;
 146         if (len <= *psize) {
 147             memcpy(buffer, nameW, len * sizeof(WCHAR));
 148             TRACE("#%u: %s\n", index, debugstr_w(buffer));
 149             return TRUE;
 150         }
 151         *psize = len;
 152         SetLastError(ERROR_INSUFFICIENT_BUFFER);
 153         return FALSE;
 154     }
 155     SetLastError(ERROR_NO_MORE_ITEMS);
 156     return FALSE;
 157 }