source/rpc_client/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 /* Shutdown a server */
  29
  30 NTSTATUS cli_reg_shutdown(struct cli_state * cli, TALLOC_CTX *mem_ctx,
  31                           const char *msg, uint32 timeout, uint16 flags)
  32 {
  33         prs_struct qbuf;
  34         prs_struct rbuf;
  35         REG_Q_SHUTDOWN q_s;
  36         REG_R_SHUTDOWN r_s;
  37         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
  38
  39         if (msg == NULL) return NT_STATUS_INVALID_PARAMETER;
  40
  41         ZERO_STRUCT (q_s);
  42         ZERO_STRUCT (r_s);
  43
  44         prs_init(&qbuf , MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
  45         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
  46
  47         /* Marshall data and send request */
  48
  49         init_reg_q_shutdown(&q_s, msg, timeout, flags);
  50
  51         if (!reg_io_q_shutdown("", &q_s, &qbuf, 0) ||
  52             !rpc_api_pipe_req(cli, REG_SHUTDOWN, &qbuf, &rbuf))
  53                 goto done;
  54
  55         /* Unmarshall response */
  56
  57         if(reg_io_r_shutdown("", &r_s, &rbuf, 0))
  58                 result = r_s.status;
  59
  60 done:
  61         prs_mem_free(&rbuf);
  62         prs_mem_free(&qbuf);
  63
  64         return result;
  65 }
  66
  67
  68 /* Abort a server shutdown */
  69
  70 NTSTATUS cli_reg_abort_shutdown(struct cli_state * cli, TALLOC_CTX *mem_ctx)
  71 {
  72         prs_struct rbuf;
  73         prs_struct qbuf;
  74         REG_Q_ABORT_SHUTDOWN q_s;
  75         REG_R_ABORT_SHUTDOWN r_s;
  76         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
  77
  78         ZERO_STRUCT (q_s);
  79         ZERO_STRUCT (r_s);
  80
  81         prs_init(&qbuf , MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
  82         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
  83
  84         /* Marshall data and send request */
  85
  86         init_reg_q_abort_shutdown(&q_s);
  87
  88         if (!reg_io_q_abort_shutdown("", &q_s, &qbuf, 0) ||
  89             !rpc_api_pipe_req(cli, REG_ABORT_SHUTDOWN, &qbuf, &rbuf))
  90                 goto done;
  91
  92                 /* Unmarshall response */
  93
  94         if (reg_io_r_abort_shutdown("", &r_s, &rbuf, 0))
  95                 result = r_s.status;
  96
  97 done:
  98         prs_mem_free(&rbuf);
  99         prs_mem_free(&qbuf );
 100
 101         return result;
 102 }