[PATCH v2 01/10] localspl: Add support for PRINTER_ATTRIBUTE_RAW_ONLY printer attribute.

[Piotr Caban]

From: Piotr Caban

---
 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;
 }