2 Unix SMB/CIFS implementation.
3 Samba utility functions
4 Copyright © Jelmer Vernooij <jelmer@samba.org> 2008
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "libcli/util/pyerrors.h"
23 #include "scripting/python/modules.h"
24 #include "../libcli/nbt/libnbt.h"
25 #include "lib/events/events.h"
27 PyAPI_DATA(PyTypeObject) nbt_node_Type;
32 struct nbt_name_socket *socket;
35 static void py_nbt_node_dealloc(PyObject *obj)
37 talloc_free(((nbt_node_Object *)obj)->mem_ctx);
41 static PyObject *py_nbt_node_init(PyTypeObject *self, PyObject *args, PyObject *kwargs)
43 struct event_context *ev;
44 nbt_node_Object *ret = PyObject_New(nbt_node_Object, &nbt_node_Type);
46 ret->mem_ctx = talloc_new(NULL);
47 if (ret->mem_ctx == NULL)
50 ev = s4_event_context_init(ret->mem_ctx);
51 ret->socket = nbt_name_socket_init(ret->mem_ctx, ev, py_iconv_convenience(ret->mem_ctx));
52 return (PyObject *)ret;
55 static bool PyObject_AsDestinationTuple(PyObject *obj, const char **dest_addr, uint16_t *dest_port)
57 if (PyString_Check(obj)) {
58 *dest_addr = PyString_AsString(obj);
59 *dest_port = NBT_NAME_SERVICE_PORT;
63 if (PyTuple_Check(obj)) {
64 if (PyTuple_Size(obj) < 1) {
65 PyErr_SetString(PyExc_TypeError, "Destination tuple size invalid");
69 if (!PyString_Check(PyTuple_GetItem(obj, 0))) {
70 PyErr_SetString(PyExc_TypeError, "Destination tuple first element not string");
74 *dest_addr = PyString_AsString(obj);
76 if (PyTuple_Size(obj) == 1) {
77 *dest_port = NBT_NAME_SERVICE_PORT;
79 } else if (PyInt_Check(PyTuple_GetItem(obj, 1))) {
80 *dest_port = PyInt_AsLong(PyTuple_GetItem(obj, 1));
83 PyErr_SetString(PyExc_TypeError, "Destination tuple second element not a port");
88 PyErr_SetString(PyExc_TypeError, "Destination tuple second element not a port");
92 static bool PyObject_AsNBTName(PyObject *obj, struct nbt_name_socket *socket, struct nbt_name *name)
94 if (PyTuple_Check(obj)) {
95 if (PyTuple_Size(obj) == 2) {
96 name->name = PyString_AsString(PyTuple_GetItem(obj, 0));
97 name->type = PyInt_AsLong(PyTuple_GetItem(obj, 1));
100 } else if (PyTuple_Size(obj) == 3) {
101 name->name = PyString_AsString(PyTuple_GetItem(obj, 0));
102 name->scope = PyString_AsString(PyTuple_GetItem(obj, 1));
103 name->type = PyInt_AsLong(PyTuple_GetItem(obj, 2));
106 PyErr_SetString(PyExc_TypeError, "Invalid tuple size");
111 if (PyString_Check(obj)) {
112 /* FIXME: Parse string to be able to interpret things like RHONWYN<02> ? */
113 name->name = PyString_AsString(obj);
119 PyErr_SetString(PyExc_TypeError, "Invalid type for object");
123 static PyObject *PyObject_FromNBTName(struct nbt_name_socket *socket, struct smb_iconv_convenience *ic,
124 struct nbt_name *name)
127 return Py_BuildValue("(ssi)", name->name, name->scope, name->type);
129 return Py_BuildValue("(si)", name->name, name->type);
133 static PyObject *py_nbt_name_query(PyObject *self, PyObject *args, PyObject *kwargs)
135 nbt_node_Object *node = (nbt_node_Object *)self;
136 PyObject *ret, *reply_addrs, *py_dest, *py_name;
137 struct nbt_name_query io;
141 const char *kwnames[] = { "name", "dest", "broadcast", "wins", "timeout",
143 io.in.broadcast = true;
144 io.in.wins_lookup = false;
148 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|bbii:query_name",
149 discard_const_p(char *, kwnames),
151 &io.in.broadcast, &io.in.wins_lookup,
152 &io.in.timeout, &io.in.retries)) {
156 if (!PyObject_AsDestinationTuple(py_dest, &io.in.dest_addr, &io.in.dest_port))
159 if (!PyObject_AsNBTName(py_name, node->socket, &io.in.name))
162 status = nbt_name_query(node->socket, NULL, &io);
164 if (NT_STATUS_IS_ERR(status)) {
165 PyErr_SetNTSTATUS(status);
169 ret = PyTuple_New(3);
172 PyTuple_SetItem(ret, 0, PyString_FromString(io.out.reply_from));
174 py_name = PyObject_FromNBTName(node->socket, py_iconv_convenience(node->socket), &io.out.name);
178 PyTuple_SetItem(ret, 1, py_name);
180 reply_addrs = PyList_New(io.out.num_addrs);
181 if (reply_addrs == NULL) {
186 for (i = 0; i < io.out.num_addrs; i++) {
187 PyList_SetItem(reply_addrs, i, PyString_FromString(io.out.reply_addrs[i]));
190 PyTuple_SetItem(ret, 2, reply_addrs);
194 static PyObject *py_nbt_name_status(PyObject *self, PyObject *args, PyObject *kwargs)
196 nbt_node_Object *node = (nbt_node_Object *)self;
197 PyObject *ret, *py_dest, *py_name, *py_names;
198 struct nbt_name_status io;
202 const char *kwnames[] = { "name", "dest", "timeout", "retries", NULL };
207 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|ii:name_status",
208 discard_const_p(char *, kwnames),
210 &io.in.timeout, &io.in.retries)) {
214 if (!PyObject_AsDestinationTuple(py_dest, &io.in.dest_addr, &io.in.dest_port))
217 if (!PyObject_AsNBTName(py_name, node->socket, &io.in.name))
220 status = nbt_name_status(node->socket, NULL, &io);
222 if (NT_STATUS_IS_ERR(status)) {
223 PyErr_SetNTSTATUS(status);
227 ret = PyTuple_New(3);
230 PyTuple_SetItem(ret, 0, PyString_FromString(io.out.reply_from));
232 py_name = PyObject_FromNBTName(node->socket, py_iconv_convenience(NULL), &io.out.name);
236 PyTuple_SetItem(ret, 1, py_name);
238 py_names = PyList_New(io.out.status.num_names);
240 for (i = 0; i < io.out.status.num_names; i++) {
241 PyList_SetItem(py_names, i, Py_BuildValue("(sii)",
242 io.out.status.names[i].name,
243 io.out.status.names[i].nb_flags,
244 io.out.status.names[i].type));
247 PyTuple_SetItem(ret, 2, py_names);
252 static PyObject *py_nbt_name_register(PyObject *self, PyObject *args, PyObject *kwargs)
254 nbt_node_Object *node = (nbt_node_Object *)self;
255 PyObject *ret, *py_dest, *py_name;
256 struct nbt_name_register io;
259 const char *kwnames[] = { "name", "address", "dest", "register_demand", "broadcast",
260 "multi_homed", "ttl", "timeout", "retries", NULL };
262 io.in.broadcast = true;
263 io.in.multi_homed = true;
264 io.in.register_demand = true;
268 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OsO|bbbiii:query_name",
269 discard_const_p(char *, kwnames),
270 &py_name, &io.in.address, &py_dest,
271 &io.in.register_demand,
272 &io.in.broadcast, &io.in.multi_homed,
273 &io.in.ttl, &io.in.timeout, &io.in.retries)) {
277 if (!PyObject_AsDestinationTuple(py_dest, &io.in.dest_addr, &io.in.dest_port))
280 if (!PyObject_AsNBTName(py_name, node->socket, &io.in.name))
283 status = nbt_name_register(node->socket, NULL, &io);
285 if (NT_STATUS_IS_ERR(status)) {
286 PyErr_SetNTSTATUS(status);
290 ret = PyTuple_New(3);
293 PyTuple_SetItem(ret, 0, PyString_FromString(io.out.reply_from));
295 py_name = PyObject_FromNBTName(node->socket, py_iconv_convenience(NULL), &io.out.name);
299 PyTuple_SetItem(ret, 1, py_name);
301 PyTuple_SetItem(ret, 2, PyString_FromString(io.out.reply_addr));
303 PyTuple_SetItem(ret, 3, PyInt_FromLong(io.out.rcode));
308 static PyObject *py_nbt_name_refresh(PyObject *self, PyObject *args, PyObject *kwargs)
310 nbt_node_Object *node = (nbt_node_Object *)self;
311 PyObject *ret, *py_dest, *py_name;
312 struct nbt_name_refresh io;
315 const char *kwnames[] = { "name", "address", "dest", "nb_flags", "broadcast",
316 "ttl", "timeout", "retries", NULL };
318 io.in.broadcast = true;
323 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OsO|ibiii:query_name",
324 discard_const_p(char *, kwnames),
325 &py_name, &io.in.address, &py_dest,
328 &io.in.ttl, &io.in.timeout, &io.in.retries)) {
332 if (!PyObject_AsDestinationTuple(py_dest, &io.in.dest_addr, &io.in.dest_port))
335 if (!PyObject_AsNBTName(py_name, node->socket, &io.in.name))
338 status = nbt_name_refresh(node->socket, NULL, &io);
340 if (NT_STATUS_IS_ERR(status)) {
341 PyErr_SetNTSTATUS(status);
345 ret = PyTuple_New(3);
348 PyTuple_SetItem(ret, 0, PyString_FromString(io.out.reply_from));
350 py_name = PyObject_FromNBTName(node->socket, py_iconv_convenience(NULL), &io.out.name);
354 PyTuple_SetItem(ret, 1, py_name);
356 PyTuple_SetItem(ret, 2, PyString_FromString(io.out.reply_addr));
358 PyTuple_SetItem(ret, 3, PyInt_FromLong(io.out.rcode));
363 static PyObject *py_nbt_name_release(PyObject *self, PyObject *args, PyObject *kwargs)
365 return Py_None; /* FIXME */
368 static PyMethodDef py_nbt_methods[] = {
369 { "query_name", (PyCFunction)py_nbt_name_query, METH_VARARGS|METH_KEYWORDS,
370 "S.query_name(name, dest, broadcast=True, wins=False, timeout=0, retries=3) -> (reply_from, name, reply_addr)\n"
371 "Query for a NetBIOS name" },
372 { "register_name", (PyCFunction)py_nbt_name_register, METH_VARARGS|METH_KEYWORDS,
373 "S.register_name(name, address, dest, register_demand=True, broadcast=True, multi_homed=True, ttl=0, timeout=0, retries=0) -> (reply_from, name, reply_addr, rcode)\n"
374 "Register a new name" },
375 { "release_name", (PyCFunction)py_nbt_name_release, METH_VARARGS|METH_KEYWORDS, "S.release_name(name, address, dest, nb_flags=0, broadcast=true, timeout=0, retries=3) -> (reply_from, name, reply_addr, rcode)\n"
376 "release a previously registered name" },
377 { "refresh_name", (PyCFunction)py_nbt_name_refresh, METH_VARARGS|METH_KEYWORDS, "S.refresh_name(name, address, dest, nb_flags=0, broadcast=True, ttl=0, timeout=0, retries=0) -> (reply_from, name, reply_addr, rcode)\n"
378 "release a previously registered name" },
379 { "name_status", (PyCFunction)py_nbt_name_status, METH_VARARGS|METH_KEYWORDS,
380 "S.name_status(name, dest, timeout=0, retries=0) -> (reply_from, name, status)\n"
381 "Find the status of a name" },
386 PyTypeObject nbt_node_Type = {
387 PyObject_HEAD_INIT(NULL) 0,
388 .tp_name = "netbios.Node",
389 .tp_basicsize = sizeof(nbt_node_Object),
390 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
391 .tp_new = py_nbt_node_init,
392 .tp_dealloc = py_nbt_node_dealloc,
393 .tp_methods = py_nbt_methods,
395 "Create a new NetBIOS node\n"
398 void initnetbios(void)
401 if (PyType_Ready(&nbt_node_Type) < 0)
404 mod = Py_InitModule3("netbios", NULL, "NetBIOS over TCP/IP support");
406 Py_INCREF((PyObject *)&nbt_node_Type);
407 PyModule_AddObject(mod, "Node", (PyObject *)&nbt_node_Type);