diff options
Diffstat (limited to 'source3/python')
| -rw-r--r-- | source3/python/py_spoolss_printers.c | 558 |
1 files changed, 558 insertions, 0 deletions
diff --git a/source3/python/py_spoolss_printers.c b/source3/python/py_spoolss_printers.c new file mode 100644 index 0000000000..192823221a --- /dev/null +++ b/source3/python/py_spoolss_printers.c @@ -0,0 +1,558 @@ +/* + Python wrappers for DCERPC/SMB client routines. + + Copyright (C) Tim Potter, 2002 + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. +*/ + +struct pyconv py_PRINTER_INFO_0[] = { + { "printer_name", PY_UNISTR, offsetof(PRINTER_INFO_0, printername) }, + { "server_name", PY_UNISTR, offsetof(PRINTER_INFO_0, servername) }, + + { "cjobs", PY_UINT32, offsetof(PRINTER_INFO_0, cjobs) }, + { "total_jobs", PY_UINT32, offsetof(PRINTER_INFO_0, total_jobs) }, + { "total_bytes", PY_UINT32, offsetof(PRINTER_INFO_0, total_bytes) }, + + { "year", PY_UINT16, offsetof(PRINTER_INFO_0, year) }, + { "month", PY_UINT16, offsetof(PRINTER_INFO_0, month) }, + { "day_of_week", PY_UINT16, offsetof(PRINTER_INFO_0, dayofweek) }, + { "day", PY_UINT16, offsetof(PRINTER_INFO_0, day) }, + { "hour", PY_UINT16, offsetof(PRINTER_INFO_0, hour) }, + { "minute", PY_UINT16, offsetof(PRINTER_INFO_0, minute) }, + { "second", PY_UINT16, offsetof(PRINTER_INFO_0, second) }, + { "milliseconds", PY_UINT16, offsetof(PRINTER_INFO_0, milliseconds) }, + + { "global_counter", PY_UINT32, offsetof(PRINTER_INFO_0, global_counter) }, + { "total_pages", PY_UINT32, offsetof(PRINTER_INFO_0, total_pages) }, + + { "major_version", PY_UINT16, offsetof(PRINTER_INFO_0, major_version) }, + { "build_version", PY_UINT16, offsetof(PRINTER_INFO_0, build_version) }, + + { "unknown7", PY_UINT32, offsetof(PRINTER_INFO_0, unknown7) }, + { "unknown8", PY_UINT32, offsetof(PRINTER_INFO_0, unknown8) }, + { "unknown9", PY_UINT32, offsetof(PRINTER_INFO_0, unknown9) }, + { "session_counter", PY_UINT32, offsetof(PRINTER_INFO_0, session_counter)}, + { "unknown11", PY_UINT32, offsetof(PRINTER_INFO_0, unknown11) }, + { "printer_errors", PY_UINT32, offsetof(PRINTER_INFO_0, printer_errors) }, + { "unknown13", PY_UINT32, offsetof(PRINTER_INFO_0, unknown13) }, + { "unknown14", PY_UINT32, offsetof(PRINTER_INFO_0, unknown14) }, + { "unknown15", PY_UINT32, offsetof(PRINTER_INFO_0, unknown15) }, + { "unknown16", PY_UINT32, offsetof(PRINTER_INFO_0, unknown16) }, + { "change_id", PY_UINT32, offsetof(PRINTER_INFO_0, change_id) }, + { "unknown18", PY_UINT32, offsetof(PRINTER_INFO_0, unknown18) }, + { "status", PY_UINT32, offsetof(PRINTER_INFO_0, status) }, + { "unknown20", PY_UINT32, offsetof(PRINTER_INFO_0, unknown20) }, + { "c_setprinter", PY_UINT32, offsetof(PRINTER_INFO_0, c_setprinter) }, + { "unknown22", PY_UINT32, offsetof(PRINTER_INFO_0, unknown22) }, + { "unknown23", PY_UINT32, offsetof(PRINTER_INFO_0, unknown23) }, + { "unknown24", PY_UINT32, offsetof(PRINTER_INFO_0, unknown24) }, + { "unknown25", PY_UINT32, offsetof(PRINTER_INFO_0, unknown25) }, + { "unknown26", PY_UINT32, offsetof(PRINTER_INFO_0, unknown26) }, + { "unknown27", PY_UINT32, offsetof(PRINTER_INFO_0, unknown27) }, + { "unknown28", PY_UINT32, offsetof(PRINTER_INFO_0, unknown28) }, + { "unknown29", PY_UINT32, offsetof(PRINTER_INFO_0, unknown29) }, + + { NULL } +}; + +struct pyconv py_PRINTER_INFO_1[] = { + { "printer_name", PY_UNISTR, offsetof(PRINTER_INFO_1, name) }, + { "description", PY_UNISTR, offsetof(PRINTER_INFO_1, description) }, + { "comment", PY_UNISTR, offsetof(PRINTER_INFO_1, comment) }, + { "flags", PY_UINT32, offsetof(PRINTER_INFO_1, flags) }, + { NULL } +}; + +struct pyconv py_PRINTER_INFO_2[] = { + { "server_name", PY_UNISTR, offsetof(PRINTER_INFO_2, servername) }, + { "printer_name", PY_UNISTR, offsetof(PRINTER_INFO_2, printername) }, + { "share_name", PY_UNISTR, offsetof(PRINTER_INFO_2, sharename) }, + { "port_name", PY_UNISTR, offsetof(PRINTER_INFO_2, portname) }, + { "driver_name", PY_UNISTR, offsetof(PRINTER_INFO_2, drivername) }, + { "comment", PY_UNISTR, offsetof(PRINTER_INFO_2, comment) }, + { "location", PY_UNISTR, offsetof(PRINTER_INFO_2, location) }, + { "datatype", PY_UNISTR, offsetof(PRINTER_INFO_2, datatype) }, + { "sepfile", PY_UNISTR, offsetof(PRINTER_INFO_2, sepfile) }, + { "print_processor", PY_UNISTR, offsetof(PRINTER_INFO_2, printprocessor) }, + { "parameters", PY_UNISTR, offsetof(PRINTER_INFO_2, parameters) }, + { "attributes", PY_UINT32, offsetof(PRINTER_INFO_2, attributes) }, + { "default_priority", PY_UINT32, offsetof(PRINTER_INFO_2, defaultpriority) }, + { "priority", PY_UINT32, offsetof(PRINTER_INFO_2, priority) }, + { "start_time", PY_UINT32, offsetof(PRINTER_INFO_2, starttime) }, + { "until_time", PY_UINT32, offsetof(PRINTER_INFO_2, untiltime) }, + { "status", PY_UINT32, offsetof(PRINTER_INFO_2, status) }, + { "cjobs", PY_UINT32, offsetof(PRINTER_INFO_2, cjobs) }, + { "average_ppm", PY_UINT32, offsetof(PRINTER_INFO_2, averageppm) }, + { NULL } +}; + +struct pyconv py_PRINTER_INFO_3[] = { + { "flags", PY_UINT32, offsetof(PRINTER_INFO_3, flags) }, + { NULL } +}; + +struct pyconv py_DEVICEMODE[] = { + { "device_name", PY_UNISTR, offsetof(DEVICEMODE, devicename) }, + { "spec_version", PY_UINT16, offsetof(DEVICEMODE, specversion) }, + { "driver_version", PY_UINT16, offsetof(DEVICEMODE, driverversion) }, + { "size", PY_UINT16, offsetof(DEVICEMODE, size) }, + { "fields", PY_UINT16, offsetof(DEVICEMODE, fields) }, + { "orientation", PY_UINT16, offsetof(DEVICEMODE, orientation) }, + { "paper_size", PY_UINT16, offsetof(DEVICEMODE, papersize) }, + { "paper_width", PY_UINT16, offsetof(DEVICEMODE, paperwidth) }, + { "paper_length", PY_UINT16, offsetof(DEVICEMODE, paperlength) }, + { "scale", PY_UINT16, offsetof(DEVICEMODE, scale) }, + { "copies", PY_UINT16, offsetof(DEVICEMODE, copies) }, + { "default_source", PY_UINT16, offsetof(DEVICEMODE, defaultsource) }, + { "print_quality", PY_UINT16, offsetof(DEVICEMODE, printquality) }, + { "color", PY_UINT16, offsetof(DEVICEMODE, color) }, + { "duplex", PY_UINT16, offsetof(DEVICEMODE, duplex) }, + { "y_resolution", PY_UINT16, offsetof(DEVICEMODE, yresolution) }, + { "tt_option", PY_UINT16, offsetof(DEVICEMODE, ttoption) }, + { "collate", PY_UINT16, offsetof(DEVICEMODE, collate) }, + { "form_name", PY_UNISTR, offsetof(DEVICEMODE, formname) }, + { "log_pixels", PY_UINT16, offsetof(DEVICEMODE, logpixels) }, + { "bits_per_pel", PY_UINT32, offsetof(DEVICEMODE, bitsperpel) }, + { "pels_width", PY_UINT32, offsetof(DEVICEMODE, pelswidth) }, + { "pels_height", PY_UINT32, offsetof(DEVICEMODE, pelsheight) }, + { "display_flags", PY_UINT32, offsetof(DEVICEMODE, displayflags) }, + { "display_frequency", PY_UINT32, offsetof(DEVICEMODE, displayfrequency) }, + { "icm_method", PY_UINT32, offsetof(DEVICEMODE, icmmethod) }, + { "icm_intent", PY_UINT32, offsetof(DEVICEMODE, icmintent) }, + { "media_type", PY_UINT32, offsetof(DEVICEMODE, mediatype) }, + { "dither_type", PY_UINT32, offsetof(DEVICEMODE, dithertype) }, + { "reserved1", PY_UINT32, offsetof(DEVICEMODE, reserved1) }, + { "reserved2", PY_UINT32, offsetof(DEVICEMODE, reserved2) }, + { "panning_width", PY_UINT32, offsetof(DEVICEMODE, panningwidth) }, + { "panning_height", PY_UINT32, offsetof(DEVICEMODE, panningheight) }, + { NULL } +}; + +/* Convert a security descriptor to a Python dict */ + +static PyObject *PySID_FromSID(DOM_SID *sid) +{ + fstring sidstr; + + if (!sid) { + Py_INCREF(Py_None); + return Py_None; + } + + if (sid_to_string(sidstr, sid)) + return PyString_FromString(sidstr); + + Py_INCREF(Py_None); + return Py_None; +} + +static PyObject *PyACE_FromACE(SEC_ACE *ace) +{ + PyObject *obj; + + if (!ace) { + Py_INCREF(Py_None); + return Py_None; + } + + obj = PyDict_New(); + + PyDict_SetItemString(obj, "type", PyInt_FromLong(ace->type)); + PyDict_SetItemString(obj, "flags", PyInt_FromLong(ace->flags)); + PyDict_SetItemString(obj, "mask", PyInt_FromLong(ace->info.mask)); + + PyDict_SetItemString(obj, "trustee", PySID_FromSID(&ace->trustee)); + + return obj; +} + +static PyObject *PyACL_FromACL(SEC_ACL *acl) +{ + PyObject *obj, *ace_list; + int i; + + if (!acl) { + Py_INCREF(Py_None); + return Py_None; + } + + obj = PyDict_New(); + + PyDict_SetItemString(obj, "revision", PyInt_FromLong(acl->revision)); + + ace_list = PyList_New(acl->num_aces); + + for (i = 0; i < acl->num_aces; i++) + PyList_SetItem(ace_list, i, PyACE_FromACE(&acl->ace[i])); + + PyDict_SetItemString(obj, "ace_list", ace_list); + + return obj; +} + +static PyObject *PySECDESC_FromSECDESC(SEC_DESC *sd) +{ + PyObject *obj = PyDict_New(); + + PyDict_SetItemString(obj, "revision", PyInt_FromLong(sd->revision)); + PyDict_SetItemString(obj, "type", PyInt_FromLong(sd->type)); + + PyDict_SetItemString(obj, "owner_sid", PySID_FromSID(sd->owner_sid)); + PyDict_SetItemString(obj, "group_sid", PySID_FromSID(sd->grp_sid)); + + PyDict_SetItemString(obj, "dacl", PyACL_FromACL(sd->dacl)); + PyDict_SetItemString(obj, "sacl", PyACL_FromACL(sd->sacl)); + + return obj; +} + +static PyObject *PyDEVICEMODE_FromDEVICEMODE(DEVICEMODE *devmode) +{ + PyObject *obj; + + obj = from_struct(devmode, py_DEVICEMODE); + + PyDict_SetItemString(obj, "private", + PyString_FromStringAndSize(devmode->private, + devmode->driverextra)); + + return obj; +} + +/* Open a printer */ + +static PyObject *spoolss_openprinter(PyObject *self, PyObject *args, + PyObject *kw) +{ + char *full_name, *computer_name = NULL; + TALLOC_CTX *mem_ctx; + POLICY_HND hnd; + WERROR werror; + PyObject *result = NULL, *creds = NULL; + static char *kwlist[] = { "printername", "creds", "access", NULL }; + uint32 desired_access = MAXIMUM_ALLOWED_ACCESS; + struct cli_state *cli; + + if (!PyArg_ParseTupleAndKeywords( + args, kw, "s|O!i", kwlist, &full_name, &PyDict_Type, &creds, + &desired_access)) { + + goto done; + } + + /* FIXME: Return name format exception for names without a UNC + prefix */ + + computer_name = strdup(full_name + 2); + + if (strchr(computer_name, '\\')) { + char *c = strchr(computer_name, '\\'); + *c = 0; + } + + if (!(cli = open_pipe_creds(computer_name, creds, + cli_spoolss_initialise, NULL))) { + fprintf(stderr, "could not initialise cli state\n"); + goto done; + } + + if (!(mem_ctx = talloc_init())) { + fprintf(stderr, "unable to initialise talloc context\n"); + goto done; + } + + werror = cli_spoolss_open_printer_ex( + cli, mem_ctx, full_name, "", desired_access, computer_name, + "", &hnd); + + if (!W_ERROR_IS_OK(werror)) { + cli_shutdown(cli); + SAFE_FREE(cli); + PyErr_SetObject(spoolss_werror, + PyInt_FromLong(W_ERROR_V(werror))); + goto done; + } + + result = new_policy_hnd_object(cli, mem_ctx, &hnd); + + done: + SAFE_FREE(computer_name); + + return result; +} + +/* Close a printer */ + +static PyObject *spoolss_closeprinter(PyObject *self, PyObject *args) +{ + PyObject *po; + spoolss_policy_hnd_object *hnd; + WERROR result; + + /* Parse parameters */ + + if (!PyArg_ParseTuple(args, "O!", &spoolss_policy_hnd_type, &po)) + return NULL; + + hnd = (spoolss_policy_hnd_object *)po; + + /* Call rpc function */ + + result = cli_spoolss_close_printer(hnd->cli, hnd->mem_ctx, &hnd->pol); + + /* Cleanup samba stuf */ + + cli_shutdown(hnd->cli); + talloc_destroy(hnd->mem_ctx); + + /* Return value */ + + Py_INCREF(Py_None); + return Py_None; +} + +/* Fetch printer information */ + +static PyObject *spoolss_getprinter(PyObject *self, PyObject *args, + PyObject *kw) +{ + spoolss_policy_hnd_object *hnd = (spoolss_policy_hnd_object *)self; + WERROR werror; + PyObject *result; + PRINTER_INFO_CTR ctr; + int level = 1; + uint32 needed; + static char *kwlist[] = {"level", NULL}; + + /* Parse parameters */ + + if (!PyArg_ParseTupleAndKeywords(args, kw, "|i", kwlist, &level)) + return NULL; + + /* Call rpc function */ + + werror = cli_spoolss_getprinter( + hnd->cli, hnd->mem_ctx, 0, &needed, &hnd->pol, level, &ctr); + + if (W_ERROR_V(werror) == ERRinsufficientbuffer) + werror = cli_spoolss_getprinter( + hnd->cli, hnd->mem_ctx, needed, NULL, &hnd->pol, + level, &ctr); + + /* Return value */ + + result = Py_None; + + if (!W_ERROR_IS_OK(werror)) + goto done; + + switch (level) { + + case 0: + result = from_struct(ctr.printers_0, py_PRINTER_INFO_0); + + break; + + case 1: + result = from_struct(ctr.printers_1, py_PRINTER_INFO_1); + + break; + + case 2: + result = from_struct(ctr.printers_2, py_PRINTER_INFO_2); + + PyDict_SetItemString(result, "security_descriptor", + PySECDESC_FromSECDESC( + ctr.printers_2->secdesc)); + + PyDict_SetItemString(result, "device_mode", + PyDEVICEMODE_FromDEVICEMODE( + ctr.printers_2->devmode)); + + break; + + case 3: + result = from_struct(ctr.printers_3, py_PRINTER_INFO_3); + + PyDict_SetItemString(result, "security_descriptor", + PySECDESC_FromSECDESC( + ctr.printers_3->secdesc)); + break; + + default: + result = Py_None; + break; + } + done: + Py_INCREF(result); + return result; +} + +/* Set printer information */ + +static PyObject *spoolss_setprinter(PyObject *self, PyObject *args, + PyObject *kw) +{ + spoolss_policy_hnd_object *hnd = (spoolss_policy_hnd_object *)self; + WERROR werror; + PyObject *result, *info; + PRINTER_INFO_CTR ctr; + int level = 1; + static char *kwlist[] = {"dict", "level", NULL}; + union { + PRINTER_INFO_0 printers_0; + PRINTER_INFO_1 printers_1; + PRINTER_INFO_2 printers_2; + PRINTER_INFO_3 printers_3; + PRINTER_INFO_4 printers_4; + PRINTER_INFO_5 printers_5; + } pinfo; + + /* Parse parameters */ + + if (!PyArg_ParseTupleAndKeywords(args, kw, "O!|i", kwlist, + &PyDict_Type, &info, &level)) + return NULL; + + /* Fill in printer info */ + + ZERO_STRUCT(ctr); + + switch (level) { + case 1: + ctr.printers_1 = &pinfo.printers_1; + to_struct(&pinfo.printers_1, info, py_PRINTER_INFO_1); + break; + default: + } + + /* Call rpc function */ + + werror = cli_spoolss_setprinter(hnd->cli, hnd->mem_ctx, &hnd->pol, + level, &ctr, 0); + + /* Return value */ + + result = Py_None; + + if (!W_ERROR_IS_OK(werror)) + goto done; + + done: + Py_INCREF(result); + return result; +} + +/* Enumerate printers */ + +static PyObject *spoolss_enumprinters(PyObject *self, PyObject *args, + PyObject *kw) +{ + WERROR werror; + PyObject *result, *creds = NULL; + PRINTER_INFO_CTR ctr; + int level = 1, flags = PRINTER_ENUM_LOCAL, i; + uint32 needed, num_printers; + static char *kwlist[] = {"server", "name", "level", "flags", + "creds", NULL}; + TALLOC_CTX *mem_ctx = NULL; + struct cli_state *cli = NULL; + char *server, *name = NULL; + + /* Parse parameters */ + + if (!PyArg_ParseTupleAndKeywords(args, kw, "s|siiO!", kwlist, + &server, &name, &level, &flags, + &PyDict_Type, &creds)) + return NULL; + + if (server[0] == '\\' && server[1] == '\\') + server += 2; + + mem_ctx = talloc_init(); + cli = open_pipe_creds(server, creds, cli_spoolss_initialise, NULL); + + /* Call rpc function */ + + werror = cli_spoolss_enum_printers( + cli, mem_ctx, 0, &needed, flags, level, + &num_printers, &ctr); + + if (W_ERROR_V(werror) == ERRinsufficientbuffer) + werror = cli_spoolss_enum_printers( + cli, mem_ctx, needed, NULL, flags, level, + &num_printers, &ctr); + + /* Return value */ + + result = Py_None; + + if (!W_ERROR_IS_OK(werror)) + goto done; + + result = PyList_New(num_printers); + + switch (level) { + case 0: + for (i = 0; i < num_printers; i++) { + PyObject *value; + + value = from_struct ( + &ctr.printers_0[i], py_PRINTER_INFO_0); + + PyList_SetItem(result, i, value); + } + + break; + case 1: + for(i = 0; i < num_printers; i++) { + PyObject *value; + + value = from_struct( + &ctr.printers_1[i], py_PRINTER_INFO_1); + + PyList_SetItem(result, i, value); + } + + break; + case 2: + for(i = 0; i < num_printers; i++) { + PyObject *value; + + value = from_struct( + &ctr.printers_2[i], py_PRINTER_INFO_2); + + PyList_SetItem(result, i, value); + } + + break; + case 3: + for(i = 0; i < num_printers; i++) { + PyObject *value; + + value = from_struct( + &ctr.printers_3[i], py_PRINTER_INFO_3); + + PyList_SetItem(result, i, value); + } + + break; + } + + done: + Py_INCREF(result); + return result; +} |