X-Git-Url: https://git.distorted.org.uk/u/mdw/putty/blobdiff_plain/a5751e2d5bb140a640ecf1810f45cf7c70028ac3..f85e6f6edb2c9415bc10bd2015479d72ea8c5ae2:/printing.c diff --git a/printing.c b/printing.c index 964f4b52..fdd96bfc 100644 --- a/printing.c +++ b/printing.c @@ -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;