lib/util/tevent_ntstatus.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Wrap unix errno around tevent_req
   4    Copyright (C) Volker Lendecke 2009
   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 "../replace/replace.h"
  21 #include "tevent_ntstatus.h"
  22 #include "libcli/util/error.h"
  23
  24 #define TEVENT_NTERROR_MAGIC (0x917b5acd)
  25
  26 bool _tevent_req_nterror(struct tevent_req *req,
  27                          NTSTATUS status,
  28                          const char *location)
  29 {
  30         uint64_t err;
  31
  32         if (NT_STATUS_IS_OK(status)) {
  33                 return false;
  34         }
  35
  36         /*
  37          * I've put this variable here, because I'm not 100% certain
  38          * how to correctly assign a 64-bit constant and left-shift it
  39          * by 32 bits in a single expression. If anyone knows, feel
  40          * free :-)
  41          */
  42         err = TEVENT_NTERROR_MAGIC;
  43         err <<= 32;
  44         err |= NT_STATUS_V(status);
  45
  46         return _tevent_req_error(req, err, location);
  47 }
  48
  49 bool tevent_req_is_nterror(struct tevent_req *req, NTSTATUS *status)
  50 {
  51         enum tevent_req_state state;
  52         uint64_t err;
  53
  54         if (!tevent_req_is_error(req, &state, &err)) {
  55                 return false;
  56         }
  57         switch (state) {
  58         case TEVENT_REQ_TIMED_OUT:
  59                 *status = NT_STATUS_IO_TIMEOUT;
  60                 break;
  61         case TEVENT_REQ_NO_MEMORY:
  62                 *status = NT_STATUS_NO_MEMORY;
  63                 break;
  64         case TEVENT_REQ_USER_ERROR:
  65                 if ((err >> 32) != TEVENT_NTERROR_MAGIC) {
  66                         abort();
  67                 }
  68                 *status = NT_STATUS(err & 0xffffffff);
  69                 break;
  70         default:
  71                 *status = NT_STATUS_INTERNAL_ERROR;
  72                 break;
  73         }
  74         return true;
  75 }
  76
  77 NTSTATUS tevent_req_simple_recv_ntstatus(struct tevent_req *req)
  78 {
  79         NTSTATUS status;
  80
  81         if (tevent_req_is_nterror(req, &status)) {
  82                 return status;
  83         }
  84         return NT_STATUS_OK;
  85 }
  86
  87 void tevent_req_simple_finish_ntstatus(struct tevent_req *subreq,
  88                                        NTSTATUS subreq_status)
  89 {
  90         struct tevent_req *req = tevent_req_callback_data(
  91                 subreq, struct tevent_req);
  92
  93         TALLOC_FREE(subreq);
  94
  95         if (!NT_STATUS_IS_OK(subreq_status)) {
  96                 tevent_req_nterror(req, subreq_status);
  97                 return;
  98         }
  99         tevent_req_done(req);
 100 }
 101
 102 bool tevent_req_poll_ntstatus(struct tevent_req *req,
 103                               struct tevent_context *ev,
 104                               NTSTATUS *status)
 105 {
 106         bool ret = tevent_req_poll(req, ev);
 107         if (!ret) {
 108                 *status = map_nt_error_from_unix_common(errno);
 109         }
 110         return ret;
 111 }