LCOV - code coverage report
Current view: top level - libcli/smb - smb2cli_write.c (source / functions) Hit Total Coverage
Test: coverage report for abartlet/fix-coverage dd10fb34 Lines: 54 63 85.7 %
Date: 2021-09-23 10:06:22 Functions: 4 4 100.0 %

          Line data    Source code
       1             : /*
       2             :    Unix SMB/CIFS implementation.
       3             :    smb2 lib
       4             :    Copyright (C) Volker Lendecke 2011
       5             : 
       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.
      10             : 
      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.
      15             : 
      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/>.
      18             : */
      19             : 
      20             : #include "includes.h"
      21             : #include "system/network.h"
      22             : #include "lib/util/tevent_ntstatus.h"
      23             : #include "smb_common.h"
      24             : #include "smbXcli_base.h"
      25             : 
      26             : struct smb2cli_write_state {
      27             :         uint8_t fixed[48];
      28             :         uint8_t dyn_pad[1];
      29             :         uint32_t written;
      30             : };
      31             : 
      32             : static void smb2cli_write_done(struct tevent_req *subreq);
      33             : 
      34       10275 : struct tevent_req *smb2cli_write_send(TALLOC_CTX *mem_ctx,
      35             :                                       struct tevent_context *ev,
      36             :                                       struct smbXcli_conn *conn,
      37             :                                       uint32_t timeout_msec,
      38             :                                       struct smbXcli_session *session,
      39             :                                       struct smbXcli_tcon *tcon,
      40             :                                       uint32_t length,
      41             :                                       uint64_t offset,
      42             :                                       uint64_t fid_persistent,
      43             :                                       uint64_t fid_volatile,
      44             :                                       uint32_t remaining_bytes,
      45             :                                       uint32_t flags,
      46             :                                       const uint8_t *data)
      47             : {
      48             :         struct tevent_req *req, *subreq;
      49             :         struct smb2cli_write_state *state;
      50             :         uint8_t *fixed;
      51             :         const uint8_t *dyn;
      52             :         size_t dyn_len;
      53             : 
      54       10275 :         req = tevent_req_create(mem_ctx, &state,
      55             :                                 struct smb2cli_write_state);
      56       10275 :         if (req == NULL) {
      57           0 :                 return NULL;
      58             :         }
      59             : 
      60       10275 :         fixed = state->fixed;
      61             : 
      62       10275 :         SSVAL(fixed, 0, 49);
      63       10275 :         SSVAL(fixed, 2, SMB2_HDR_BODY + 48);
      64       10275 :         SIVAL(fixed, 4, length);
      65       10275 :         SBVAL(fixed, 8, offset);
      66       10275 :         SBVAL(fixed, 16, fid_persistent);
      67       10275 :         SBVAL(fixed, 24, fid_volatile);
      68       10275 :         SIVAL(fixed, 36, remaining_bytes);
      69       10275 :         SIVAL(fixed, 44, flags);
      70             : 
      71       10275 :         if (length > 0) {
      72       10275 :                 dyn = data;
      73       10275 :                 dyn_len = length;
      74             :         } else {
      75           0 :                 dyn = state->dyn_pad;;
      76           0 :                 dyn_len = sizeof(state->dyn_pad);
      77             :         }
      78             : 
      79       17375 :         subreq = smb2cli_req_send(state, ev, conn, SMB2_OP_WRITE,
      80             :                                   0, 0, /* flags */
      81             :                                   timeout_msec,
      82             :                                   tcon,
      83             :                                   session,
      84       10275 :                                   state->fixed, sizeof(state->fixed),
      85             :                                   dyn, dyn_len,
      86             :                                   0); /* max_dyn_len */
      87       10275 :         if (tevent_req_nomem(subreq, req)) {
      88           0 :                 return tevent_req_post(req, ev);
      89             :         }
      90       10275 :         tevent_req_set_callback(subreq, smb2cli_write_done, req);
      91       10275 :         return req;
      92             : }
      93             : 
      94       10275 : static void smb2cli_write_done(struct tevent_req *subreq)
      95             : {
      96        7100 :         struct tevent_req *req =
      97       10275 :                 tevent_req_callback_data(subreq,
      98             :                 struct tevent_req);
      99        7100 :         struct smb2cli_write_state *state =
     100       10275 :                 tevent_req_data(req,
     101             :                 struct smb2cli_write_state);
     102             :         NTSTATUS status;
     103             :         struct iovec *iov;
     104             :         static const struct smb2cli_req_expected_response expected[] = {
     105             :         {
     106             :                 .status = NT_STATUS_OK,
     107             :                 .body_size = 0x11
     108             :         }
     109             :         };
     110             : 
     111       10275 :         status = smb2cli_req_recv(subreq, state, &iov,
     112             :                                   expected, ARRAY_SIZE(expected));
     113       10275 :         TALLOC_FREE(subreq);
     114       10275 :         if (tevent_req_nterror(req, status)) {
     115           8 :                 return;
     116             :         }
     117       10267 :         state->written = IVAL(iov[1].iov_base, 4);
     118       10267 :         tevent_req_done(req);
     119             : }
     120             : 
     121       10275 : NTSTATUS smb2cli_write_recv(struct tevent_req *req, uint32_t *written)
     122             : {
     123        7100 :         struct smb2cli_write_state *state =
     124       10275 :                 tevent_req_data(req,
     125             :                 struct smb2cli_write_state);
     126             :         NTSTATUS status;
     127             : 
     128       10275 :         if (tevent_req_is_nterror(req, &status)) {
     129           8 :                 tevent_req_received(req);
     130           8 :                 return status;
     131             :         }
     132       10267 :         if (written) {
     133       10242 :                 *written = state->written;
     134             :         }
     135       10267 :         tevent_req_received(req);
     136       10267 :         return NT_STATUS_OK;
     137             : }
     138             : 
     139          33 : NTSTATUS smb2cli_write(struct smbXcli_conn *conn,
     140             :                        uint32_t timeout_msec,
     141             :                        struct smbXcli_session *session,
     142             :                        struct smbXcli_tcon *tcon,
     143             :                        uint32_t length,
     144             :                        uint64_t offset,
     145             :                        uint64_t fid_persistent,
     146             :                        uint64_t fid_volatile,
     147             :                        uint32_t remaining_bytes,
     148             :                        uint32_t flags,
     149             :                        const uint8_t *data,
     150             :                        uint32_t *written)
     151             : {
     152          33 :         TALLOC_CTX *frame = talloc_stackframe();
     153             :         struct tevent_context *ev;
     154             :         struct tevent_req *req;
     155          33 :         NTSTATUS status = NT_STATUS_NO_MEMORY;
     156             : 
     157          33 :         if (smbXcli_conn_has_async_calls(conn)) {
     158             :                 /*
     159             :                  * Can't use sync call while an async call is in flight
     160             :                  */
     161           0 :                 status = NT_STATUS_INVALID_PARAMETER;
     162           0 :                 goto fail;
     163             :         }
     164          33 :         ev = samba_tevent_context_init(frame);
     165          33 :         if (ev == NULL) {
     166           0 :                 goto fail;
     167             :         }
     168          33 :         req = smb2cli_write_send(frame, ev, conn, timeout_msec,
     169             :                                  session, tcon,
     170             :                                  length, offset,
     171             :                                  fid_persistent, fid_volatile,
     172             :                                  remaining_bytes, flags, data);
     173          33 :         if (req == NULL) {
     174           0 :                 goto fail;
     175             :         }
     176          33 :         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
     177           0 :                 goto fail;
     178             :         }
     179          33 :         status = smb2cli_write_recv(req, written);
     180          33 :  fail:
     181          33 :         TALLOC_FREE(frame);
     182          33 :         return status;
     183             : }

Generated by: LCOV version 1.13