| 1 | /*
 | 
|---|
| 2 |    Unix SMB/Netbios implementation.
 | 
|---|
| 3 |    Version 3.0
 | 
|---|
| 4 |    async_io read handling using POSIX async io.
 | 
|---|
| 5 |    Copyright (C) Jeremy Allison 2005.
 | 
|---|
| 6 | 
 | 
|---|
| 7 |    This program is free software; you can redistribute it and/or modify
 | 
|---|
| 8 |    it under the terms of the GNU General Public License as published by
 | 
|---|
| 9 |    the Free Software Foundation; either version 3 of the License, or
 | 
|---|
| 10 |    (at your option) any later version.
 | 
|---|
| 11 | 
 | 
|---|
| 12 |    This program is distributed in the hope that it will be useful,
 | 
|---|
| 13 |    but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
|---|
| 14 |    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
|---|
| 15 |    GNU General Public License for more details.
 | 
|---|
| 16 | 
 | 
|---|
| 17 |    You should have received a copy of the GNU General Public License
 | 
|---|
| 18 |    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | 
|---|
| 19 | */
 | 
|---|
| 20 | 
 | 
|---|
| 21 | #include "includes.h"
 | 
|---|
| 22 | #include "smbd/globals.h"
 | 
|---|
| 23 | 
 | 
|---|
| 24 | #if defined(WITH_AIO)
 | 
|---|
| 25 | 
 | 
|---|
| 26 | /* The signal we'll use to signify aio done. */
 | 
|---|
| 27 | #ifndef RT_SIGNAL_AIO
 | 
|---|
| 28 | #define RT_SIGNAL_AIO   (SIGRTMIN+3)
 | 
|---|
| 29 | #endif
 | 
|---|
| 30 | 
 | 
|---|
| 31 | #ifndef HAVE_STRUCT_SIGEVENT_SIGEV_VALUE_SIVAL_PTR
 | 
|---|
| 32 | #ifdef HAVE_STRUCT_SIGEVENT_SIGEV_VALUE_SIGVAL_PTR
 | 
|---|
| 33 | #define sival_int       sigval_int
 | 
|---|
| 34 | #define sival_ptr       sigval_ptr
 | 
|---|
| 35 | #endif
 | 
|---|
| 36 | #endif
 | 
|---|
| 37 | 
 | 
|---|
| 38 | /****************************************************************************
 | 
|---|
| 39 |  The buffer we keep around whilst an aio request is in process.
 | 
|---|
| 40 | *****************************************************************************/
 | 
|---|
| 41 | 
 | 
|---|
| 42 | struct aio_extra {
 | 
|---|
| 43 |         struct aio_extra *next, *prev;
 | 
|---|
| 44 |         SMB_STRUCT_AIOCB acb;
 | 
|---|
| 45 |         files_struct *fsp;
 | 
|---|
| 46 |         struct smb_request *req;
 | 
|---|
| 47 |         char *outbuf;
 | 
|---|
| 48 |         int (*handle_completion)(struct aio_extra *ex, int errcode);
 | 
|---|
| 49 | };
 | 
|---|
| 50 | 
 | 
|---|
| 51 | static int handle_aio_read_complete(struct aio_extra *aio_ex, int errcode);
 | 
|---|
| 52 | static int handle_aio_write_complete(struct aio_extra *aio_ex, int errcode);
 | 
|---|
| 53 | 
 | 
|---|
| 54 | static int aio_extra_destructor(struct aio_extra *aio_ex)
 | 
|---|
| 55 | {
 | 
|---|
| 56 |         DLIST_REMOVE(aio_list_head, aio_ex);
 | 
|---|
| 57 |         return 0;
 | 
|---|
| 58 | }
 | 
|---|
| 59 | 
 | 
|---|
| 60 | /****************************************************************************
 | 
|---|
| 61 |  Create the extended aio struct we must keep around for the lifetime
 | 
|---|
| 62 |  of the aio call.
 | 
|---|
| 63 | *****************************************************************************/
 | 
|---|
| 64 | 
 | 
|---|
| 65 | static struct aio_extra *create_aio_extra(files_struct *fsp, size_t buflen)
 | 
|---|
| 66 | {
 | 
|---|
| 67 |         struct aio_extra *aio_ex = TALLOC_ZERO_P(NULL, struct aio_extra);
 | 
|---|
| 68 | 
 | 
|---|
| 69 |         if (!aio_ex) {
 | 
|---|
| 70 |                 return NULL;
 | 
|---|
| 71 |         }
 | 
|---|
| 72 | 
 | 
|---|
| 73 |         /* The output buffer stored in the aio_ex is the start of
 | 
|---|
| 74 |            the smb return buffer. The buffer used in the acb
 | 
|---|
| 75 |            is the start of the reply data portion of that buffer. */
 | 
|---|
| 76 | 
 | 
|---|
| 77 |         aio_ex->outbuf = TALLOC_ARRAY(aio_ex, char, buflen);
 | 
|---|
| 78 |         if (!aio_ex->outbuf) {
 | 
|---|
| 79 |                 TALLOC_FREE(aio_ex);
 | 
|---|
| 80 |                 return NULL;
 | 
|---|
| 81 |         }
 | 
|---|
| 82 |         DLIST_ADD(aio_list_head, aio_ex);
 | 
|---|
| 83 |         talloc_set_destructor(aio_ex, aio_extra_destructor);
 | 
|---|
| 84 |         aio_ex->fsp = fsp;
 | 
|---|
| 85 |         return aio_ex;
 | 
|---|
| 86 | }
 | 
|---|
| 87 | 
 | 
|---|
| 88 | /****************************************************************************
 | 
|---|
| 89 |  Given the mid find the extended aio struct containing it.
 | 
|---|
| 90 | *****************************************************************************/
 | 
|---|
| 91 | 
 | 
|---|
| 92 | static struct aio_extra *find_aio_ex(uint16 mid)
 | 
|---|
| 93 | {
 | 
|---|
| 94 |         struct aio_extra *p;
 | 
|---|
| 95 | 
 | 
|---|
| 96 |         for( p = aio_list_head; p; p = p->next) {
 | 
|---|
| 97 |                 if (mid == p->req->mid) {
 | 
|---|
| 98 |                         return p;
 | 
|---|
| 99 |                 }
 | 
|---|
| 100 |         }
 | 
|---|
| 101 |         return NULL;
 | 
|---|
| 102 | }
 | 
|---|
| 103 | 
 | 
|---|
| 104 | /****************************************************************************
 | 
|---|
| 105 |  We can have these many aio buffers in flight.
 | 
|---|
| 106 | *****************************************************************************/
 | 
|---|
| 107 | 
 | 
|---|
| 108 | /****************************************************************************
 | 
|---|
| 109 |  Set up an aio request from a SMBreadX call.
 | 
|---|
| 110 | *****************************************************************************/
 | 
|---|
| 111 | 
 | 
|---|
| 112 | bool schedule_aio_read_and_X(connection_struct *conn,
 | 
|---|
| 113 |                              struct smb_request *req,
 | 
|---|
| 114 |                              files_struct *fsp, SMB_OFF_T startpos,
 | 
|---|
| 115 |                              size_t smb_maxcnt)
 | 
|---|
| 116 | {
 | 
|---|
| 117 |         struct aio_extra *aio_ex;
 | 
|---|
| 118 |         SMB_STRUCT_AIOCB *a;
 | 
|---|
| 119 |         size_t bufsize;
 | 
|---|
| 120 |         size_t min_aio_read_size = lp_aio_read_size(SNUM(conn));
 | 
|---|
| 121 |         int ret;
 | 
|---|
| 122 | 
 | 
|---|
| 123 |         if (fsp->base_fsp != NULL) {
 | 
|---|
| 124 |                 /* No AIO on streams yet */
 | 
|---|
| 125 |                 DEBUG(10, ("AIO on streams not yet supported\n"));
 | 
|---|
| 126 |                 return false;
 | 
|---|
| 127 |         }
 | 
|---|
| 128 | 
 | 
|---|
| 129 |         if ((!min_aio_read_size || (smb_maxcnt < min_aio_read_size))
 | 
|---|
| 130 |             && !SMB_VFS_AIO_FORCE(fsp)) {
 | 
|---|
| 131 |                 /* Too small a read for aio request. */
 | 
|---|
| 132 |                 DEBUG(10,("schedule_aio_read_and_X: read size (%u) too small "
 | 
|---|
| 133 |                           "for minimum aio_read of %u\n",
 | 
|---|
| 134 |                           (unsigned int)smb_maxcnt,
 | 
|---|
| 135 |                           (unsigned int)min_aio_read_size ));
 | 
|---|
| 136 |                 return False;
 | 
|---|
| 137 |         }
 | 
|---|
| 138 | 
 | 
|---|
| 139 |         /* Only do this on non-chained and non-chaining reads not using the
 | 
|---|
| 140 |          * write cache. */
 | 
|---|
| 141 |         if (req_is_in_chain(req) || (lp_write_cache_size(SNUM(conn)) != 0)) {
 | 
|---|
| 142 |                 return False;
 | 
|---|
| 143 |         }
 | 
|---|
| 144 | 
 | 
|---|
| 145 |         if (outstanding_aio_calls >= aio_pending_size) {
 | 
|---|
| 146 |                 DEBUG(10,("schedule_aio_read_and_X: Already have %d aio "
 | 
|---|
| 147 |                           "activities outstanding.\n",
 | 
|---|
| 148 |                           outstanding_aio_calls ));
 | 
|---|
| 149 |                 return False;
 | 
|---|
| 150 |         }
 | 
|---|
| 151 | 
 | 
|---|
| 152 |         /* The following is safe from integer wrap as we've already checked
 | 
|---|
| 153 |            smb_maxcnt is 128k or less. Wct is 12 for read replies */
 | 
|---|
| 154 | 
 | 
|---|
| 155 |         bufsize = smb_size + 12 * 2 + smb_maxcnt;
 | 
|---|
| 156 | 
 | 
|---|
| 157 |         if ((aio_ex = create_aio_extra(fsp, bufsize)) == NULL) {
 | 
|---|
| 158 |                 DEBUG(10,("schedule_aio_read_and_X: malloc fail.\n"));
 | 
|---|
| 159 |                 return False;
 | 
|---|
| 160 |         }
 | 
|---|
| 161 |         aio_ex->handle_completion = handle_aio_read_complete;
 | 
|---|
| 162 | 
 | 
|---|
| 163 |         construct_reply_common_req(req, aio_ex->outbuf);
 | 
|---|
| 164 |         srv_set_message(aio_ex->outbuf, 12, 0, True);
 | 
|---|
| 165 |         SCVAL(aio_ex->outbuf,smb_vwv0,0xFF); /* Never a chained reply. */
 | 
|---|
| 166 | 
 | 
|---|
| 167 |         a = &aio_ex->acb;
 | 
|---|
| 168 | 
 | 
|---|
| 169 |         /* Now set up the aio record for the read call. */
 | 
|---|
| 170 | 
 | 
|---|
| 171 |         a->aio_fildes = fsp->fh->fd;
 | 
|---|
| 172 |         a->aio_buf = smb_buf(aio_ex->outbuf);
 | 
|---|
| 173 |         a->aio_nbytes = smb_maxcnt;
 | 
|---|
| 174 |         a->aio_offset = startpos;
 | 
|---|
| 175 |         a->aio_sigevent.sigev_notify = SIGEV_SIGNAL;
 | 
|---|
| 176 |         a->aio_sigevent.sigev_signo  = RT_SIGNAL_AIO;
 | 
|---|
| 177 |         a->aio_sigevent.sigev_value.sival_int = req->mid;
 | 
|---|
| 178 | 
 | 
|---|
| 179 |         ret = SMB_VFS_AIO_READ(fsp, a);
 | 
|---|
| 180 |         if (ret == -1) {
 | 
|---|
| 181 |                 DEBUG(0,("schedule_aio_read_and_X: aio_read failed. "
 | 
|---|
| 182 |                          "Error %s\n", strerror(errno) ));
 | 
|---|
| 183 |                 TALLOC_FREE(aio_ex);
 | 
|---|
| 184 |                 return False;
 | 
|---|
| 185 |         }
 | 
|---|
| 186 | 
 | 
|---|
| 187 |         outstanding_aio_calls++;
 | 
|---|
| 188 |         aio_ex->req = talloc_move(aio_ex, &req);
 | 
|---|
| 189 | 
 | 
|---|
| 190 |         DEBUG(10,("schedule_aio_read_and_X: scheduled aio_read for file %s, "
 | 
|---|
| 191 |                   "offset %.0f, len = %u (mid = %u)\n",
 | 
|---|
| 192 |                   fsp_str_dbg(fsp), (double)startpos, (unsigned int)smb_maxcnt,
 | 
|---|
| 193 |                   (unsigned int)aio_ex->req->mid ));
 | 
|---|
| 194 | 
 | 
|---|
| 195 |         return True;
 | 
|---|
| 196 | }
 | 
|---|
| 197 | 
 | 
|---|
| 198 | /****************************************************************************
 | 
|---|
| 199 |  Set up an aio request from a SMBwriteX call.
 | 
|---|
| 200 | *****************************************************************************/
 | 
|---|
| 201 | 
 | 
|---|
| 202 | bool schedule_aio_write_and_X(connection_struct *conn,
 | 
|---|
| 203 |                               struct smb_request *req,
 | 
|---|
| 204 |                               files_struct *fsp, char *data,
 | 
|---|
| 205 |                               SMB_OFF_T startpos,
 | 
|---|
| 206 |                               size_t numtowrite)
 | 
|---|
| 207 | {
 | 
|---|
| 208 |         struct aio_extra *aio_ex;
 | 
|---|
| 209 |         SMB_STRUCT_AIOCB *a;
 | 
|---|
| 210 |         size_t bufsize;
 | 
|---|
| 211 |         bool write_through = BITSETW(req->vwv+7,0);
 | 
|---|
| 212 |         size_t min_aio_write_size = lp_aio_write_size(SNUM(conn));
 | 
|---|
| 213 |         int ret;
 | 
|---|
| 214 | 
 | 
|---|
| 215 |         if (fsp->base_fsp != NULL) {
 | 
|---|
| 216 |                 /* No AIO on streams yet */
 | 
|---|
| 217 |                 DEBUG(10, ("AIO on streams not yet supported\n"));
 | 
|---|
| 218 |                 return false;
 | 
|---|
| 219 |         }
 | 
|---|
| 220 | 
 | 
|---|
| 221 |         if ((!min_aio_write_size || (numtowrite < min_aio_write_size))
 | 
|---|
| 222 |             && !SMB_VFS_AIO_FORCE(fsp)) {
 | 
|---|
| 223 |                 /* Too small a write for aio request. */
 | 
|---|
| 224 |                 DEBUG(10,("schedule_aio_write_and_X: write size (%u) too "
 | 
|---|
| 225 |                           "small for minimum aio_write of %u\n",
 | 
|---|
| 226 |                           (unsigned int)numtowrite,
 | 
|---|
| 227 |                           (unsigned int)min_aio_write_size ));
 | 
|---|
| 228 |                 return False;
 | 
|---|
| 229 |         }
 | 
|---|
| 230 | 
 | 
|---|
| 231 |         /* Only do this on non-chained and non-chaining reads not using the
 | 
|---|
| 232 |          * write cache. */
 | 
|---|
| 233 |         if (req_is_in_chain(req) || (lp_write_cache_size(SNUM(conn)) != 0)) {
 | 
|---|
| 234 |                 return False;
 | 
|---|
| 235 |         }
 | 
|---|
| 236 | 
 | 
|---|
| 237 |         if (outstanding_aio_calls >= aio_pending_size) {
 | 
|---|
| 238 |                 DEBUG(3,("schedule_aio_write_and_X: Already have %d aio "
 | 
|---|
| 239 |                          "activities outstanding.\n",
 | 
|---|
| 240 |                           outstanding_aio_calls ));
 | 
|---|
| 241 |                 DEBUG(10,("schedule_aio_write_and_X: failed to schedule "
 | 
|---|
| 242 |                           "aio_write for file %s, offset %.0f, len = %u "
 | 
|---|
| 243 |                           "(mid = %u)\n",
 | 
|---|
| 244 |                           fsp_str_dbg(fsp), (double)startpos,
 | 
|---|
| 245 |                           (unsigned int)numtowrite,
 | 
|---|
| 246 |                           (unsigned int)req->mid ));
 | 
|---|
| 247 |                 return False;
 | 
|---|
| 248 |         }
 | 
|---|
| 249 | 
 | 
|---|
| 250 |         bufsize = smb_size + 6*2;
 | 
|---|
| 251 | 
 | 
|---|
| 252 |         if (!(aio_ex = create_aio_extra(fsp, bufsize))) {
 | 
|---|
| 253 |                 DEBUG(0,("schedule_aio_write_and_X: malloc fail.\n"));
 | 
|---|
| 254 |                 return False;
 | 
|---|
| 255 |         }
 | 
|---|
| 256 |         aio_ex->handle_completion = handle_aio_write_complete;
 | 
|---|
| 257 | 
 | 
|---|
| 258 |         construct_reply_common_req(req, aio_ex->outbuf);
 | 
|---|
| 259 |         srv_set_message(aio_ex->outbuf, 6, 0, True);
 | 
|---|
| 260 |         SCVAL(aio_ex->outbuf,smb_vwv0,0xFF); /* Never a chained reply. */
 | 
|---|
| 261 | 
 | 
|---|
| 262 |         a = &aio_ex->acb;
 | 
|---|
| 263 | 
 | 
|---|
| 264 |         /* Now set up the aio record for the write call. */
 | 
|---|
| 265 | 
 | 
|---|
| 266 |         a->aio_fildes = fsp->fh->fd;
 | 
|---|
| 267 |         a->aio_buf = data;
 | 
|---|
| 268 |         a->aio_nbytes = numtowrite;
 | 
|---|
| 269 |         a->aio_offset = startpos;
 | 
|---|
| 270 |         a->aio_sigevent.sigev_notify = SIGEV_SIGNAL;
 | 
|---|
| 271 |         a->aio_sigevent.sigev_signo  = RT_SIGNAL_AIO;
 | 
|---|
| 272 |         a->aio_sigevent.sigev_value.sival_int = req->mid;
 | 
|---|
| 273 | 
 | 
|---|
| 274 |         ret = SMB_VFS_AIO_WRITE(fsp, a);
 | 
|---|
| 275 |         if (ret == -1) {
 | 
|---|
| 276 |                 DEBUG(3,("schedule_aio_wrote_and_X: aio_write failed. "
 | 
|---|
| 277 |                          "Error %s\n", strerror(errno) ));
 | 
|---|
| 278 |                 TALLOC_FREE(aio_ex);
 | 
|---|
| 279 |                 return False;
 | 
|---|
| 280 |         }
 | 
|---|
| 281 | 
 | 
|---|
| 282 |         outstanding_aio_calls++;
 | 
|---|
| 283 |         aio_ex->req = talloc_move(aio_ex, &req);
 | 
|---|
| 284 | 
 | 
|---|
| 285 |         /* This should actually be improved to span the write. */
 | 
|---|
| 286 |         contend_level2_oplocks_begin(fsp, LEVEL2_CONTEND_WRITE);
 | 
|---|
| 287 |         contend_level2_oplocks_end(fsp, LEVEL2_CONTEND_WRITE);
 | 
|---|
| 288 | 
 | 
|---|
| 289 |         if (!write_through && !lp_syncalways(SNUM(fsp->conn))
 | 
|---|
| 290 |             && fsp->aio_write_behind) {
 | 
|---|
| 291 |                 /* Lie to the client and immediately claim we finished the
 | 
|---|
| 292 |                  * write. */
 | 
|---|
| 293 |                 SSVAL(aio_ex->outbuf,smb_vwv2,numtowrite);
 | 
|---|
| 294 |                 SSVAL(aio_ex->outbuf,smb_vwv4,(numtowrite>>16)&1);
 | 
|---|
| 295 |                 show_msg(aio_ex->outbuf);
 | 
|---|
| 296 |                 if (!srv_send_smb(smbd_server_fd(),aio_ex->outbuf,
 | 
|---|
| 297 |                                 true, aio_ex->req->seqnum+1,
 | 
|---|
| 298 |                                 IS_CONN_ENCRYPTED(fsp->conn),
 | 
|---|
| 299 |                                 &aio_ex->req->pcd)) {
 | 
|---|
| 300 |                         exit_server_cleanly("handle_aio_write: srv_send_smb "
 | 
|---|
| 301 |                                             "failed.");
 | 
|---|
| 302 |                 }
 | 
|---|
| 303 |                 DEBUG(10,("schedule_aio_write_and_X: scheduled aio_write "
 | 
|---|
| 304 |                           "behind for file %s\n", fsp_str_dbg(fsp)));
 | 
|---|
| 305 |         }
 | 
|---|
| 306 | 
 | 
|---|
| 307 |         DEBUG(10,("schedule_aio_write_and_X: scheduled aio_write for file "
 | 
|---|
| 308 |                   "%s, offset %.0f, len = %u (mid = %u) "
 | 
|---|
| 309 |                   "outstanding_aio_calls = %d\n",
 | 
|---|
| 310 |                   fsp_str_dbg(fsp), (double)startpos, (unsigned int)numtowrite,
 | 
|---|
| 311 |                   (unsigned int)aio_ex->req->mid, outstanding_aio_calls ));
 | 
|---|
| 312 | 
 | 
|---|
| 313 |         return True;
 | 
|---|
| 314 | }
 | 
|---|
| 315 | 
 | 
|---|
| 316 | 
 | 
|---|
| 317 | /****************************************************************************
 | 
|---|
| 318 |  Complete the read and return the data or error back to the client.
 | 
|---|
| 319 |  Returns errno or zero if all ok.
 | 
|---|
| 320 | *****************************************************************************/
 | 
|---|
| 321 | 
 | 
|---|
| 322 | static int handle_aio_read_complete(struct aio_extra *aio_ex, int errcode)
 | 
|---|
| 323 | {
 | 
|---|
| 324 |         int outsize;
 | 
|---|
| 325 |         char *outbuf = aio_ex->outbuf;
 | 
|---|
| 326 |         char *data = smb_buf(outbuf);
 | 
|---|
| 327 |         ssize_t nread = SMB_VFS_AIO_RETURN(aio_ex->fsp,&aio_ex->acb);
 | 
|---|
| 328 | 
 | 
|---|
| 329 |         if (nread < 0) {
 | 
|---|
| 330 |                 /* We're relying here on the fact that if the fd is
 | 
|---|
| 331 |                    closed then the aio will complete and aio_return
 | 
|---|
| 332 |                    will return an error. Hopefully this is
 | 
|---|
| 333 |                    true.... JRA. */
 | 
|---|
| 334 | 
 | 
|---|
| 335 |                 DEBUG( 3,( "handle_aio_read_complete: file %s nread == %d. "
 | 
|---|
| 336 |                            "Error = %s\n",
 | 
|---|
| 337 |                            fsp_str_dbg(aio_ex->fsp), (int)nread, strerror(errcode)));
 | 
|---|
| 338 | 
 | 
|---|
| 339 |                 ERROR_NT(map_nt_error_from_unix(errcode));
 | 
|---|
| 340 |                 outsize = srv_set_message(outbuf,0,0,true);
 | 
|---|
| 341 |         } else {
 | 
|---|
| 342 |                 outsize = srv_set_message(outbuf,12,nread,False);
 | 
|---|
| 343 |                 SSVAL(outbuf,smb_vwv2,0xFFFF); /* Remaining - must be * -1. */
 | 
|---|
| 344 |                 SSVAL(outbuf,smb_vwv5,nread);
 | 
|---|
| 345 |                 SSVAL(outbuf,smb_vwv6,smb_offset(data,outbuf));
 | 
|---|
| 346 |                 SSVAL(outbuf,smb_vwv7,((nread >> 16) & 1));
 | 
|---|
| 347 |                 SSVAL(smb_buf(outbuf),-2,nread);
 | 
|---|
| 348 | 
 | 
|---|
| 349 |                 aio_ex->fsp->fh->pos = aio_ex->acb.aio_offset + nread;
 | 
|---|
| 350 |                 aio_ex->fsp->fh->position_information = aio_ex->fsp->fh->pos;
 | 
|---|
| 351 | 
 | 
|---|
| 352 |                 DEBUG( 3, ( "handle_aio_read_complete file %s max=%d "
 | 
|---|
| 353 |                             "nread=%d\n",
 | 
|---|
| 354 |                             fsp_str_dbg(aio_ex->fsp),
 | 
|---|
| 355 |                             (int)aio_ex->acb.aio_nbytes, (int)nread ) );
 | 
|---|
| 356 | 
 | 
|---|
| 357 |         }
 | 
|---|
| 358 |         smb_setlen(outbuf,outsize - 4);
 | 
|---|
| 359 |         show_msg(outbuf);
 | 
|---|
| 360 |         if (!srv_send_smb(smbd_server_fd(),outbuf,
 | 
|---|
| 361 |                         true, aio_ex->req->seqnum+1,
 | 
|---|
| 362 |                         IS_CONN_ENCRYPTED(aio_ex->fsp->conn), NULL)) {
 | 
|---|
| 363 |                 exit_server_cleanly("handle_aio_read_complete: srv_send_smb "
 | 
|---|
| 364 |                                     "failed.");
 | 
|---|
| 365 |         }
 | 
|---|
| 366 | 
 | 
|---|
| 367 |         DEBUG(10,("handle_aio_read_complete: scheduled aio_read completed "
 | 
|---|
| 368 |                   "for file %s, offset %.0f, len = %u\n",
 | 
|---|
| 369 |                   fsp_str_dbg(aio_ex->fsp), (double)aio_ex->acb.aio_offset,
 | 
|---|
| 370 |                   (unsigned int)nread ));
 | 
|---|
| 371 | 
 | 
|---|
| 372 |         return errcode;
 | 
|---|
| 373 | }
 | 
|---|
| 374 | 
 | 
|---|
| 375 | /****************************************************************************
 | 
|---|
| 376 |  Complete the write and return the data or error back to the client.
 | 
|---|
| 377 |  Returns error code or zero if all ok.
 | 
|---|
| 378 | *****************************************************************************/
 | 
|---|
| 379 | 
 | 
|---|
| 380 | static int handle_aio_write_complete(struct aio_extra *aio_ex, int errcode)
 | 
|---|
| 381 | {
 | 
|---|
| 382 |         files_struct *fsp = aio_ex->fsp;
 | 
|---|
| 383 |         char *outbuf = aio_ex->outbuf;
 | 
|---|
| 384 |         ssize_t numtowrite = aio_ex->acb.aio_nbytes;
 | 
|---|
| 385 |         ssize_t nwritten = SMB_VFS_AIO_RETURN(fsp,&aio_ex->acb);
 | 
|---|
| 386 | 
 | 
|---|
| 387 |         if (fsp->aio_write_behind) {
 | 
|---|
| 388 |                 if (nwritten != numtowrite) {
 | 
|---|
| 389 |                         if (nwritten == -1) {
 | 
|---|
| 390 |                                 DEBUG(5,("handle_aio_write_complete: "
 | 
|---|
| 391 |                                          "aio_write_behind failed ! File %s "
 | 
|---|
| 392 |                                          "is corrupt ! Error %s\n",
 | 
|---|
| 393 |                                          fsp_str_dbg(fsp), strerror(errcode)));
 | 
|---|
| 394 |                         } else {
 | 
|---|
| 395 |                                 DEBUG(0,("handle_aio_write_complete: "
 | 
|---|
| 396 |                                          "aio_write_behind failed ! File %s "
 | 
|---|
| 397 |                                          "is corrupt ! Wanted %u bytes but "
 | 
|---|
| 398 |                                          "only wrote %d\n", fsp_str_dbg(fsp),
 | 
|---|
| 399 |                                          (unsigned int)numtowrite,
 | 
|---|
| 400 |                                          (int)nwritten ));
 | 
|---|
| 401 |                                 errcode = EIO;
 | 
|---|
| 402 |                         }
 | 
|---|
| 403 |                 } else {
 | 
|---|
| 404 |                         DEBUG(10,("handle_aio_write_complete: "
 | 
|---|
| 405 |                                   "aio_write_behind completed for file %s\n",
 | 
|---|
| 406 |                                   fsp_str_dbg(fsp)));
 | 
|---|
| 407 |                 }
 | 
|---|
| 408 |                 /* TODO: should no return 0 in case of an error !!! */
 | 
|---|
| 409 |                 return 0;
 | 
|---|
| 410 |         }
 | 
|---|
| 411 | 
 | 
|---|
| 412 |         /* We don't need outsize or set_message here as we've already set the
 | 
|---|
| 413 |            fixed size length when we set up the aio call. */
 | 
|---|
| 414 | 
 | 
|---|
| 415 |         if(nwritten == -1) {
 | 
|---|
| 416 |                 DEBUG( 3,( "handle_aio_write: file %s wanted %u bytes. "
 | 
|---|
| 417 |                            "nwritten == %d. Error = %s\n",
 | 
|---|
| 418 |                            fsp_str_dbg(fsp), (unsigned int)numtowrite,
 | 
|---|
| 419 |                            (int)nwritten, strerror(errcode) ));
 | 
|---|
| 420 | 
 | 
|---|
| 421 |                 ERROR_NT(map_nt_error_from_unix(errcode));
 | 
|---|
| 422 |                 srv_set_message(outbuf,0,0,true);
 | 
|---|
| 423 |         } else {
 | 
|---|
| 424 |                 bool write_through = BITSETW(aio_ex->req->vwv+7,0);
 | 
|---|
| 425 |                 NTSTATUS status;
 | 
|---|
| 426 | 
 | 
|---|
| 427 |                 SSVAL(outbuf,smb_vwv2,nwritten);
 | 
|---|
| 428 |                 SSVAL(outbuf,smb_vwv4,(nwritten>>16)&1);
 | 
|---|
| 429 |                 if (nwritten < (ssize_t)numtowrite) {
 | 
|---|
| 430 |                         SCVAL(outbuf,smb_rcls,ERRHRD);
 | 
|---|
| 431 |                         SSVAL(outbuf,smb_err,ERRdiskfull);
 | 
|---|
| 432 |                 }
 | 
|---|
| 433 | 
 | 
|---|
| 434 |                 DEBUG(3,("handle_aio_write: fnum=%d num=%d wrote=%d\n",
 | 
|---|
| 435 |                          fsp->fnum, (int)numtowrite, (int)nwritten));
 | 
|---|
| 436 |                 status = sync_file(fsp->conn,fsp, write_through);
 | 
|---|
| 437 |                 if (!NT_STATUS_IS_OK(status)) {
 | 
|---|
| 438 |                         errcode = errno;
 | 
|---|
| 439 |                         ERROR_BOTH(map_nt_error_from_unix(errcode),
 | 
|---|
| 440 |                                    ERRHRD, ERRdiskfull);
 | 
|---|
| 441 |                         srv_set_message(outbuf,0,0,true);
 | 
|---|
| 442 |                         DEBUG(5,("handle_aio_write: sync_file for %s returned %s\n",
 | 
|---|
| 443 |                                  fsp_str_dbg(fsp), nt_errstr(status)));
 | 
|---|
| 444 |                 }
 | 
|---|
| 445 | 
 | 
|---|
| 446 |                 aio_ex->fsp->fh->pos = aio_ex->acb.aio_offset + nwritten;
 | 
|---|
| 447 |         }
 | 
|---|
| 448 | 
 | 
|---|
| 449 |         show_msg(outbuf);
 | 
|---|
| 450 |         if (!srv_send_smb(smbd_server_fd(),outbuf,
 | 
|---|
| 451 |                           true, aio_ex->req->seqnum+1,
 | 
|---|
| 452 |                           IS_CONN_ENCRYPTED(fsp->conn),
 | 
|---|
| 453 |                           NULL)) {
 | 
|---|
| 454 |                 exit_server_cleanly("handle_aio_write: srv_send_smb failed.");
 | 
|---|
| 455 |         }
 | 
|---|
| 456 | 
 | 
|---|
| 457 |         DEBUG(10,("handle_aio_write_complete: scheduled aio_write completed "
 | 
|---|
| 458 |                   "for file %s, offset %.0f, requested %u, written = %u\n",
 | 
|---|
| 459 |                   fsp_str_dbg(fsp), (double)aio_ex->acb.aio_offset,
 | 
|---|
| 460 |                   (unsigned int)numtowrite, (unsigned int)nwritten ));
 | 
|---|
| 461 | 
 | 
|---|
| 462 |         return errcode;
 | 
|---|
| 463 | }
 | 
|---|
| 464 | 
 | 
|---|
| 465 | /****************************************************************************
 | 
|---|
| 466 |  Handle any aio completion. Returns True if finished (and sets *perr if err
 | 
|---|
| 467 |  was non-zero), False if not.
 | 
|---|
| 468 | *****************************************************************************/
 | 
|---|
| 469 | 
 | 
|---|
| 470 | static bool handle_aio_completed(struct aio_extra *aio_ex, int *perr)
 | 
|---|
| 471 | {
 | 
|---|
| 472 |         int err;
 | 
|---|
| 473 | 
 | 
|---|
| 474 |         if(!aio_ex) {
 | 
|---|
| 475 |                 DEBUG(3, ("handle_aio_completed: Non-existing aio_ex passed\n"));
 | 
|---|
| 476 |                 return false;
 | 
|---|
| 477 |         }
 | 
|---|
| 478 | 
 | 
|---|
| 479 |         /* Ensure the operation has really completed. */
 | 
|---|
| 480 |         err = SMB_VFS_AIO_ERROR(aio_ex->fsp, &aio_ex->acb);
 | 
|---|
| 481 |         if (err == EINPROGRESS) {
 | 
|---|
| 482 |                 DEBUG(10,( "handle_aio_completed: operation mid %u still in "
 | 
|---|
| 483 |                            "process for file %s\n",
 | 
|---|
| 484 |                            aio_ex->req->mid, fsp_str_dbg(aio_ex->fsp)));
 | 
|---|
| 485 |                 return False;
 | 
|---|
| 486 |         } else if (err == ECANCELED) {
 | 
|---|
| 487 |                 /* If error is ECANCELED then don't return anything to the
 | 
|---|
| 488 |                  * client. */
 | 
|---|
| 489 |                 DEBUG(10,( "handle_aio_completed: operation mid %u"
 | 
|---|
| 490 |                            " canceled\n", aio_ex->req->mid));
 | 
|---|
| 491 |                 return True;
 | 
|---|
| 492 |         }
 | 
|---|
| 493 | 
 | 
|---|
| 494 |         err = aio_ex->handle_completion(aio_ex, err);
 | 
|---|
| 495 |         if (err) {
 | 
|---|
| 496 |                 *perr = err; /* Only save non-zero errors. */
 | 
|---|
| 497 |         }
 | 
|---|
| 498 | 
 | 
|---|
| 499 |         return True;
 | 
|---|
| 500 | }
 | 
|---|
| 501 | 
 | 
|---|
| 502 | /****************************************************************************
 | 
|---|
| 503 |  Handle any aio completion inline.
 | 
|---|
| 504 | *****************************************************************************/
 | 
|---|
| 505 | 
 | 
|---|
| 506 | void smbd_aio_complete_mid(unsigned int mid)
 | 
|---|
| 507 | {
 | 
|---|
| 508 |         files_struct *fsp = NULL;
 | 
|---|
| 509 |         struct aio_extra *aio_ex = find_aio_ex(mid);
 | 
|---|
| 510 |         int ret = 0;
 | 
|---|
| 511 | 
 | 
|---|
| 512 |         outstanding_aio_calls--;
 | 
|---|
| 513 | 
 | 
|---|
| 514 |         DEBUG(10,("smbd_aio_complete_mid: mid[%u]\n", mid));
 | 
|---|
| 515 | 
 | 
|---|
| 516 |         if (!aio_ex) {
 | 
|---|
| 517 |                 DEBUG(3,("smbd_aio_complete_mid: Can't find record to "
 | 
|---|
| 518 |                          "match mid %u.\n", mid));
 | 
|---|
| 519 |                 return;
 | 
|---|
| 520 |         }
 | 
|---|
| 521 | 
 | 
|---|
| 522 |         fsp = aio_ex->fsp;
 | 
|---|
| 523 |         if (fsp == NULL) {
 | 
|---|
| 524 |                 /* file was closed whilst I/O was outstanding. Just
 | 
|---|
| 525 |                  * ignore. */
 | 
|---|
| 526 |                 DEBUG( 3,( "smbd_aio_complete_mid: file closed whilst "
 | 
|---|
| 527 |                            "aio outstanding (mid[%u]).\n", mid));
 | 
|---|
| 528 |                 return;
 | 
|---|
| 529 |         }
 | 
|---|
| 530 | 
 | 
|---|
| 531 |         if (!handle_aio_completed(aio_ex, &ret)) {
 | 
|---|
| 532 |                 return;
 | 
|---|
| 533 |         }
 | 
|---|
| 534 | 
 | 
|---|
| 535 |         TALLOC_FREE(aio_ex);
 | 
|---|
| 536 | }
 | 
|---|
| 537 | 
 | 
|---|
| 538 | static void smbd_aio_signal_handler(struct tevent_context *ev_ctx,
 | 
|---|
| 539 |                                     struct tevent_signal *se,
 | 
|---|
| 540 |                                     int signum, int count,
 | 
|---|
| 541 |                                     void *_info, void *private_data)
 | 
|---|
| 542 | {
 | 
|---|
| 543 |         siginfo_t *info = (siginfo_t *)_info;
 | 
|---|
| 544 |         unsigned int mid = (unsigned int)info->si_value.sival_int;
 | 
|---|
| 545 | 
 | 
|---|
| 546 |         smbd_aio_complete_mid(mid);
 | 
|---|
| 547 | }
 | 
|---|
| 548 | 
 | 
|---|
| 549 | /****************************************************************************
 | 
|---|
| 550 |  We're doing write behind and the client closed the file. Wait up to 30
 | 
|---|
| 551 |  seconds (my arbitrary choice) for the aio to complete. Return 0 if all writes
 | 
|---|
| 552 |  completed, errno to return if not.
 | 
|---|
| 553 | *****************************************************************************/
 | 
|---|
| 554 | 
 | 
|---|
| 555 | #define SMB_TIME_FOR_AIO_COMPLETE_WAIT 29
 | 
|---|
| 556 | 
 | 
|---|
| 557 | int wait_for_aio_completion(files_struct *fsp)
 | 
|---|
| 558 | {
 | 
|---|
| 559 |         struct aio_extra *aio_ex;
 | 
|---|
| 560 |         const SMB_STRUCT_AIOCB **aiocb_list;
 | 
|---|
| 561 |         int aio_completion_count = 0;
 | 
|---|
| 562 |         time_t start_time = time(NULL);
 | 
|---|
| 563 |         int seconds_left;
 | 
|---|
| 564 | 
 | 
|---|
| 565 |         for (seconds_left = SMB_TIME_FOR_AIO_COMPLETE_WAIT;
 | 
|---|
| 566 |              seconds_left >= 0;) {
 | 
|---|
| 567 |                 int err = 0;
 | 
|---|
| 568 |                 int i;
 | 
|---|
| 569 |                 struct timespec ts;
 | 
|---|
| 570 | 
 | 
|---|
| 571 |                 aio_completion_count = 0;
 | 
|---|
| 572 |                 for( aio_ex = aio_list_head; aio_ex; aio_ex = aio_ex->next) {
 | 
|---|
| 573 |                         if (aio_ex->fsp == fsp) {
 | 
|---|
| 574 |                                 aio_completion_count++;
 | 
|---|
| 575 |                         }
 | 
|---|
| 576 |                 }
 | 
|---|
| 577 | 
 | 
|---|
| 578 |                 if (!aio_completion_count) {
 | 
|---|
| 579 |                         return 0;
 | 
|---|
| 580 |                 }
 | 
|---|
| 581 | 
 | 
|---|
| 582 |                 DEBUG(3,("wait_for_aio_completion: waiting for %d aio events "
 | 
|---|
| 583 |                          "to complete.\n", aio_completion_count ));
 | 
|---|
| 584 | 
 | 
|---|
| 585 |                 aiocb_list = SMB_MALLOC_ARRAY(const SMB_STRUCT_AIOCB *,
 | 
|---|
| 586 |                                               aio_completion_count);
 | 
|---|
| 587 |                 if (!aiocb_list) {
 | 
|---|
| 588 |                         return ENOMEM;
 | 
|---|
| 589 |                 }
 | 
|---|
| 590 | 
 | 
|---|
| 591 |                 for( i = 0, aio_ex = aio_list_head;
 | 
|---|
| 592 |                      aio_ex;
 | 
|---|
| 593 |                      aio_ex = aio_ex->next) {
 | 
|---|
| 594 |                         if (aio_ex->fsp == fsp) {
 | 
|---|
| 595 |                                 aiocb_list[i++] = &aio_ex->acb;
 | 
|---|
| 596 |                         }
 | 
|---|
| 597 |                 }
 | 
|---|
| 598 | 
 | 
|---|
| 599 |                 /* Now wait up to seconds_left for completion. */
 | 
|---|
| 600 |                 ts.tv_sec = seconds_left;
 | 
|---|
| 601 |                 ts.tv_nsec = 0;
 | 
|---|
| 602 | 
 | 
|---|
| 603 |                 DEBUG(10,("wait_for_aio_completion: %d events, doing a wait "
 | 
|---|
| 604 |                           "of %d seconds.\n",
 | 
|---|
| 605 |                           aio_completion_count, seconds_left ));
 | 
|---|
| 606 | 
 | 
|---|
| 607 |                 err = SMB_VFS_AIO_SUSPEND(fsp, aiocb_list,
 | 
|---|
| 608 |                                           aio_completion_count, &ts);
 | 
|---|
| 609 | 
 | 
|---|
| 610 |                 DEBUG(10,("wait_for_aio_completion: returned err = %d, "
 | 
|---|
| 611 |                           "errno = %s\n", err, strerror(errno) ));
 | 
|---|
| 612 | 
 | 
|---|
| 613 |                 if (err == -1 && errno == EAGAIN) {
 | 
|---|
| 614 |                         DEBUG(0,("wait_for_aio_completion: aio_suspend timed "
 | 
|---|
| 615 |                                  "out waiting for %d events after a wait of "
 | 
|---|
| 616 |                                  "%d seconds\n", aio_completion_count,
 | 
|---|
| 617 |                                  seconds_left));
 | 
|---|
| 618 |                         /* Timeout. */
 | 
|---|
| 619 |                         cancel_aio_by_fsp(fsp);
 | 
|---|
| 620 |                         SAFE_FREE(aiocb_list);
 | 
|---|
| 621 |                         return EIO;
 | 
|---|
| 622 |                 }
 | 
|---|
| 623 | 
 | 
|---|
| 624 |                 /* One or more events might have completed - process them if
 | 
|---|
| 625 |                  * so. */
 | 
|---|
| 626 |                 for( i = 0; i < aio_completion_count; i++) {
 | 
|---|
| 627 |                         uint16 mid = aiocb_list[i]->aio_sigevent.sigev_value.sival_int;
 | 
|---|
| 628 | 
 | 
|---|
| 629 |                         aio_ex = find_aio_ex(mid);
 | 
|---|
| 630 | 
 | 
|---|
| 631 |                         if (!aio_ex) {
 | 
|---|
| 632 |                                 DEBUG(0, ("wait_for_aio_completion: mid %u "
 | 
|---|
| 633 |                                           "doesn't match an aio record\n",
 | 
|---|
| 634 |                                           (unsigned int)mid ));
 | 
|---|
| 635 |                                 continue;
 | 
|---|
| 636 |                         }
 | 
|---|
| 637 | 
 | 
|---|
| 638 |                         if (!handle_aio_completed(aio_ex, &err)) {
 | 
|---|
| 639 |                                 continue;
 | 
|---|
| 640 |                         }
 | 
|---|
| 641 |                         TALLOC_FREE(aio_ex);
 | 
|---|
| 642 |                 }
 | 
|---|
| 643 | 
 | 
|---|
| 644 |                 SAFE_FREE(aiocb_list);
 | 
|---|
| 645 |                 seconds_left = SMB_TIME_FOR_AIO_COMPLETE_WAIT
 | 
|---|
| 646 |                         - (time(NULL) - start_time);
 | 
|---|
| 647 |         }
 | 
|---|
| 648 | 
 | 
|---|
| 649 |         /* We timed out - we don't know why. Return ret if already an error,
 | 
|---|
| 650 |          * else EIO. */
 | 
|---|
| 651 |         DEBUG(10,("wait_for_aio_completion: aio_suspend timed out waiting "
 | 
|---|
| 652 |                   "for %d events\n",
 | 
|---|
| 653 |                   aio_completion_count));
 | 
|---|
| 654 | 
 | 
|---|
| 655 |         return EIO;
 | 
|---|
| 656 | }
 | 
|---|
| 657 | 
 | 
|---|
| 658 | /****************************************************************************
 | 
|---|
| 659 |  Cancel any outstanding aio requests. The client doesn't care about the reply.
 | 
|---|
| 660 | *****************************************************************************/
 | 
|---|
| 661 | 
 | 
|---|
| 662 | void cancel_aio_by_fsp(files_struct *fsp)
 | 
|---|
| 663 | {
 | 
|---|
| 664 |         struct aio_extra *aio_ex;
 | 
|---|
| 665 | 
 | 
|---|
| 666 |         for( aio_ex = aio_list_head; aio_ex; aio_ex = aio_ex->next) {
 | 
|---|
| 667 |                 if (aio_ex->fsp == fsp) {
 | 
|---|
| 668 |                         /* Don't delete the aio_extra record as we may have
 | 
|---|
| 669 |                            completed and don't yet know it. Just do the
 | 
|---|
| 670 |                            aio_cancel call and return. */
 | 
|---|
| 671 |                         SMB_VFS_AIO_CANCEL(fsp, &aio_ex->acb);
 | 
|---|
| 672 |                         aio_ex->fsp = NULL; /* fsp will be closed when we
 | 
|---|
| 673 |                                              * return. */
 | 
|---|
| 674 |                 }
 | 
|---|
| 675 |         }
 | 
|---|
| 676 | }
 | 
|---|
| 677 | 
 | 
|---|
| 678 | /****************************************************************************
 | 
|---|
| 679 |  Initialize the signal handler for aio read/write.
 | 
|---|
| 680 | *****************************************************************************/
 | 
|---|
| 681 | 
 | 
|---|
| 682 | void initialize_async_io_handler(void)
 | 
|---|
| 683 | {
 | 
|---|
| 684 |         aio_signal_event = tevent_add_signal(smbd_event_context(),
 | 
|---|
| 685 |                                              smbd_event_context(),
 | 
|---|
| 686 |                                              RT_SIGNAL_AIO, SA_SIGINFO,
 | 
|---|
| 687 |                                              smbd_aio_signal_handler,
 | 
|---|
| 688 |                                              NULL);
 | 
|---|
| 689 |         if (!aio_signal_event) {
 | 
|---|
| 690 |                 exit_server("Failed to setup RT_SIGNAL_AIO handler");
 | 
|---|
| 691 |         }
 | 
|---|
| 692 | 
 | 
|---|
| 693 |         /* tevent supports 100 signal with SA_SIGINFO */
 | 
|---|
| 694 |         aio_pending_size = 100;
 | 
|---|
| 695 | }
 | 
|---|
| 696 | 
 | 
|---|
| 697 | #else
 | 
|---|
| 698 | void initialize_async_io_handler(void)
 | 
|---|
| 699 | {
 | 
|---|
| 700 | }
 | 
|---|
| 701 | 
 | 
|---|
| 702 | bool schedule_aio_read_and_X(connection_struct *conn,
 | 
|---|
| 703 |                              struct smb_request *req,
 | 
|---|
| 704 |                              files_struct *fsp, SMB_OFF_T startpos,
 | 
|---|
| 705 |                              size_t smb_maxcnt)
 | 
|---|
| 706 | {
 | 
|---|
| 707 |         return False;
 | 
|---|
| 708 | }
 | 
|---|
| 709 | 
 | 
|---|
| 710 | bool schedule_aio_write_and_X(connection_struct *conn,
 | 
|---|
| 711 |                               struct smb_request *req,
 | 
|---|
| 712 |                               files_struct *fsp, char *data,
 | 
|---|
| 713 |                               SMB_OFF_T startpos,
 | 
|---|
| 714 |                               size_t numtowrite)
 | 
|---|
| 715 | {
 | 
|---|
| 716 |         return False;
 | 
|---|
| 717 | }
 | 
|---|
| 718 | 
 | 
|---|
| 719 | void cancel_aio_by_fsp(files_struct *fsp)
 | 
|---|
| 720 | {
 | 
|---|
| 721 | }
 | 
|---|
| 722 | 
 | 
|---|
| 723 | int wait_for_aio_completion(files_struct *fsp)
 | 
|---|
| 724 | {
 | 
|---|
| 725 |         return ENOSYS;
 | 
|---|
| 726 | }
 | 
|---|
| 727 | 
 | 
|---|
| 728 | void smbd_aio_complete_mid(unsigned int mid);
 | 
|---|
| 729 | 
 | 
|---|
| 730 | #endif
 | 
|---|