Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 1 | /* |
2 | * File descriptors management functions. | ||||
3 | * | ||||
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 4 | * Copyright 2000-2007 Willy Tarreau <w@1wt.eu> |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 5 | * |
6 | * This program is free software; you can redistribute it and/or | ||||
7 | * modify it under the terms of the GNU General Public License | ||||
8 | * as published by the Free Software Foundation; either version | ||||
9 | * 2 of the License, or (at your option) any later version. | ||||
10 | * | ||||
11 | */ | ||||
12 | |||||
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 13 | #include <string.h> |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 14 | #include <unistd.h> |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 15 | #include <sys/types.h> |
16 | |||||
Willy Tarreau | 2dd0d47 | 2006-06-29 17:53:05 +0200 | [diff] [blame] | 17 | #include <common/compat.h> |
18 | #include <common/config.h> | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 19 | |
20 | #include <types/fd.h> | ||||
21 | #include <types/global.h> | ||||
22 | |||||
Willy Tarreau | 2a42950 | 2006-10-15 14:52:29 +0200 | [diff] [blame] | 23 | #include <proto/fd.h> |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 24 | |
25 | struct fdtab *fdtab = NULL; /* array of all the file descriptors */ | ||||
26 | int maxfd; /* # of the highest fd + 1 */ | ||||
27 | int totalconn; /* total # of terminated sessions */ | ||||
28 | int actconn; /* # of active sessions */ | ||||
29 | |||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 30 | int cfg_polling_mechanism = 0; /* POLL_USE_{SELECT|POLL|EPOLL} */ |
31 | |||||
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 32 | struct poller pollers[MAX_POLLERS]; |
33 | struct poller cur_poller; | ||||
34 | int nbpollers = 0; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 35 | |
36 | |||||
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 37 | /********************* |
38 | * generic functions | ||||
39 | *********************/ | ||||
Willy Tarreau | 2a42950 | 2006-10-15 14:52:29 +0200 | [diff] [blame] | 40 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 41 | extern int select_register(struct poller *p); |
42 | #if defined(ENABLE_POLL) | ||||
43 | extern int poll_register(struct poller *p); | ||||
Willy Tarreau | 2a42950 | 2006-10-15 14:52:29 +0200 | [diff] [blame] | 44 | #endif |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 45 | #if defined(ENABLE_EPOLL) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 46 | extern int epoll_register(struct poller *p); |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 47 | #endif |
48 | |||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 49 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 50 | /* Deletes an FD from the fdsets, and recomputes the maxfd limit. |
51 | * The file descriptor is also closed. | ||||
52 | */ | ||||
53 | void fd_delete(int fd) | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 54 | { |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 55 | EV_FD_CLO(fd); |
56 | close(fd); | ||||
57 | fdtab[fd].state = FD_STCLOSE; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 58 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 59 | while ((maxfd-1 >= 0) && (fdtab[maxfd-1].state == FD_STCLOSE)) |
60 | maxfd--; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 61 | } |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 62 | |
63 | |||||
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 64 | /* registers all known pollers */ |
65 | void register_pollers() | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 66 | { |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 67 | if (select_register(&pollers[nbpollers])) |
68 | nbpollers++; | ||||
69 | #if defined(ENABLE_POLL) | ||||
70 | poll_register(&pollers[nbpollers]); | ||||
71 | nbpollers++; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 72 | #endif |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 73 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 74 | #if defined(ENABLE_EPOLL) |
75 | epoll_register(&pollers[nbpollers]); | ||||
76 | nbpollers++; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 77 | #endif |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 78 | } |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 79 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 80 | /* disable the specified poller */ |
81 | void disable_poller(const char *poller_name) | ||||
82 | { | ||||
83 | int p; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 84 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 85 | for (p = 0; p < nbpollers; p++) |
86 | if (strcmp(pollers[p].name, poller_name) == 0) | ||||
87 | pollers[p].pref = 0; | ||||
88 | } | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 89 | |
90 | /* | ||||
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 91 | * Initialize the pollers till the best one is found. |
92 | * If none works, returns 0, otherwise 1. | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 93 | */ |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 94 | int init_pollers() |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 95 | { |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 96 | int p; |
97 | struct poller *bp; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 98 | |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 99 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 100 | do { |
101 | bp = NULL; | ||||
102 | for (p = 0; p < nbpollers; p++) | ||||
103 | if (!bp || (pollers[p].pref > bp->pref)) | ||||
104 | bp = &pollers[p]; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 105 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 106 | if (!bp || bp->pref == 0) |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 107 | break; |
108 | |||||
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 109 | if (bp->init(bp)) { |
110 | memcpy(&cur_poller, bp, sizeof(*bp)); | ||||
111 | return 1; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 112 | } |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame^] | 113 | } while (!bp || bp->pref == 0); |
114 | return 0; | ||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 115 | } |
116 | |||||
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 117 | /* |
118 | * Local variables: | ||||
119 | * c-indent-level: 8 | ||||
120 | * c-basic-offset: 8 | ||||
121 | * End: | ||||
122 | */ |