s3:rpcclient: Use C99 initializer for cmd_set in cmd_eventlog
[vlendec/samba-autobuild/.git] / source3 / rpcclient / cmd_eventlog.c
index e212452e5d64634f7c757303a8c7469aa0c5537b..a22ab8a18ff3805235206a684bae53c4ad69cd69 100644 (file)
 
 #include "includes.h"
 #include "rpcclient.h"
+#include "../librpc/gen_ndr/ndr_eventlog.h"
+#include "../librpc/gen_ndr/ndr_eventlog_c.h"
+#include "rpc_client/init_lsa.h"
 
 static NTSTATUS get_eventlog_handle(struct rpc_pipe_client *cli,
                                    TALLOC_CTX *mem_ctx,
                                    const char *log,
                                    struct policy_handle *handle)
 {
-       NTSTATUS status;
+       NTSTATUS status, result;
        struct eventlog_OpenUnknown0 unknown0;
        struct lsa_String logname, servername;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        unknown0.unknown0 = 0x005c;
        unknown0.unknown1 = 0x0001;
@@ -36,18 +40,19 @@ static NTSTATUS get_eventlog_handle(struct rpc_pipe_client *cli,
        init_lsa_String(&logname, log);
        init_lsa_String(&servername, NULL);
 
-       status = rpccli_eventlog_OpenEventLogW(cli, mem_ctx,
+       status = dcerpc_eventlog_OpenEventLogW(b, mem_ctx,
                                               &unknown0,
                                               &logname,
                                               &servername,
                                               0x00000001, /* major */
                                               0x00000001, /* minor */
-                                              handle);
+                                              handle,
+                                              &result);
        if (!NT_STATUS_IS_OK(status)) {
                return status;
        }
 
-       return NT_STATUS_OK;
+       return result;
 }
 
 static NTSTATUS cmd_eventlog_readlog(struct rpc_pipe_client *cli,
@@ -56,13 +61,15 @@ static NTSTATUS cmd_eventlog_readlog(struct rpc_pipe_client *cli,
                                     const char **argv)
 {
        NTSTATUS status = NT_STATUS_OK;
+       NTSTATUS result = NT_STATUS_OK;
        struct policy_handle handle;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        uint32_t flags = EVENTLOG_BACKWARDS_READ |
                         EVENTLOG_SEQUENTIAL_READ;
        uint32_t offset = 0;
        uint32_t number_of_bytes = 0;
-       uint8_t *data = NULL;
+       uint8_t *data;
        uint32_t sent_size = 0;
        uint32_t real_size = 0;
 
@@ -77,10 +84,6 @@ static NTSTATUS cmd_eventlog_readlog(struct rpc_pipe_client *cli,
 
        if (argc >= 4) {
                number_of_bytes = atoi(argv[3]);
-               data = talloc_array(mem_ctx, uint8_t, number_of_bytes);
-               if (!data) {
-                       goto done;
-               }
        }
 
        status = get_eventlog_handle(cli, mem_ctx, argv[1], &handle);
@@ -88,6 +91,11 @@ static NTSTATUS cmd_eventlog_readlog(struct rpc_pipe_client *cli,
                return status;
        }
 
+       data = talloc_array(mem_ctx, uint8_t, number_of_bytes);
+       if (data == NULL) {
+               goto done;
+       }
+
        do {
 
                enum ndr_err_code ndr_err;
@@ -96,33 +104,41 @@ static NTSTATUS cmd_eventlog_readlog(struct rpc_pipe_client *cli,
                uint32_t size = 0;
                uint32_t pos = 0;
 
-               status = rpccli_eventlog_ReadEventLogW(cli, mem_ctx,
+               status = dcerpc_eventlog_ReadEventLogW(b, mem_ctx,
                                                       &handle,
                                                       flags,
                                                       offset,
                                                       number_of_bytes,
                                                       data,
                                                       &sent_size,
-                                                      &real_size);
-               if (NT_STATUS_EQUAL(status, NT_STATUS_BUFFER_TOO_SMALL) &&
+                                                      &real_size,
+                                                      &result);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+               if (NT_STATUS_EQUAL(result, NT_STATUS_BUFFER_TOO_SMALL) &&
                    real_size > 0 ) {
                        number_of_bytes = real_size;
-                       data = talloc_array(mem_ctx, uint8_t, real_size);
-                       if (!data) {
+                       data = talloc_realloc(mem_ctx, data, uint8_t, real_size);
+                       if (data == NULL) {
                                goto done;
                        }
-                       status = rpccli_eventlog_ReadEventLogW(cli, mem_ctx,
+                       status = dcerpc_eventlog_ReadEventLogW(b, mem_ctx,
                                                               &handle,
                                                               flags,
                                                               offset,
                                                               number_of_bytes,
                                                               data,
                                                               &sent_size,
-                                                              &real_size);
+                                                              &real_size,
+                                                              &result);
+                       if (!NT_STATUS_IS_OK(status)) {
+                               return status;
+                       }
                }
 
-               if (!NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE) &&
-                   !NT_STATUS_IS_OK(status)) {
+               if (!NT_STATUS_EQUAL(result, NT_STATUS_END_OF_FILE) &&
+                   !NT_STATUS_IS_OK(result)) {
                        goto done;
                }
 
@@ -134,7 +150,7 @@ static NTSTATUS cmd_eventlog_readlog(struct rpc_pipe_client *cli,
 
                        blob = data_blob_const(data + pos, size);
                        /* dump_data(0, blob.data, blob.length); */
-                       ndr_err = ndr_pull_struct_blob_all(&blob, mem_ctx, NULL, &r,
+                       ndr_err = ndr_pull_struct_blob_all(&blob, mem_ctx, &r,
                                           (ndr_pull_flags_fn_t)ndr_pull_EVENTLOGRECORD);
                        if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
                                status = ndr_map_error2ntstatus(ndr_err);
@@ -154,10 +170,10 @@ static NTSTATUS cmd_eventlog_readlog(struct rpc_pipe_client *cli,
 
                offset++;
 
-       } while (NT_STATUS_IS_OK(status));
+       } while (NT_STATUS_IS_OK(result));
 
  done:
-       rpccli_eventlog_CloseEventLog(cli, mem_ctx, &handle);
+       dcerpc_eventlog_CloseEventLog(b, mem_ctx, &handle, &result);
 
        return status;
 }
@@ -167,9 +183,10 @@ static NTSTATUS cmd_eventlog_numrecords(struct rpc_pipe_client *cli,
                                        int argc,
                                        const char **argv)
 {
-       NTSTATUS status;
+       NTSTATUS status, result;
        struct policy_handle handle;
        uint32_t number = 0;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        if (argc != 2) {
                printf("Usage: %s logname\n", argv[0]);
@@ -181,17 +198,22 @@ static NTSTATUS cmd_eventlog_numrecords(struct rpc_pipe_client *cli,
                return status;
        }
 
-       status = rpccli_eventlog_GetNumRecords(cli, mem_ctx,
+       status = dcerpc_eventlog_GetNumRecords(b, mem_ctx,
                                               &handle,
-                                              &number);
+                                              &number,
+                                              &result);
        if (!NT_STATUS_IS_OK(status)) {
                goto done;
        }
+       if (!NT_STATUS_IS_OK(result)) {
+               status = result;
+               goto done;
+       }
 
        printf("number of records: %d\n", number);
 
  done:
-       rpccli_eventlog_CloseEventLog(cli, mem_ctx, &handle);
+       dcerpc_eventlog_CloseEventLog(b, mem_ctx, &handle, &result);
 
        return status;
 }
@@ -201,9 +223,10 @@ static NTSTATUS cmd_eventlog_oldestrecord(struct rpc_pipe_client *cli,
                                          int argc,
                                          const char **argv)
 {
-       NTSTATUS status;
+       NTSTATUS status, result;
        struct policy_handle handle;
        uint32_t oldest_entry = 0;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        if (argc != 2) {
                printf("Usage: %s logname\n", argv[0]);
@@ -215,17 +238,22 @@ static NTSTATUS cmd_eventlog_oldestrecord(struct rpc_pipe_client *cli,
                return status;
        }
 
-       status = rpccli_eventlog_GetOldestRecord(cli, mem_ctx,
+       status = dcerpc_eventlog_GetOldestRecord(b, mem_ctx,
                                                 &handle,
-                                                &oldest_entry);
+                                                &oldest_entry,
+                                                &result);
        if (!NT_STATUS_IS_OK(status)) {
                goto done;
        }
+       if (!NT_STATUS_IS_OK(result)) {
+               status = result;
+               goto done;
+       }
 
        printf("oldest entry: %d\n", oldest_entry);
 
  done:
-       rpccli_eventlog_CloseEventLog(cli, mem_ctx, &handle);
+       dcerpc_eventlog_CloseEventLog(b, mem_ctx, &handle, &result);
 
        return status;
 }
@@ -235,8 +263,9 @@ static NTSTATUS cmd_eventlog_reportevent(struct rpc_pipe_client *cli,
                                         int argc,
                                         const char **argv)
 {
-       NTSTATUS status;
+       NTSTATUS status, result;
        struct policy_handle handle;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        uint16_t num_of_strings = 1;
        uint32_t data_size = 0;
@@ -264,7 +293,7 @@ static NTSTATUS cmd_eventlog_reportevent(struct rpc_pipe_client *cli,
        init_lsa_String(&strings[0], "test event written by rpcclient\n");
        init_lsa_String(&servername, NULL);
 
-       status = rpccli_eventlog_ReportEventW(cli, mem_ctx,
+       status = dcerpc_eventlog_ReportEventW(b, mem_ctx,
                                              &handle,
                                              time(NULL),
                                              EVENTLOG_INFORMATION_TYPE,
@@ -278,17 +307,22 @@ static NTSTATUS cmd_eventlog_reportevent(struct rpc_pipe_client *cli,
                                              data,
                                              0, /* flags */
                                              &record_number,
-                                             &time_written);
+                                             &time_written,
+                                             &result);
 
        if (!NT_STATUS_IS_OK(status)) {
                goto done;
        }
+       if (!NT_STATUS_IS_OK(result)) {
+               status = result;
+               goto done;
+       }
 
        printf("entry: %d written at %s\n", record_number,
                http_timestring(talloc_tos(), time_written));
 
  done:
-       rpccli_eventlog_CloseEventLog(cli, mem_ctx, &handle);
+       dcerpc_eventlog_CloseEventLog(b, mem_ctx, &handle, &result);
 
        return status;
 }
@@ -298,8 +332,9 @@ static NTSTATUS cmd_eventlog_reporteventsource(struct rpc_pipe_client *cli,
                                               int argc,
                                               const char **argv)
 {
-       NTSTATUS status;
+       NTSTATUS status, result;
        struct policy_handle handle;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        uint16_t num_of_strings = 1;
        uint32_t data_size = 0;
@@ -328,7 +363,7 @@ static NTSTATUS cmd_eventlog_reporteventsource(struct rpc_pipe_client *cli,
        init_lsa_String(&servername, NULL);
        init_lsa_String(&sourcename, "rpcclient");
 
-       status = rpccli_eventlog_ReportEventAndSourceW(cli, mem_ctx,
+       status = dcerpc_eventlog_ReportEventAndSourceW(b, mem_ctx,
                                                       &handle,
                                                       time(NULL),
                                                       EVENTLOG_INFORMATION_TYPE,
@@ -343,16 +378,21 @@ static NTSTATUS cmd_eventlog_reporteventsource(struct rpc_pipe_client *cli,
                                                       data,
                                                       0, /* flags */
                                                       &record_number,
-                                                      &time_written);
+                                                      &time_written,
+                                                      &result);
        if (!NT_STATUS_IS_OK(status)) {
                goto done;
        }
+       if (!NT_STATUS_IS_OK(result)) {
+               status = result;
+               goto done;
+       }
 
        printf("entry: %d written at %s\n", record_number,
                http_timestring(talloc_tos(), time_written));
 
  done:
-       rpccli_eventlog_CloseEventLog(cli, mem_ctx, &handle);
+       dcerpc_eventlog_CloseEventLog(b, mem_ctx, &handle, &result);
 
        return status;
 }
@@ -362,10 +402,11 @@ static NTSTATUS cmd_eventlog_registerevsource(struct rpc_pipe_client *cli,
                                              int argc,
                                              const char **argv)
 {
-       NTSTATUS status;
+       NTSTATUS status, result;
        struct policy_handle log_handle;
        struct lsa_String module_name, reg_module_name;
        struct eventlog_OpenUnknown0 unknown0;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        unknown0.unknown0 = 0x005c;
        unknown0.unknown1 = 0x0001;
@@ -378,19 +419,24 @@ static NTSTATUS cmd_eventlog_registerevsource(struct rpc_pipe_client *cli,
        init_lsa_String(&module_name, "rpcclient");
        init_lsa_String(&reg_module_name, NULL);
 
-       status = rpccli_eventlog_RegisterEventSourceW(cli, mem_ctx,
+       status = dcerpc_eventlog_RegisterEventSourceW(b, mem_ctx,
                                                      &unknown0,
                                                      &module_name,
                                                      &reg_module_name,
                                                      1, /* major_version */
                                                      1, /* minor_version */
-                                                     &log_handle);
+                                                     &log_handle,
+                                                     &result);
        if (!NT_STATUS_IS_OK(status)) {
                goto done;
        }
+       if (!NT_STATUS_IS_OK(result)) {
+               status = result;
+               goto done;
+       }
 
  done:
-       rpccli_eventlog_DeregisterEventSource(cli, mem_ctx, &log_handle);
+       dcerpc_eventlog_DeregisterEventSource(b, mem_ctx, &log_handle, &result);
 
        return status;
 }
@@ -400,10 +446,11 @@ static NTSTATUS cmd_eventlog_backuplog(struct rpc_pipe_client *cli,
                                       int argc,
                                       const char **argv)
 {
-       NTSTATUS status;
+       NTSTATUS status, result;
        struct policy_handle handle;
        struct lsa_String backup_filename;
        const char *tmp;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        if (argc != 3) {
                printf("Usage: %s logname backupname\n", argv[0]);
@@ -423,12 +470,20 @@ static NTSTATUS cmd_eventlog_backuplog(struct rpc_pipe_client *cli,
 
        init_lsa_String(&backup_filename, tmp);
 
-       status = rpccli_eventlog_BackupEventLogW(cli, mem_ctx,
+       status = dcerpc_eventlog_BackupEventLogW(b, mem_ctx,
                                                 &handle,
-                                                &backup_filename);
+                                                &backup_filename,
+                                                &result);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
+       }
+       if (!NT_STATUS_IS_OK(result)) {
+               status = result;
+               goto done;
+       }
 
  done:
-       rpccli_eventlog_CloseEventLog(cli, mem_ctx, &handle);
+       dcerpc_eventlog_CloseEventLog(b, mem_ctx, &handle, &result);
 
        return status;
 }
@@ -438,11 +493,12 @@ static NTSTATUS cmd_eventlog_loginfo(struct rpc_pipe_client *cli,
                                     int argc,
                                     const char **argv)
 {
-       NTSTATUS status;
+       NTSTATUS status, result;
        struct policy_handle handle;
        uint8_t *buffer = NULL;
        uint32_t buf_size = 0;
        uint32_t bytes_needed = 0;
+       struct dcerpc_binding_handle *b = cli->binding_handle;
 
        if (argc != 2) {
                printf("Usage: %s logname\n", argv[0]);
@@ -454,50 +510,141 @@ static NTSTATUS cmd_eventlog_loginfo(struct rpc_pipe_client *cli,
                return status;
        }
 
-       status = rpccli_eventlog_GetLogIntormation(cli, mem_ctx,
+       buffer = talloc_array(mem_ctx, uint8_t, bytes_needed);
+       if (buffer == NULL) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+
+       status = dcerpc_eventlog_GetLogInformation(b, mem_ctx,
                                                   &handle,
                                                   0, /* level */
                                                   buffer,
                                                   buf_size,
-                                                  &bytes_needed);
-       if (!NT_STATUS_IS_OK(status) &&
-           !NT_STATUS_EQUAL(status, NT_STATUS_BUFFER_TOO_SMALL)) {
+                                                  &bytes_needed,
+                                                  &result);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
+       }
+       if (!NT_STATUS_IS_OK(result) &&
+           !NT_STATUS_EQUAL(result, NT_STATUS_BUFFER_TOO_SMALL)) {
                goto done;
        }
 
        buf_size = bytes_needed;
-       buffer = talloc_array(mem_ctx, uint8_t, bytes_needed);
-       if (!buffer) {
+       buffer = talloc_realloc(mem_ctx, buffer, uint8_t, bytes_needed);
+       if (buffer == NULL) {
                status = NT_STATUS_NO_MEMORY;
                goto done;
        }
 
-       status = rpccli_eventlog_GetLogIntormation(cli, mem_ctx,
+       status = dcerpc_eventlog_GetLogInformation(b, mem_ctx,
                                                   &handle,
                                                   0, /* level */
                                                   buffer,
                                                   buf_size,
-                                                  &bytes_needed);
+                                                  &bytes_needed,
+                                                  &result);
        if (!NT_STATUS_IS_OK(status)) {
                goto done;
        }
+       if (!NT_STATUS_IS_OK(result)) {
+               status = result;
+               goto done;
+       }
 
  done:
-       rpccli_eventlog_CloseEventLog(cli, mem_ctx, &handle);
+       dcerpc_eventlog_CloseEventLog(b, mem_ctx, &handle, &result);
 
        return status;
 }
 
 
 struct cmd_set eventlog_commands[] = {
-       { "EVENTLOG" },
-       { "eventlog_readlog",           RPC_RTYPE_NTSTATUS,     cmd_eventlog_readlog,           NULL,   &ndr_table_eventlog.syntax_id,  NULL,   "Read Eventlog", "" },
-       { "eventlog_numrecord",         RPC_RTYPE_NTSTATUS,     cmd_eventlog_numrecords,        NULL,   &ndr_table_eventlog.syntax_id,  NULL,   "Get number of records", "" },
-       { "eventlog_oldestrecord",      RPC_RTYPE_NTSTATUS,     cmd_eventlog_oldestrecord,      NULL,   &ndr_table_eventlog.syntax_id,  NULL,   "Get oldest record", "" },
-       { "eventlog_reportevent",       RPC_RTYPE_NTSTATUS,     cmd_eventlog_reportevent,       NULL,   &ndr_table_eventlog.syntax_id,  NULL,   "Report event", "" },
-       { "eventlog_reporteventsource", RPC_RTYPE_NTSTATUS,     cmd_eventlog_reporteventsource, NULL,   &ndr_table_eventlog.syntax_id,  NULL,   "Report event and source", "" },
-       { "eventlog_registerevsource",  RPC_RTYPE_NTSTATUS,     cmd_eventlog_registerevsource,  NULL,   &ndr_table_eventlog.syntax_id,  NULL,   "Register event source", "" },
-       { "eventlog_backuplog",         RPC_RTYPE_NTSTATUS,     cmd_eventlog_backuplog,         NULL,   &ndr_table_eventlog.syntax_id,  NULL,   "Backup Eventlog File", "" },
-       { "eventlog_loginfo",           RPC_RTYPE_NTSTATUS,     cmd_eventlog_loginfo,           NULL,   &ndr_table_eventlog.syntax_id,  NULL,   "Get Eventlog Information", "" },
-       { NULL }
+       {
+               .name = "EVENTLOG",
+       },
+       {
+               .name               = "eventlog_readlog",
+               .returntype         = RPC_RTYPE_NTSTATUS,
+               .ntfn               = cmd_eventlog_readlog,
+               .wfn                = NULL,
+               .table              = &ndr_table_eventlog,
+               .rpc_pipe           = NULL,
+               .description        = "Read Eventlog",
+               .usage              = "",
+       },
+       {
+               .name               = "eventlog_numrecord",
+               .returntype         = RPC_RTYPE_NTSTATUS,
+               .ntfn               = cmd_eventlog_numrecords,
+               .wfn                = NULL,
+               .table              = &ndr_table_eventlog,
+               .rpc_pipe           = NULL,
+               .description        = "Get number of records",
+               .usage              = "",
+       },
+       {
+               .name               = "eventlog_oldestrecord",
+               .returntype         = RPC_RTYPE_NTSTATUS,
+               .ntfn               = cmd_eventlog_oldestrecord,
+               .wfn                = NULL,
+               .table              = &ndr_table_eventlog,
+               .rpc_pipe           = NULL,
+               .description        = "Get oldest record",
+               .usage              = "",
+       },
+       {
+               .name               = "eventlog_reportevent",
+               .returntype         = RPC_RTYPE_NTSTATUS,
+               .ntfn               = cmd_eventlog_reportevent,
+               .wfn                = NULL,
+               .table              = &ndr_table_eventlog,
+               .rpc_pipe           = NULL,
+               .description        = "Report event",
+               .usage              = "",
+       },
+       {
+               .name               = "eventlog_reporteventsource",
+               .returntype         = RPC_RTYPE_NTSTATUS,
+               .ntfn               = cmd_eventlog_reporteventsource,
+               .wfn                = NULL,
+               .table              = &ndr_table_eventlog,
+               .rpc_pipe           = NULL,
+               .description        = "Report event and source",
+               .usage              = "",
+       },
+       {
+               .name               = "eventlog_registerevsource",
+               .returntype         = RPC_RTYPE_NTSTATUS,
+               .ntfn               = cmd_eventlog_registerevsource,
+               .wfn                = NULL,
+               .table              = &ndr_table_eventlog,
+               .rpc_pipe           = NULL,
+               .description        = "Register event source",
+               .usage              = "",
+       },
+       {
+               .name               = "eventlog_backuplog",
+               .returntype         = RPC_RTYPE_NTSTATUS,
+               .ntfn               = cmd_eventlog_backuplog,
+               .wfn                = NULL,
+               .table              = &ndr_table_eventlog,
+               .rpc_pipe           = NULL,
+               .description        = "Backup Eventlog File",
+               .usage              = "",
+       },
+       {
+               .name               = "eventlog_loginfo",
+               .returntype         = RPC_RTYPE_NTSTATUS,
+               .ntfn               = cmd_eventlog_loginfo,
+               .wfn                = NULL,
+               .table              = &ndr_table_eventlog,
+               .rpc_pipe           = NULL,
+               .description        = "Get Eventlog Information",
+               .usage              = "",
+       },
+       {
+               .name = NULL,
+       },
 };