pylibsmb: Add py_cli_qfileinfo

Signed-off-by: Volker Lendecke <vl@samba.org>
Reviewed-by: Jeremy Allison <jra@samba.org>

diff --git a/source3/libsmb/pylibsmb.c b/source3/libsmb/pylibsmb.c
index aab17474e549b1c3d6adca03b897abecedd02e41..091cfee76bd64d072e7c9dff40e008633d86d88d 100644 (file)
--- a/source3/libsmb/pylibsmb.c
+++ b/source3/libsmb/pylibsmb.c
@@ -1263,6 +1263,61 @@ static PyObject *py_cli_close(struct py_cli_state *self, PyObject *args)
        Py_RETURN_NONE;
 }
 
+static PyObject *py_cli_qfileinfo(struct py_cli_state *self, PyObject *args)
+{
+       struct tevent_req *req = NULL;
+       int fnum, level;
+       uint16_t recv_flags2;
+       uint8_t *rdata = NULL;
+       uint32_t num_rdata;
+       PyObject *result = NULL;
+       NTSTATUS status;
+
+       if (!PyArg_ParseTuple(args, "ii", &fnum, &level)) {
+               return NULL;
+       }
+
+       req = cli_qfileinfo_send(
+               NULL, self->ev, self->cli, fnum, level, 0, UINT32_MAX);
+       if (!py_tevent_req_wait_exc(self, req)) {
+               return NULL;
+       }
+       status = cli_qfileinfo_recv(
+               req, NULL, &recv_flags2, &rdata, &num_rdata);
+       TALLOC_FREE(req);
+
+       if (!NT_STATUS_IS_OK(status)) {
+               PyErr_SetNTSTATUS(status);
+               return NULL;
+       }
+
+       switch (level) {
+       case FSCC_FILE_ATTRIBUTE_TAG_INFORMATION: {
+               uint32_t mode = PULL_LE_U32(rdata, 0);
+               uint32_t tag = PULL_LE_U32(rdata, 4);
+
+               if (num_rdata != 8) {
+                       PyErr_SetNTSTATUS(NT_STATUS_INVALID_NETWORK_RESPONSE);
+                       return NULL;
+               }
+
+               result = Py_BuildValue("{s:K,s:K}",
+                                      "mode",
+                                      (unsigned long long)mode,
+                                      "tag",
+                                      (unsigned long long)tag);
+               break;
+       }
+       default:
+               result = PyBytes_FromStringAndSize((char *)rdata, num_rdata);
+               break;
+       }
+
+       TALLOC_FREE(rdata);
+
+       return result;
+}
+
 static PyObject *py_cli_rename(
        struct py_cli_state *self, PyObject *args, PyObject *kwds)
 {
@@ -2787,6 +2842,12 @@ static PyMethodDef py_cli_state_methods[] = {
          METH_VARARGS|METH_KEYWORDS,
          "fsctl(fnum, ctl_code, in_bytes, max_out) -> out_bytes",
        },
+       {
+               "qfileinfo",
+               (PyCFunction)py_cli_qfileinfo,
+               METH_VARARGS | METH_KEYWORDS,
+               "qfileinfo(fnum, level) -> blob",
+       },
        { "mknod",
          PY_DISCARD_FUNC_SIG(PyCFunction, py_cli_mknod),
          METH_VARARGS|METH_KEYWORDS,