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/debug.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(ntprint);
  36
  37 typedef struct {
  38   LPMONITOR_INFO_2W mi2;    /* Buffer for installed Monitors */
  39   DWORD installed;          /* Number of installed Monitors */
  40 } monitorinfo_t;
  41
  42 /*****************************************************
  43  *  PSetupCreateMonitorInfo  [NTPRINT.@]
  44  *
  45  *
  46  */
  47
  48 HANDLE WINAPI PSetupCreateMonitorInfo(DWORD unknown1, WCHAR *server)
  49 {
  50     monitorinfo_t * mi=NULL;
  51     DWORD needed;
  52     DWORD res;
  53
  54     TRACE("(%ld, %s)\n", unknown1, debugstr_w(server));
  55
  56     mi = HeapAlloc(GetProcessHeap(), 0, sizeof(monitorinfo_t));
  57     if (!mi) {
  58         /* FIXME: SetLastError() needed? */
  59         return NULL;
  60     }
  61
  62     /* Get the needed size for all Monitors */
  63     res = EnumMonitorsW(server, 2, NULL, 0, &needed, &mi->installed);
  64     if (!res && (GetLastError() == ERROR_INSUFFICIENT_BUFFER)) {
  65         mi->mi2 = HeapAlloc(GetProcessHeap(), 0, needed);
  66         res = EnumMonitorsW(server, 2, (LPBYTE) mi->mi2, needed, &needed, &mi->installed);
  67     }
  68
  69     if (!res) {
  70         HeapFree(GetProcessHeap(), 0, mi);
  71         return NULL;
  72     }
  73
  74     TRACE("=> %p (%lu monitors installed)\n", mi, mi->installed);
  75     return mi;
  76 }
  77
  78 /*****************************************************
  79  *  PSetupDestroyMonitorInfo  [NTPRINT.@]
  80  *
  81  */
  82
  83 VOID WINAPI PSetupDestroyMonitorInfo(HANDLE monitorinfo)
  84 {
  85     monitorinfo_t * mi = monitorinfo;
  86
  87     TRACE("(%p)\n", mi);
  88     if (mi) {
  89         if (mi->installed) HeapFree(GetProcessHeap(), 0, mi->mi2);
  90         HeapFree(GetProcessHeap(), 0, mi);
  91     }
  92 }
  93
  94 /*****************************************************
  95  *  PSetupEnumMonitor  [NTPRINT.@]
  96  *
  97  * Copy the selected Monitorname to a buffer
  98  *
  99  * PARAMS
 100  *  monitorinfo [I]  HANDLE from PSetupCreateMonitorInfo
 101  *  index       [I]  Nr. of the Monitorname to copy
 102  *  buffer      [I]  Target, that receive the Monitorname
 103  *  psize       [IO] PTR to a DWORD that hold the size of the buffer and receive
 104  *                   the needed size, when the buffer is too small
 105  *
 106  * RETURNS
 107  *  Success:  TRUE
 108  *  Failure:  FALSE
 109  *
 110  * NOTES
 111  *   size is in Bytes on w2k and WCHAR on XP
 112  *
 113  */
 114
 115 BOOL WINAPI PSetupEnumMonitor(HANDLE monitorinfo, DWORD index, LPWSTR buffer, LPDWORD psize)
 116 {
 117     monitorinfo_t * mi = monitorinfo;
 118     LPWSTR  nameW;
 119     DWORD   len;
 120
 121     TRACE("(%p, %lu, %p, %p) => %ld\n", mi, index, buffer, psize, psize ? *psize : 0);
 122
 123     if (index < mi->installed) {
 124         nameW = mi->mi2[index].pName;
 125         len = lstrlenW(nameW) + 1;
 126         if (len <= *psize) {
 127             memcpy(buffer, nameW, len * sizeof(WCHAR));
 128             TRACE("#%lu: %s\n", index, debugstr_w(buffer));
 129             return TRUE;
 130         }
 131         *psize = len;
 132         SetLastError(ERROR_INSUFFICIENT_BUFFER);
 133         return FALSE;
 134     }
 135     SetLastError(ERROR_NO_MORE_ITEMS);
 136     return FALSE;
 137 }