source/libsmb/cli_reg.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    RPC Pipe client
   4
   5    Copyright (C) Andrew Tridgell              1992-1998,
   6    Copyright (C) Luke Kenneth Casson Leighton 1996-1998,
   7    Copyright (C) Paul Ashton                  1997-1998.
   8    Copyright (C) Jeremy Allison                    1999.
   9    Copyright (C) Simo Sorce                        2001
  10
  11    This program is free software; you can redistribute it and/or modify
  12    it under the terms of the GNU General Public License as published by
  13    the Free Software Foundation; either version 2 of the License, or
  14    (at your option) any later version.
  15
  16    This program is distributed in the hope that it will be useful,
  17    but WITHOUT ANY WARRANTY; without even the implied warranty of
  18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19    GNU General Public License for more details.
  20
  21    You should have received a copy of the GNU General Public License
  22    along with this program; if not, write to the Free Software
  23    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  24 */
  25
  26 #include "includes.h"
  27
  28 /* Opens a SMB connection to the WINREG pipe */
  29
  30 struct cli_state *cli_winreg_initialise(struct cli_state *cli,
  31                                         char *system_name,
  32                                         struct ntuser_creds *creds)
  33 {
  34         return cli_pipe_initialise(cli, system_name, PIPE_WINREG, creds);
  35 }
  36
  37 /* Shutdown a server */
  38
  39 NTSTATUS cli_reg_shutdown(struct cli_state * cli, TALLOC_CTX *mem_ctx,
  40                           const char *msg, uint32 timeout, uint16 flags)
  41 {
  42         prs_struct qbuf;
  43         prs_struct rbuf;
  44         REG_Q_SHUTDOWN q_s;
  45         REG_R_SHUTDOWN r_s;
  46         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
  47
  48         if (msg == NULL) return NT_STATUS_INVALID_PARAMETER;
  49
  50         ZERO_STRUCT (q_s);
  51         ZERO_STRUCT (r_s);
  52
  53         prs_init(&qbuf , MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
  54         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
  55
  56         /* Marshall data and send request */
  57
  58         init_reg_q_shutdown(&q_s, msg, timeout, flags);
  59
  60         if (!reg_io_q_shutdown("", &q_s, &qbuf, 0) ||
  61             !rpc_api_pipe_req(cli, REG_SHUTDOWN, &qbuf, &rbuf))
  62                 goto done;
  63
  64         /* Unmarshall response */
  65
  66         if(reg_io_r_shutdown("", &r_s, &rbuf, 0))
  67                 result = r_s.status;
  68
  69 done:
  70         prs_mem_free(&rbuf);
  71         prs_mem_free(&qbuf);
  72
  73         return result;
  74 }
  75
  76
  77 /* Abort a server shutdown */
  78
  79 NTSTATUS cli_reg_abort_shutdown(struct cli_state * cli, TALLOC_CTX *mem_ctx)
  80 {
  81         prs_struct rbuf;
  82         prs_struct qbuf;
  83         REG_Q_ABORT_SHUTDOWN q_s;
  84         REG_R_ABORT_SHUTDOWN r_s;
  85         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
  86
  87         ZERO_STRUCT (q_s);
  88         ZERO_STRUCT (r_s);
  89
  90         prs_init(&qbuf , MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
  91         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
  92
  93         /* Marshall data and send request */
  94
  95         init_reg_q_abort_shutdown(&q_s);
  96
  97         if (!reg_io_q_abort_shutdown("", &q_s, &qbuf, 0) ||
  98             !rpc_api_pipe_req(cli, REG_ABORT_SHUTDOWN, &qbuf, &rbuf))
  99                 goto done;
 100
 101                 /* Unmarshall response */
 102
 103         if (reg_io_r_abort_shutdown("", &r_s, &rbuf, 0))
 104                 result = r_s.status;
 105
 106 done:
 107         prs_mem_free(&rbuf);
 108         prs_mem_free(&qbuf );
 109
 110         return result;
 111 }