Piotr Caban : localspl: Add support for PRINTER_ATTRIBUTE_RAW_ONLY printer attribute.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: dd4603d3ab77adf342c4efb5c5b110129fde8f8c
URL:    https://gitlab.winehq.org/wine/wine/-/commit/dd4603d3ab77adf342c4efb5c5b1101...

Author: Piotr Caban <piotr(a)codeweavers.com>
Date:   Thu Apr 27 11:24:50 2023 +0200

localspl: Add support for PRINTER_ATTRIBUTE_RAW_ONLY printer attribute.

---

 dlls/localspl/provider.c | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/dlls/localspl/provider.c b/dlls/localspl/provider.c
index d09ac4420c3..59f2b8c8cd0 100644
--- a/dlls/localspl/provider.c
+++ b/dlls/localspl/provider.c
@@ -237,6 +237,7 @@ typedef struct {
     WCHAR *port;
     WCHAR *print_proc;
     WCHAR *datatype;
+    DWORD attributes;
 
     CRITICAL_SECTION jobs_cs;
     struct list jobs;
@@ -546,6 +547,17 @@ static WCHAR * reg_query_value(HKEY key, const WCHAR *name)
     return ret;
 }
 
+static DWORD reg_query_dword(HKEY hkey, const WCHAR *name)
+{
+    DWORD type, val, size = sizeof(size);
+
+    if (RegQueryValueExW(hkey, name, 0, &type, (BYTE*)&val, &size))
+        return 0;
+    if (type != REG_DWORD)
+        return 0;
+    return val;
+}
+
 static printer_info_t* get_printer_info(const WCHAR *name)
 {
     HKEY hkey, hprinter = NULL;
@@ -582,6 +594,7 @@ static printer_info_t* get_printer_info(const WCHAR *name)
     info->port = reg_query_value(hprinter, L"Port");
     info->print_proc = reg_query_value(hprinter, L"Print Processor");
     info->datatype = reg_query_value(hprinter, L"Datatype");
+    info->attributes = reg_query_dword(hprinter, L"Attributes");
     RegCloseKey(hprinter);
 
     if (!info->name || !info->port || !info->print_proc || !info->datatype)
@@ -3281,6 +3294,7 @@ static DWORD WINAPI fpStartDocPrinter(HANDLE hprinter, DWORD level, BYTE *doc_in
 {
     printer_t *printer = (printer_t *)hprinter;
     DOC_INFO_1W *info = (DOC_INFO_1W *)doc_info;
+    WCHAR *datatype;
 
     TRACE("(%p %ld %p {pDocName = %s, pOutputFile = %s, pDatatype = %s})\n",
             hprinter, level, doc_info, debugstr_w(info->pDocName),
@@ -3318,6 +3332,21 @@ static DWORD WINAPI fpStartDocPrinter(HANDLE hprinter, DWORD level, BYTE *doc_in
         return 0;
     }
 
+    if (info->pDatatype)
+        datatype = info->pDatatype;
+    else if (printer->datatype)
+        datatype = printer->datatype;
+    else
+        datatype = printer->info->datatype;
+
+    if (!datatype || ((printer->info->attributes & PRINTER_ATTRIBUTE_RAW_ONLY) &&
+                wcsicmp(datatype, L"RAW")))
+    {
+        TRACE("non RAW datatype specified on RAW-only printer (%s)\n", debugstr_w(datatype));
+        SetLastError(ERROR_INVALID_DATATYPE);
+        return 0;
+    }
+
     printer->doc = add_job(printer, info, TRUE);
     return printer->doc ? printer->doc->id : 0;
 }