]> git.ipfire.org Git - thirdparty/squid.git/blame - src/DiskIO/DiskThreads/aiops.cc
Ported: urllogin ACL from squid 2.7
[thirdparty/squid.git] / src / DiskIO / DiskThreads / aiops.cc
CommitLineData
cd748f27 1/*
cd748f27 2 * DEBUG: section 43 AIOPS
3 * AUTHOR: Stewart Forster <slf@connect.com.au>
4 *
2b6662ba 5 * SQUID Web Proxy Cache http://www.squid-cache.org/
cd748f27 6 * ----------------------------------------------------------
7 *
2b6662ba 8 * Squid is the result of efforts by numerous individuals from
9 * the Internet community; see the CONTRIBUTORS file for full
10 * details. Many organizations have provided support for Squid's
11 * development; see the SPONSORS file for full details. Squid is
12 * Copyrighted (C) 2001 by the Regents of the University of
13 * California; see the COPYRIGHT file for full details. Squid
14 * incorporates software developed and/or copyrighted by other
15 * sources; see the CREDITS file for full details.
cd748f27 16 *
17 * This program is free software; you can redistribute it and/or modify
18 * it under the terms of the GNU General Public License as published by
19 * the Free Software Foundation; either version 2 of the License, or
20 * (at your option) any later version.
26ac0430 21 *
cd748f27 22 * This program is distributed in the hope that it will be useful,
23 * but WITHOUT ANY WARRANTY; without even the implied warranty of
24 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
25 * GNU General Public License for more details.
26ac0430 26 *
cd748f27 27 * You should have received a copy of the GNU General Public License
28 * along with this program; if not, write to the Free Software
29 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
30 *
31 */
32
e4116bdb 33#ifndef _REENTRANT
34#error "_REENTRANT MUST be defined to build squid async io support."
35#endif
36
582c2af2 37#include "squid.h"
b9ae18aa 38#include "DiskThreads.h"
4d5904f7 39#include "SquidConfig.h"
cd748f27 40
4d5904f7
FC
41#include <stdio.h>
42#include <sys/stat.h>
43#include <fcntl.h>
44#include <pthread.h>
45#include <errno.h>
46#include <dirent.h>
47#include <signal.h>
cd748f27 48#if HAVE_SCHED_H
4d5904f7 49#include <sched.h>
cd748f27 50#endif
1ff991dc 51#include "DiskIO/DiskThreads/CommIO.h"
985c86bc 52#include "SquidTime.h"
a553a5a3 53#include "Store.h"
cd748f27 54
55#define RIDICULOUS_LENGTH 4096
56
c04d4f40 57enum _squidaio_thread_status {
cd748f27 58 _THREAD_STARTING = 0,
59 _THREAD_WAITING,
60 _THREAD_BUSY,
61 _THREAD_FAILED,
62 _THREAD_DONE
63};
c04d4f40 64typedef enum _squidaio_thread_status squidaio_thread_status;
cd748f27 65
26ac0430 66typedef struct squidaio_request_t {
62e76326 67
c04d4f40 68 struct squidaio_request_t *next;
69 squidaio_request_type request_type;
cd748f27 70 int cancelled;
71 char *path;
72 int oflag;
73 mode_t mode;
74 int fd;
75 char *bufferp;
ee139403 76 size_t buflen;
cd748f27 77 off_t offset;
78 int whence;
79 int ret;
80 int err;
62e76326 81
cd748f27 82 struct stat *tmpstatp;
62e76326 83
cd748f27 84 struct stat *statp;
c04d4f40 85 squidaio_result_t *resultp;
2fadd50d 86} squidaio_request_t;
cd748f27 87
26ac0430 88typedef struct squidaio_request_queue_t {
55f0e6f7 89 pthread_mutex_t mutex;
90 pthread_cond_t cond;
c04d4f40 91 squidaio_request_t *volatile head;
92 squidaio_request_t *volatile *volatile tailp;
55f0e6f7 93 unsigned long requests;
f0debecb 94 unsigned long blocked; /* main failed to lock the queue */
2fadd50d 95} squidaio_request_queue_t;
55f0e6f7 96
c04d4f40 97typedef struct squidaio_thread_t squidaio_thread_t;
62e76326 98
26ac0430 99struct squidaio_thread_t {
c04d4f40 100 squidaio_thread_t *next;
cd748f27 101 pthread_t thread;
c04d4f40 102 squidaio_thread_status status;
62e76326 103
c04d4f40 104 struct squidaio_request_t *current_req;
55f0e6f7 105 unsigned long requests;
106};
cd748f27 107
c04d4f40 108static void squidaio_queue_request(squidaio_request_t *);
109static void squidaio_cleanup_request(squidaio_request_t *);
d9c252f2 110void *squidaio_thread_loop(void *);
c04d4f40 111static void squidaio_do_open(squidaio_request_t *);
112static void squidaio_do_read(squidaio_request_t *);
113static void squidaio_do_write(squidaio_request_t *);
114static void squidaio_do_close(squidaio_request_t *);
115static void squidaio_do_stat(squidaio_request_t *);
efa3acd1 116static void squidaio_do_unlink(squidaio_request_t *);
cd748f27 117#if AIO_OPENDIR
c04d4f40 118static void *squidaio_do_opendir(squidaio_request_t *);
cd748f27 119#endif
c04d4f40 120static void squidaio_debug(squidaio_request_t *);
121static void squidaio_poll_queues(void);
cd748f27 122
c04d4f40 123static squidaio_thread_t *threads = NULL;
124static int squidaio_initialised = 0;
cd748f27 125
58cd5bbd 126#define AIO_LARGE_BUFS 16384
127#define AIO_MEDIUM_BUFS AIO_LARGE_BUFS >> 1
128#define AIO_SMALL_BUFS AIO_LARGE_BUFS >> 2
129#define AIO_TINY_BUFS AIO_LARGE_BUFS >> 3
55f0e6f7 130#define AIO_MICRO_BUFS 128
58cd5bbd 131
a3efa961 132static MemAllocator *squidaio_large_bufs = NULL; /* 16K */
133static MemAllocator *squidaio_medium_bufs = NULL; /* 8K */
134static MemAllocator *squidaio_small_bufs = NULL; /* 4K */
135static MemAllocator *squidaio_tiny_bufs = NULL; /* 2K */
136static MemAllocator *squidaio_micro_bufs = NULL; /* 128K */
58cd5bbd 137
cd748f27 138static int request_queue_len = 0;
a3efa961 139static MemAllocator *squidaio_request_pool = NULL;
140static MemAllocator *squidaio_thread_pool = NULL;
c04d4f40 141static squidaio_request_queue_t request_queue;
62e76326 142
26ac0430 143static struct {
c04d4f40 144 squidaio_request_t *head, **tailp;
62e76326 145}
f0debecb 146
62e76326 147request_queue2 = {
148
26ac0430
AJ
149 NULL, &request_queue2.head
150};
c04d4f40 151static squidaio_request_queue_t done_queue;
62e76326 152
26ac0430 153static struct {
c04d4f40 154 squidaio_request_t *head, **tailp;
62e76326 155}
f0debecb 156
62e76326 157done_requests = {
158
26ac0430
AJ
159 NULL, &done_requests.head
160};
cd748f27 161static pthread_attr_t globattr;
5e5c622b 162#if HAVE_SCHED_H
62e76326 163
cd748f27 164static struct sched_param globsched;
5e5c622b 165#endif
cd748f27 166static pthread_t main_thread;
167
a3efa961 168static MemAllocator *
c04d4f40 169squidaio_get_pool(int size)
58cd5bbd 170{
58cd5bbd 171 if (size <= AIO_LARGE_BUFS) {
62e76326 172 if (size <= AIO_MICRO_BUFS)
a3efa961 173 return squidaio_micro_bufs;
62e76326 174 else if (size <= AIO_TINY_BUFS)
a3efa961 175 return squidaio_tiny_bufs;
62e76326 176 else if (size <= AIO_SMALL_BUFS)
a3efa961 177 return squidaio_small_bufs;
62e76326 178 else if (size <= AIO_MEDIUM_BUFS)
a3efa961 179 return squidaio_medium_bufs;
62e76326 180 else
a3efa961 181 return squidaio_large_bufs;
182 }
62e76326 183
a3efa961 184 return NULL;
58cd5bbd 185}
186
211f1d0b 187void *
c04d4f40 188squidaio_xmalloc(int size)
58cd5bbd 189{
190 void *p;
a3efa961 191 MemAllocator *pool;
58cd5bbd 192
c04d4f40 193 if ((pool = squidaio_get_pool(size)) != NULL) {
b001e822 194 p = pool->alloc();
58cd5bbd 195 } else
62e76326 196 p = xmalloc(size);
58cd5bbd 197
198 return p;
199}
200
55f0e6f7 201static char *
c04d4f40 202squidaio_xstrdup(const char *str)
55f0e6f7 203{
204 char *p;
f0debecb 205 int len = strlen(str) + 1;
55f0e6f7 206
e6ccf245 207 p = (char *)squidaio_xmalloc(len);
55f0e6f7 208 strncpy(p, str, len);
209
210 return p;
211}
212
211f1d0b 213void
c04d4f40 214squidaio_xfree(void *p, int size)
58cd5bbd 215{
a3efa961 216 MemAllocator *pool;
58cd5bbd 217
c04d4f40 218 if ((pool = squidaio_get_pool(size)) != NULL) {
dc47f531 219 pool->freeOne(p);
58cd5bbd 220 } else
62e76326 221 xfree(p);
58cd5bbd 222}
223
55f0e6f7 224static void
c04d4f40 225squidaio_xstrfree(char *str)
55f0e6f7 226{
a3efa961 227 MemAllocator *pool;
f0debecb 228 int len = strlen(str) + 1;
55f0e6f7 229
c04d4f40 230 if ((pool = squidaio_get_pool(len)) != NULL) {
dc47f531 231 pool->freeOne(str);
55f0e6f7 232 } else
62e76326 233 xfree(str);
55f0e6f7 234}
235
d06925a4 236void
c04d4f40 237squidaio_init(void)
cd748f27 238{
239 int i;
c04d4f40 240 squidaio_thread_t *threadp;
cd748f27 241
c04d4f40 242 if (squidaio_initialised)
62e76326 243 return;
cd748f27 244
245 pthread_attr_init(&globattr);
62e76326 246
cd748f27 247#if HAVE_PTHREAD_ATTR_SETSCOPE
62e76326 248
cd748f27 249 pthread_attr_setscope(&globattr, PTHREAD_SCOPE_SYSTEM);
62e76326 250
cd748f27 251#endif
5e5c622b 252#if HAVE_SCHED_H
62e76326 253
cd748f27 254 globsched.sched_priority = 1;
62e76326 255
5e5c622b 256#endif
62e76326 257
cd748f27 258 main_thread = pthread_self();
62e76326 259
5e5c622b 260#if HAVE_SCHED_H && HAVE_PTHREAD_SETSCHEDPARAM
62e76326 261
cd748f27 262 pthread_setschedparam(main_thread, SCHED_OTHER, &globsched);
62e76326 263
cd748f27 264#endif
5e5c622b 265#if HAVE_SCHED_H
62e76326 266
cd748f27 267 globsched.sched_priority = 2;
62e76326 268
5e5c622b 269#endif
270#if HAVE_SCHED_H && HAVE_PTHREAD_ATTR_SETSCHEDPARAM
62e76326 271
cd748f27 272 pthread_attr_setschedparam(&globattr, &globsched);
62e76326 273
cd748f27 274#endif
275
5a3501d2 276 /* Give each thread a smaller 256KB stack, should be more than sufficient */
277 pthread_attr_setstacksize(&globattr, 256 * 1024);
278
55f0e6f7 279 /* Initialize request queue */
280 if (pthread_mutex_init(&(request_queue.mutex), NULL))
62e76326 281 fatal("Failed to create mutex");
282
55f0e6f7 283 if (pthread_cond_init(&(request_queue.cond), NULL))
62e76326 284 fatal("Failed to create condition variable");
285
55f0e6f7 286 request_queue.head = NULL;
62e76326 287
55f0e6f7 288 request_queue.tailp = &request_queue.head;
62e76326 289
55f0e6f7 290 request_queue.requests = 0;
62e76326 291
55f0e6f7 292 request_queue.blocked = 0;
293
294 /* Initialize done queue */
295 if (pthread_mutex_init(&(done_queue.mutex), NULL))
62e76326 296 fatal("Failed to create mutex");
297
55f0e6f7 298 if (pthread_cond_init(&(done_queue.cond), NULL))
62e76326 299 fatal("Failed to create condition variable");
300
55f0e6f7 301 done_queue.head = NULL;
62e76326 302
55f0e6f7 303 done_queue.tailp = &done_queue.head;
62e76326 304
55f0e6f7 305 done_queue.requests = 0;
62e76326 306
55f0e6f7 307 done_queue.blocked = 0;
cd748f27 308
55f0e6f7 309 /* Create threads and get them to sit in their wait loop */
04eb0689 310 squidaio_thread_pool = memPoolCreate("aio_thread", sizeof(squidaio_thread_t));
62e76326 311
9fbbe256 312 assert(NUMTHREADS);
62e76326 313
cb4185f1 314 for (i = 0; i < NUMTHREADS; ++i) {
b001e822 315 threadp = (squidaio_thread_t *)squidaio_thread_pool->alloc();
62e76326 316 threadp->status = _THREAD_STARTING;
317 threadp->current_req = NULL;
318 threadp->requests = 0;
319 threadp->next = threads;
320 threads = threadp;
321
322 if (pthread_create(&threadp->thread, &globattr, squidaio_thread_loop, threadp)) {
323 fprintf(stderr, "Thread creation failed\n");
324 threadp->status = _THREAD_FAILED;
325 continue;
326 }
cd748f27 327 }
328
329 /* Create request pool */
04eb0689 330 squidaio_request_pool = memPoolCreate("aio_request", sizeof(squidaio_request_t));
62e76326 331
04eb0689 332 squidaio_large_bufs = memPoolCreate("squidaio_large_bufs", AIO_LARGE_BUFS);
62e76326 333
04eb0689 334 squidaio_medium_bufs = memPoolCreate("squidaio_medium_bufs", AIO_MEDIUM_BUFS);
62e76326 335
04eb0689 336 squidaio_small_bufs = memPoolCreate("squidaio_small_bufs", AIO_SMALL_BUFS);
62e76326 337
04eb0689 338 squidaio_tiny_bufs = memPoolCreate("squidaio_tiny_bufs", AIO_TINY_BUFS);
62e76326 339
04eb0689 340 squidaio_micro_bufs = memPoolCreate("squidaio_micro_bufs", AIO_MICRO_BUFS);
c04d4f40 341
342 squidaio_initialised = 1;
cd748f27 343}
344
d06925a4 345void
346squidaio_shutdown(void)
347{
348 if (!squidaio_initialised)
349 return;
350
351 /* This is the same as in squidaio_sync */
352 do {
353 squidaio_poll_queues();
354 } while (request_queue_len > 0);
355
356 CommIO::NotifyIOClose();
357
358 squidaio_initialised = 0;
359}
cd748f27 360
59a09b98 361void *
c04d4f40 362squidaio_thread_loop(void *ptr)
cd748f27 363{
e6ccf245 364 squidaio_thread_t *threadp = (squidaio_thread_t *)ptr;
c04d4f40 365 squidaio_request_t *request;
e6ccf245 366 sigset_t newSig;
cd748f27 367
368 /*
369 * Make sure to ignore signals which may possibly get sent to
370 * the parent squid thread. Causes havoc with mutex's and
371 * condition waits otherwise
372 */
373
e6ccf245 374 sigemptyset(&newSig);
375 sigaddset(&newSig, SIGPIPE);
376 sigaddset(&newSig, SIGCHLD);
605f2c3e 377#if defined(_SQUID_LINUX_THREADS_)
62e76326 378
e6ccf245 379 sigaddset(&newSig, SIGQUIT);
380 sigaddset(&newSig, SIGTRAP);
cd748f27 381#else
62e76326 382
e6ccf245 383 sigaddset(&newSig, SIGUSR1);
384 sigaddset(&newSig, SIGUSR2);
cd748f27 385#endif
62e76326 386
e6ccf245 387 sigaddset(&newSig, SIGHUP);
388 sigaddset(&newSig, SIGTERM);
389 sigaddset(&newSig, SIGINT);
390 sigaddset(&newSig, SIGALRM);
391 pthread_sigmask(SIG_BLOCK, &newSig, NULL);
cd748f27 392
cd748f27 393 while (1) {
62e76326 394 threadp->current_req = request = NULL;
395 request = NULL;
396 /* Get a request to process */
397 threadp->status = _THREAD_WAITING;
398 pthread_mutex_lock(&request_queue.mutex);
399
400 while (!request_queue.head) {
401 pthread_cond_wait(&request_queue.cond, &request_queue.mutex);
402 }
403
404 request = request_queue.head;
405
406 if (request)
407 request_queue.head = request->next;
408
409 if (!request_queue.head)
410 request_queue.tailp = &request_queue.head;
411
412 pthread_mutex_unlock(&request_queue.mutex);
413
414 /* process the request */
415 threadp->status = _THREAD_BUSY;
416
417 request->next = NULL;
418
419 threadp->current_req = request;
420
421 errno = 0;
422
423 if (!request->cancelled) {
424 switch (request->request_type) {
425
426 case _AIO_OP_OPEN:
427 squidaio_do_open(request);
428 break;
429
430 case _AIO_OP_READ:
431 squidaio_do_read(request);
432 break;
433
434 case _AIO_OP_WRITE:
435 squidaio_do_write(request);
436 break;
437
438 case _AIO_OP_CLOSE:
439 squidaio_do_close(request);
440 break;
441
efa3acd1 442 case _AIO_OP_UNLINK:
443 squidaio_do_unlink(request);
444 break;
445
cd748f27 446#if AIO_OPENDIR /* Opendir not implemented yet */
62e76326 447
448 case _AIO_OP_OPENDIR:
449 squidaio_do_opendir(request);
450 break;
cd748f27 451#endif
62e76326 452
453 case _AIO_OP_STAT:
454 squidaio_do_stat(request);
455 break;
456
457 default:
458 request->ret = -1;
459 request->err = EINVAL;
460 break;
461 }
462 } else { /* cancelled */
463 request->ret = -1;
464 request->err = EINTR;
465 }
466
467 threadp->status = _THREAD_DONE;
468 /* put the request in the done queue */
469 pthread_mutex_lock(&done_queue.mutex);
470 *done_queue.tailp = request;
471 done_queue.tailp = &request->next;
472 pthread_mutex_unlock(&done_queue.mutex);
473 CommIO::NotifyIOCompleted();
cb4185f1 474 ++ threadp->requests;
55f0e6f7 475 } /* while forever */
62e76326 476
cd748f27 477 return NULL;
c04d4f40 478} /* squidaio_thread_loop */
cd748f27 479
480static void
c04d4f40 481squidaio_queue_request(squidaio_request_t * request)
cd748f27 482{
cd748f27 483 static int high_start = 0;
bf8fe701 484 debugs(43, 9, "squidaio_queue_request: " << request << " type=" << request->request_type << " result=" << request->resultp);
cd748f27 485 /* Mark it as not executed (failing result, no error) */
55f0e6f7 486 request->ret = -1;
487 request->err = 0;
488 /* Internal housekeeping */
489 request_queue_len += 1;
490 request->resultp->_data = request;
491 /* Play some tricks with the request_queue2 queue */
492 request->next = NULL;
62e76326 493
40173038 494 if (pthread_mutex_trylock(&request_queue.mutex) == 0) {
62e76326 495 if (request_queue2.head) {
496 /* Grab blocked requests */
497 *request_queue.tailp = request_queue2.head;
498 request_queue.tailp = request_queue2.tailp;
499 }
500
501 /* Enqueue request */
502 *request_queue.tailp = request;
503
504 request_queue.tailp = &request->next;
505
506 pthread_cond_signal(&request_queue.cond);
507
508 pthread_mutex_unlock(&request_queue.mutex);
509
510 if (request_queue2.head) {
511 /* Clear queue of blocked requests */
512 request_queue2.head = NULL;
513 request_queue2.tailp = &request_queue2.head;
514 }
40173038 515 } else {
62e76326 516 /* Oops, the request queue is blocked, use request_queue2 */
517 *request_queue2.tailp = request;
518 request_queue2.tailp = &request->next;
55f0e6f7 519 }
62e76326 520
55f0e6f7 521 if (request_queue2.head) {
62e76326 522 static int filter = 0;
523 static int filter_limit = 8;
524
525 if (++filter >= filter_limit) {
526 filter_limit += filter;
527 filter = 0;
e0236918 528 debugs(43, DBG_IMPORTANT, "squidaio_queue_request: WARNING - Queue congestion");
62e76326 529 }
cd748f27 530 }
62e76326 531
cd748f27 532 /* Warn if out of threads */
55f0e6f7 533 if (request_queue_len > MAGIC1) {
62e76326 534 static int last_warn = 0;
535 static int queue_high, queue_low;
536
537 if (high_start == 0) {
538 high_start = squid_curtime;
539 queue_high = request_queue_len;
540 queue_low = request_queue_len;
541 }
542
543 if (request_queue_len > queue_high)
544 queue_high = request_queue_len;
545
546 if (request_queue_len < queue_low)
547 queue_low = request_queue_len;
548
549 if (squid_curtime >= (last_warn + 15) &&
550 squid_curtime >= (high_start + 5)) {
e0236918 551 debugs(43, DBG_IMPORTANT, "squidaio_queue_request: WARNING - Disk I/O overloading");
62e76326 552
553 if (squid_curtime >= (high_start + 15))
e0236918 554 debugs(43, DBG_IMPORTANT, "squidaio_queue_request: Queue Length: current=" <<
bf8fe701 555 request_queue_len << ", high=" << queue_high <<
556 ", low=" << queue_low << ", duration=" <<
557 (long int) (squid_curtime - high_start));
62e76326 558
559 last_warn = squid_curtime;
560 }
cd748f27 561 } else {
62e76326 562 high_start = 0;
cd748f27 563 }
62e76326 564
55f0e6f7 565 /* Warn if seriously overloaded */
cd748f27 566 if (request_queue_len > RIDICULOUS_LENGTH) {
fa84c01d
FC
567 debugs(43, DBG_CRITICAL, "squidaio_queue_request: Async request queue growing uncontrollably!");
568 debugs(43, DBG_CRITICAL, "squidaio_queue_request: Syncing pending I/O operations.. (blocking)");
62e76326 569 squidaio_sync();
fa84c01d 570 debugs(43, DBG_CRITICAL, "squidaio_queue_request: Synced");
cd748f27 571 }
c04d4f40 572} /* squidaio_queue_request */
cd748f27 573
cd748f27 574static void
c04d4f40 575squidaio_cleanup_request(squidaio_request_t * requestp)
cd748f27 576{
c04d4f40 577 squidaio_result_t *resultp = requestp->resultp;
cd748f27 578 int cancelled = requestp->cancelled;
579
580 /* Free allocated structures and copy data back to user space if the */
581 /* request hasn't been cancelled */
62e76326 582
cd748f27 583 switch (requestp->request_type) {
62e76326 584
cd748f27 585 case _AIO_OP_STAT:
62e76326 586
587 if (!cancelled && requestp->ret == 0)
41d00cd3 588 memcpy(requestp->statp, requestp->tmpstatp, sizeof(struct stat));
62e76326 589
590 squidaio_xfree(requestp->tmpstatp, sizeof(struct stat));
591
592 squidaio_xstrfree(requestp->path);
593
594 break;
595
cd748f27 596 case _AIO_OP_OPEN:
62e76326 597 if (cancelled && requestp->ret >= 0)
598 /* The open() was cancelled but completed */
599 close(requestp->ret);
600
601 squidaio_xstrfree(requestp->path);
602
603 break;
604
cd748f27 605 case _AIO_OP_CLOSE:
62e76326 606 if (cancelled && requestp->ret < 0)
607 /* The close() was cancelled and never got executed */
608 close(requestp->fd);
609
610 break;
611
cd748f27 612 case _AIO_OP_UNLINK:
62e76326 613
cd748f27 614 case _AIO_OP_OPENDIR:
62e76326 615 squidaio_xstrfree(requestp->path);
616
617 break;
618
cd748f27 619 case _AIO_OP_READ:
62e76326 620 break;
621
cd748f27 622 case _AIO_OP_WRITE:
62e76326 623 break;
624
cd748f27 625 default:
62e76326 626 break;
cd748f27 627 }
62e76326 628
cd748f27 629 if (resultp != NULL && !cancelled) {
62e76326 630 resultp->aio_return = requestp->ret;
631 resultp->aio_errno = requestp->err;
cd748f27 632 }
62e76326 633
dc47f531 634 squidaio_request_pool->freeOne(requestp);
c04d4f40 635} /* squidaio_cleanup_request */
cd748f27 636
cd748f27 637int
c04d4f40 638squidaio_cancel(squidaio_result_t * resultp)
cd748f27 639{
e6ccf245 640 squidaio_request_t *request = (squidaio_request_t *)resultp->_data;
cd748f27 641
55f0e6f7 642 if (request && request->resultp == resultp) {
bf8fe701 643 debugs(43, 9, "squidaio_cancel: " << request << " type=" << request->request_type << " result=" << request->resultp);
62e76326 644 request->cancelled = 1;
645 request->resultp = NULL;
646 resultp->_data = NULL;
647 resultp->result_type = _AIO_OP_NONE;
648 return 0;
55f0e6f7 649 }
62e76326 650
cd748f27 651 return 1;
c04d4f40 652} /* squidaio_cancel */
cd748f27 653
cd748f27 654int
c04d4f40 655squidaio_open(const char *path, int oflag, mode_t mode, squidaio_result_t * resultp)
cd748f27 656{
e65c313f 657 squidaio_init();
c04d4f40 658 squidaio_request_t *requestp;
cd748f27 659
b001e822 660 requestp = (squidaio_request_t *)squidaio_request_pool->alloc();
62e76326 661
c04d4f40 662 requestp->path = (char *) squidaio_xstrdup(path);
62e76326 663
cd748f27 664 requestp->oflag = oflag;
62e76326 665
cd748f27 666 requestp->mode = mode;
62e76326 667
cd748f27 668 requestp->resultp = resultp;
62e76326 669
cd748f27 670 requestp->request_type = _AIO_OP_OPEN;
62e76326 671
cd748f27 672 requestp->cancelled = 0;
62e76326 673
12e137b0 674 resultp->result_type = _AIO_OP_OPEN;
cd748f27 675
c04d4f40 676 squidaio_queue_request(requestp);
62e76326 677
cd748f27 678 return 0;
679}
680
cd748f27 681static void
c04d4f40 682squidaio_do_open(squidaio_request_t * requestp)
cd748f27 683{
684 requestp->ret = open(requestp->path, requestp->oflag, requestp->mode);
685 requestp->err = errno;
686}
687
cd748f27 688int
ee139403 689squidaio_read(int fd, char *bufp, size_t bufs, off_t offset, int whence, squidaio_result_t * resultp)
cd748f27 690{
c04d4f40 691 squidaio_request_t *requestp;
cd748f27 692
b001e822 693 requestp = (squidaio_request_t *)squidaio_request_pool->alloc();
62e76326 694
cd748f27 695 requestp->fd = fd;
62e76326 696
cd748f27 697 requestp->bufferp = bufp;
62e76326 698
cd748f27 699 requestp->buflen = bufs;
62e76326 700
cd748f27 701 requestp->offset = offset;
62e76326 702
cd748f27 703 requestp->whence = whence;
62e76326 704
cd748f27 705 requestp->resultp = resultp;
62e76326 706
cd748f27 707 requestp->request_type = _AIO_OP_READ;
62e76326 708
cd748f27 709 requestp->cancelled = 0;
62e76326 710
12e137b0 711 resultp->result_type = _AIO_OP_READ;
cd748f27 712
c04d4f40 713 squidaio_queue_request(requestp);
62e76326 714
cd748f27 715 return 0;
716}
717
cd748f27 718static void
c04d4f40 719squidaio_do_read(squidaio_request_t * requestp)
cd748f27 720{
721 lseek(requestp->fd, requestp->offset, requestp->whence);
211f1d0b 722 requestp->ret = read(requestp->fd, requestp->bufferp, requestp->buflen);
cd748f27 723 requestp->err = errno;
724}
725
cd748f27 726int
ee139403 727squidaio_write(int fd, char *bufp, size_t bufs, off_t offset, int whence, squidaio_result_t * resultp)
cd748f27 728{
c04d4f40 729 squidaio_request_t *requestp;
cd748f27 730
b001e822 731 requestp = (squidaio_request_t *)squidaio_request_pool->alloc();
62e76326 732
cd748f27 733 requestp->fd = fd;
62e76326 734
d06925a4 735 requestp->bufferp = bufp;
62e76326 736
cd748f27 737 requestp->buflen = bufs;
62e76326 738
cd748f27 739 requestp->offset = offset;
62e76326 740
cd748f27 741 requestp->whence = whence;
62e76326 742
cd748f27 743 requestp->resultp = resultp;
62e76326 744
cd748f27 745 requestp->request_type = _AIO_OP_WRITE;
62e76326 746
cd748f27 747 requestp->cancelled = 0;
62e76326 748
12e137b0 749 resultp->result_type = _AIO_OP_WRITE;
cd748f27 750
c04d4f40 751 squidaio_queue_request(requestp);
62e76326 752
cd748f27 753 return 0;
754}
755
cd748f27 756static void
c04d4f40 757squidaio_do_write(squidaio_request_t * requestp)
cd748f27 758{
d06925a4 759 requestp->ret = write(requestp->fd, requestp->bufferp, requestp->buflen);
cd748f27 760 requestp->err = errno;
761}
762
cd748f27 763int
c04d4f40 764squidaio_close(int fd, squidaio_result_t * resultp)
cd748f27 765{
c04d4f40 766 squidaio_request_t *requestp;
cd748f27 767
b001e822 768 requestp = (squidaio_request_t *)squidaio_request_pool->alloc();
62e76326 769
cd748f27 770 requestp->fd = fd;
62e76326 771
cd748f27 772 requestp->resultp = resultp;
62e76326 773
cd748f27 774 requestp->request_type = _AIO_OP_CLOSE;
62e76326 775
cd748f27 776 requestp->cancelled = 0;
62e76326 777
12e137b0 778 resultp->result_type = _AIO_OP_CLOSE;
cd748f27 779
c04d4f40 780 squidaio_queue_request(requestp);
62e76326 781
cd748f27 782 return 0;
783}
784
cd748f27 785static void
c04d4f40 786squidaio_do_close(squidaio_request_t * requestp)
cd748f27 787{
788 requestp->ret = close(requestp->fd);
789 requestp->err = errno;
790}
791
cd748f27 792int
62e76326 793
c04d4f40 794squidaio_stat(const char *path, struct stat *sb, squidaio_result_t * resultp)
cd748f27 795{
e65c313f 796 squidaio_init();
c04d4f40 797 squidaio_request_t *requestp;
cd748f27 798
b001e822 799 requestp = (squidaio_request_t *)squidaio_request_pool->alloc();
62e76326 800
c04d4f40 801 requestp->path = (char *) squidaio_xstrdup(path);
62e76326 802
cd748f27 803 requestp->statp = sb;
62e76326 804
c04d4f40 805 requestp->tmpstatp = (struct stat *) squidaio_xmalloc(sizeof(struct stat));
62e76326 806
cd748f27 807 requestp->resultp = resultp;
62e76326 808
cd748f27 809 requestp->request_type = _AIO_OP_STAT;
62e76326 810
cd748f27 811 requestp->cancelled = 0;
62e76326 812
12e137b0 813 resultp->result_type = _AIO_OP_STAT;
cd748f27 814
c04d4f40 815 squidaio_queue_request(requestp);
62e76326 816
cd748f27 817 return 0;
818}
819
cd748f27 820static void
c04d4f40 821squidaio_do_stat(squidaio_request_t * requestp)
cd748f27 822{
823 requestp->ret = stat(requestp->path, requestp->tmpstatp);
824 requestp->err = errno;
825}
826
15a47d1d 827int
efa3acd1 828squidaio_unlink(const char *path, squidaio_result_t * resultp)
15a47d1d 829{
e65c313f 830 squidaio_init();
c04d4f40 831 squidaio_request_t *requestp;
15a47d1d 832
b001e822 833 requestp = (squidaio_request_t *)squidaio_request_pool->alloc();
62e76326 834
efa3acd1 835 requestp->path = squidaio_xstrdup(path);
62e76326 836
15a47d1d 837 requestp->resultp = resultp;
62e76326 838
efa3acd1 839 requestp->request_type = _AIO_OP_UNLINK;
62e76326 840
15a47d1d 841 requestp->cancelled = 0;
62e76326 842
efa3acd1 843 resultp->result_type = _AIO_OP_UNLINK;
15a47d1d 844
c04d4f40 845 squidaio_queue_request(requestp);
62e76326 846
15a47d1d 847 return 0;
848}
849
15a47d1d 850static void
efa3acd1 851squidaio_do_unlink(squidaio_request_t * requestp)
15a47d1d 852{
efa3acd1 853 requestp->ret = unlink(requestp->path);
15a47d1d 854 requestp->err = errno;
855}
856
cd748f27 857#if AIO_OPENDIR
c04d4f40 858/* XXX squidaio_opendir NOT implemented yet.. */
cd748f27 859
860int
c04d4f40 861squidaio_opendir(const char *path, squidaio_result_t * resultp)
cd748f27 862{
c04d4f40 863 squidaio_request_t *requestp;
cd748f27 864 int len;
865
b001e822 866 requestp = squidaio_request_pool->alloc();
62e76326 867
12e137b0 868 resultp->result_type = _AIO_OP_OPENDIR;
62e76326 869
cd748f27 870 return -1;
871}
872
873static void
c04d4f40 874squidaio_do_opendir(squidaio_request_t * requestp)
cd748f27 875{
876 /* NOT IMPLEMENTED */
877}
878
879#endif
880
55f0e6f7 881static void
c04d4f40 882squidaio_poll_queues(void)
55f0e6f7 883{
884 /* kick "overflow" request queue */
62e76326 885
55f0e6f7 886 if (request_queue2.head &&
62e76326 887 pthread_mutex_trylock(&request_queue.mutex) == 0) {
888 *request_queue.tailp = request_queue2.head;
889 request_queue.tailp = request_queue2.tailp;
890 pthread_cond_signal(&request_queue.cond);
891 pthread_mutex_unlock(&request_queue.mutex);
892 request_queue2.head = NULL;
893 request_queue2.tailp = &request_queue2.head;
55f0e6f7 894 }
62e76326 895
55f0e6f7 896 /* poll done queue */
897 if (done_queue.head && pthread_mutex_trylock(&done_queue.mutex) == 0) {
62e76326 898
899 struct squidaio_request_t *requests = done_queue.head;
900 done_queue.head = NULL;
901 done_queue.tailp = &done_queue.head;
902 pthread_mutex_unlock(&done_queue.mutex);
903 *done_requests.tailp = requests;
904 request_queue_len -= 1;
905
906 while (requests->next) {
907 requests = requests->next;
908 request_queue_len -= 1;
909 }
910
911 done_requests.tailp = &requests->next;
55f0e6f7 912 }
55f0e6f7 913}
cd748f27 914
c04d4f40 915squidaio_result_t *
916squidaio_poll_done(void)
cd748f27 917{
c04d4f40 918 squidaio_request_t *request;
919 squidaio_result_t *resultp;
cd748f27 920 int cancelled;
55f0e6f7 921 int polled = 0;
cd748f27 922
62e76326 923AIO_REPOLL:
55f0e6f7 924 request = done_requests.head;
62e76326 925
55f0e6f7 926 if (request == NULL && !polled) {
62e76326 927 CommIO::ResetNotifications();
928 squidaio_poll_queues();
929 polled = 1;
930 request = done_requests.head;
55f0e6f7 931 }
62e76326 932
55f0e6f7 933 if (!request) {
62e76326 934 return NULL;
cd748f27 935 }
62e76326 936
bf8fe701 937 debugs(43, 9, "squidaio_poll_done: " << request << " type=" << request->request_type << " result=" << request->resultp);
55f0e6f7 938 done_requests.head = request->next;
62e76326 939
55f0e6f7 940 if (!done_requests.head)
62e76326 941 done_requests.tailp = &done_requests.head;
942
55f0e6f7 943 resultp = request->resultp;
62e76326 944
55f0e6f7 945 cancelled = request->cancelled;
62e76326 946
c04d4f40 947 squidaio_debug(request);
62e76326 948
bf8fe701 949 debugs(43, 5, "DONE: " << request->ret << " -> " << request->err);
62e76326 950
c04d4f40 951 squidaio_cleanup_request(request);
62e76326 952
cd748f27 953 if (cancelled)
62e76326 954 goto AIO_REPOLL;
955
cd748f27 956 return resultp;
c04d4f40 957} /* squidaio_poll_done */
cd748f27 958
959int
c04d4f40 960squidaio_operations_pending(void)
cd748f27 961{
55f0e6f7 962 return request_queue_len + (done_requests.head ? 1 : 0);
cd748f27 963}
964
965int
c04d4f40 966squidaio_sync(void)
cd748f27 967{
55f0e6f7 968 /* XXX This might take a while if the queue is large.. */
62e76326 969
cd748f27 970 do {
62e76326 971 squidaio_poll_queues();
cd748f27 972 } while (request_queue_len > 0);
62e76326 973
c04d4f40 974 return squidaio_operations_pending();
cd748f27 975}
976
977int
c04d4f40 978squidaio_get_queue_len(void)
cd748f27 979{
980 return request_queue_len;
981}
982
983static void
c04d4f40 984squidaio_debug(squidaio_request_t * request)
cd748f27 985{
55f0e6f7 986 switch (request->request_type) {
62e76326 987
cd748f27 988 case _AIO_OP_OPEN:
bf8fe701 989 debugs(43, 5, "OPEN of " << request->path << " to FD " << request->ret);
62e76326 990 break;
991
cd748f27 992 case _AIO_OP_READ:
bf8fe701 993 debugs(43, 5, "READ on fd: " << request->fd);
62e76326 994 break;
995
cd748f27 996 case _AIO_OP_WRITE:
bf8fe701 997 debugs(43, 5, "WRITE on fd: " << request->fd);
62e76326 998 break;
999
cd748f27 1000 case _AIO_OP_CLOSE:
bf8fe701 1001 debugs(43, 5, "CLOSE of fd: " << request->fd);
62e76326 1002 break;
1003
cd748f27 1004 case _AIO_OP_UNLINK:
bf8fe701 1005 debugs(43, 5, "UNLINK of " << request->path);
62e76326 1006 break;
1007
cd748f27 1008 default:
62e76326 1009 break;
cd748f27 1010 }
1011}
b0465494 1012
1013void
1014squidaio_stats(StoreEntry * sentry)
1015{
1016 squidaio_thread_t *threadp;
1017 int i;
1018
1019 if (!squidaio_initialised)
1020 return;
1021
1022 storeAppendPrintf(sentry, "\n\nThreads Status:\n");
1023
1024 storeAppendPrintf(sentry, "#\tID\t# Requests\n");
1025
1026 threadp = threads;
1027
cb4185f1 1028 for (i = 0; i < NUMTHREADS; ++i) {
691aef5a 1029 storeAppendPrintf(sentry, "%i\t0x%lx\t%ld\n", i + 1, (unsigned long)threadp->thread, threadp->requests);
b0465494 1030 threadp = threadp->next;
1031 }
1032}