X-Git-Url: https://git.distorted.org.uk/u/mdw/putty/blobdiff_plain/eba72b0b15c6df31a53bb13f664cb2e19823e78b..f5dd8adbec0c362f609e2f115ff5859ce4969d09:/printing.c diff --git a/printing.c b/printing.c index ebec7ac6..fdd96bfc 100644 --- a/printing.c +++ b/printing.c @@ -11,7 +11,7 @@ * under PRINTER_ENUM_CONNECTIONS on NT 4. I don't pretend to * understand this... */ -#if 1 +#if 0 #define ENUM_LEVEL 5 #define ENUM_PTR LPPRINTER_INFO_5 #define ENUM_TYPE PRINTER_INFO_5 @@ -37,16 +37,20 @@ static char *printer_add_enum(int param, char *buffer, { DWORD needed, nprinters; - buffer = srealloc(buffer, offset+512); + buffer = sresize(buffer, offset+512, char); - if (EnumPrinters(param, NULL, ENUM_LEVEL, buffer+offset, - 512, &needed, &nprinters) == 0) - return NULL; + /* + * Exploratory call to EnumPrinters to determine how much space + * we'll need for the output. Discard the return value since it + * will almost certainly be a failure due to lack of space. + */ + EnumPrinters(param, NULL, ENUM_LEVEL, buffer+offset, 512, + &needed, &nprinters); if (needed < 512) needed = 512; - buffer = srealloc(buffer, offset+needed); + buffer = sresize(buffer, offset+needed, char); if (EnumPrinters(param, NULL, ENUM_LEVEL, buffer+offset, needed, &needed, &nprinters) == 0) @@ -59,11 +63,11 @@ static char *printer_add_enum(int param, char *buffer, printer_enum *printer_start_enum(int *nprinters_ptr) { - printer_enum *ret = smalloc(sizeof(printer_enum)); + printer_enum *ret = snew(printer_enum); char *buffer = NULL, *retval; *nprinters_ptr = 0; /* default return value */ - buffer = smalloc(512); + buffer = snewn(512, char); retval = printer_add_enum(PRINTER_ENUM_LOCAL | PRINTER_ENUM_CONNECTIONS, buffer, 0, nprinters_ptr); @@ -103,7 +107,7 @@ void printer_finish_enum(printer_enum *pe) printer_job *printer_start_job(char *printer) { - printer_job *ret = smalloc(sizeof(printer_job)); + printer_job *ret = snew(printer_job); DOC_INFO_1 docinfo; int jobstarted = 0, pagestarted = 0;