diff options
Diffstat (limited to 'ot_mutex.c')
| -rw-r--r-- | ot_mutex.c | 221 |
1 files changed, 0 insertions, 221 deletions
| @@ -31,7 +31,6 @@ static pthread_mutex_t bucket_mutex[OT_BUCKET_COUNT]; | |||
| 31 | static _Atomic size_t g_torrent_count; | 31 | static _Atomic size_t g_torrent_count; |
| 32 | 32 | ||
| 33 | /* Self pipe from opentracker.c */ | 33 | /* Self pipe from opentracker.c */ |
| 34 | extern int g_self_pipe[2]; | ||
| 35 | 34 | ||
| 36 | ot_vector *mutex_bucket_lock(int bucket) { | 35 | ot_vector *mutex_bucket_lock(int bucket) { |
| 37 | pthread_mutex_lock(bucket_mutex + bucket); | 36 | pthread_mutex_lock(bucket_mutex + bucket); |
| @@ -60,237 +59,17 @@ void mutex_bucket_unlock_by_hash(ot_hash const hash, int delta_torrentcount) { | |||
| 60 | 59 | ||
| 61 | size_t mutex_get_torrent_count(void) { return atomic_load_explicit(&g_torrent_count, memory_order_relaxed); } | 60 | size_t mutex_get_torrent_count(void) { return atomic_load_explicit(&g_torrent_count, memory_order_relaxed); } |
| 62 | 61 | ||
| 63 | /* TaskQueue Magic */ | ||
| 64 | |||
| 65 | struct ot_task { | ||
| 66 | ot_taskid taskid; | ||
| 67 | ot_tasktype tasktype; | ||
| 68 | int64_t sock; | ||
| 69 | size_t iovec_entries; | ||
| 70 | struct iovec *iovec; | ||
| 71 | struct ot_task *next; | ||
| 72 | }; | ||
| 73 | |||
| 74 | static ot_taskid next_free_taskid = 1; | ||
| 75 | static struct ot_task *tasklist; | ||
| 76 | static pthread_mutex_t tasklist_mutex; | ||
| 77 | static pthread_cond_t tasklist_being_filled; | ||
| 78 | |||
| 79 | int mutex_workqueue_pushtask(int64_t sock, ot_tasktype tasktype) { | ||
| 80 | struct ot_task **tmptask, *task; | ||
| 81 | |||
| 82 | task = malloc(sizeof(struct ot_task)); | ||
| 83 | if (!task) | ||
| 84 | return -1; | ||
| 85 | |||
| 86 | task->taskid = 0; | ||
| 87 | task->tasktype = tasktype; | ||
| 88 | task->sock = sock; | ||
| 89 | task->iovec_entries = 0; | ||
| 90 | task->iovec = NULL; | ||
| 91 | task->next = 0; | ||
| 92 | |||
| 93 | /* Want exclusive access to tasklist */ | ||
| 94 | pthread_mutex_lock(&tasklist_mutex); | ||
| 95 | |||
| 96 | /* Skip to end of list */ | ||
| 97 | tmptask = &tasklist; | ||
| 98 | while (*tmptask) | ||
| 99 | tmptask = &(*tmptask)->next; | ||
| 100 | *tmptask = task; | ||
| 101 | |||
| 102 | /* Inform waiting workers and release lock */ | ||
| 103 | pthread_cond_broadcast(&tasklist_being_filled); | ||
| 104 | pthread_mutex_unlock(&tasklist_mutex); | ||
| 105 | return 0; | ||
| 106 | } | ||
| 107 | |||
| 108 | void mutex_workqueue_canceltask(int64_t sock) { | ||
| 109 | struct ot_task **task; | ||
| 110 | |||
| 111 | /* Want exclusive access to tasklist */ | ||
| 112 | pthread_mutex_lock(&tasklist_mutex); | ||
| 113 | |||
| 114 | for (task = &tasklist; *task; task = &((*task)->next)) | ||
| 115 | if ((*task)->sock == sock) { | ||
| 116 | struct iovec *iovec = (*task)->iovec; | ||
| 117 | struct ot_task *ptask = *task; | ||
| 118 | size_t i; | ||
| 119 | |||
| 120 | /* Free task's iovec, the buffers it points at and the vector itself */ | ||
| 121 | for (i = 0; i < (*task)->iovec_entries; ++i) | ||
| 122 | free(iovec[i].iov_base); | ||
| 123 | free(iovec); | ||
| 124 | |||
| 125 | *task = (*task)->next; | ||
| 126 | free(ptask); | ||
| 127 | break; | ||
| 128 | } | ||
| 129 | |||
| 130 | /* Release lock */ | ||
| 131 | pthread_mutex_unlock(&tasklist_mutex); | ||
| 132 | } | ||
| 133 | |||
| 134 | ot_taskid mutex_workqueue_poptask(ot_tasktype *tasktype) { | ||
| 135 | struct ot_task *task; | ||
| 136 | ot_taskid taskid = 0; | ||
| 137 | |||
| 138 | /* Want exclusive access to tasklist */ | ||
| 139 | pthread_mutex_lock(&tasklist_mutex); | ||
| 140 | |||
| 141 | while (!taskid) { | ||
| 142 | /* Skip to the first unassigned task this worker wants to do */ | ||
| 143 | for (task = tasklist; task; task = task->next) | ||
| 144 | if (!task->taskid && (TASK_CLASS_MASK & task->tasktype) == *tasktype) { | ||
| 145 | /* If we found an outstanding task, assign a taskid to it | ||
| 146 | and leave the loop */ | ||
| 147 | task->taskid = taskid = ++next_free_taskid; | ||
| 148 | *tasktype = task->tasktype; | ||
| 149 | break; | ||
| 150 | } | ||
| 151 | |||
| 152 | /* Wait until the next task is being fed, unless the tracker is on its | ||
| 153 | way out, in which case the worker is meant to leave rather than wait */ | ||
| 154 | if (!taskid) { | ||
| 155 | if (!g_opentracker_running) | ||
| 156 | break; | ||
| 157 | pthread_cond_wait(&tasklist_being_filled, &tasklist_mutex); | ||
| 158 | } | ||
| 159 | } | ||
| 160 | |||
| 161 | /* Release lock */ | ||
| 162 | pthread_mutex_unlock(&tasklist_mutex); | ||
| 163 | |||
| 164 | return taskid; | ||
| 165 | } | ||
| 166 | |||
| 167 | void mutex_workqueue_pushsuccess(ot_taskid taskid) { | ||
| 168 | struct ot_task **task; | ||
| 169 | |||
| 170 | /* Want exclusive access to tasklist */ | ||
| 171 | pthread_mutex_lock(&tasklist_mutex); | ||
| 172 | |||
| 173 | for (task = &tasklist; *task; task = &((*task)->next)) | ||
| 174 | if ((*task)->taskid == taskid) { | ||
| 175 | struct ot_task *ptask = *task; | ||
| 176 | *task = (*task)->next; | ||
| 177 | free(ptask); | ||
| 178 | break; | ||
| 179 | } | ||
| 180 | |||
| 181 | /* Release lock */ | ||
| 182 | pthread_mutex_unlock(&tasklist_mutex); | ||
| 183 | } | ||
| 184 | |||
| 185 | int mutex_workqueue_pushresult(ot_taskid taskid, size_t iovec_entries, struct iovec *iovec) { | ||
| 186 | struct ot_task *task; | ||
| 187 | const char byte = 'o'; | ||
| 188 | |||
| 189 | /* Want exclusive access to tasklist */ | ||
| 190 | pthread_mutex_lock(&tasklist_mutex); | ||
| 191 | |||
| 192 | for (task = tasklist; task; task = task->next) | ||
| 193 | if (task->taskid == taskid) { | ||
| 194 | task->iovec_entries = iovec_entries; | ||
| 195 | task->iovec = iovec; | ||
| 196 | task->tasktype = TASK_DONE; | ||
| 197 | break; | ||
| 198 | } | ||
| 199 | |||
| 200 | /* Release lock */ | ||
| 201 | pthread_mutex_unlock(&tasklist_mutex); | ||
| 202 | |||
| 203 | /* A full pipe has already done this nudge's job, so a failure is fine */ | ||
| 204 | (void)!write(g_self_pipe[1], &byte, 1); | ||
| 205 | |||
| 206 | /* Indicate whether the worker has to throw away results */ | ||
| 207 | return task ? 0 : -1; | ||
| 208 | } | ||
| 209 | |||
| 210 | int mutex_workqueue_pushchunked(ot_taskid taskid, struct iovec *iovec) { | ||
| 211 | struct ot_task *task; | ||
| 212 | const char byte = 'o'; | ||
| 213 | |||
| 214 | /* Want exclusive access to tasklist */ | ||
| 215 | pthread_mutex_lock(&tasklist_mutex); | ||
| 216 | |||
| 217 | for (task = tasklist; task; task = task->next) | ||
| 218 | if (task->taskid == taskid) { | ||
| 219 | if (iovec) { | ||
| 220 | if (iovec_append(&task->iovec_entries, &task->iovec, iovec)) | ||
| 221 | task->tasktype = TASK_DONE_PARTIAL; | ||
| 222 | else | ||
| 223 | task = NULL; | ||
| 224 | } else | ||
| 225 | task->tasktype = TASK_DONE; | ||
| 226 | break; | ||
| 227 | } | ||
| 228 | |||
| 229 | /* Release lock */ | ||
| 230 | pthread_mutex_unlock(&tasklist_mutex); | ||
| 231 | |||
| 232 | /* A full pipe has already done this nudge's job, so a failure is fine */ | ||
| 233 | (void)!write(g_self_pipe[1], &byte, 1); | ||
| 234 | |||
| 235 | /* Indicate whether the worker has to throw away results */ | ||
| 236 | return task ? 0 : -1; | ||
| 237 | } | ||
| 238 | |||
| 239 | int64_t mutex_workqueue_popresult(size_t *iovec_entries, struct iovec **iovec, int *is_partial) { | ||
| 240 | struct ot_task **task; | ||
| 241 | int64_t sock = -1; | ||
| 242 | |||
| 243 | *is_partial = 0; | ||
| 244 | |||
| 245 | /* Want exclusive access to tasklist */ | ||
| 246 | pthread_mutex_lock(&tasklist_mutex); | ||
| 247 | |||
| 248 | for (task = &tasklist; *task; task = &((*task)->next)) | ||
| 249 | if (((*task)->tasktype & TASK_CLASS_MASK) == TASK_DONE) { | ||
| 250 | struct ot_task *ptask = *task; | ||
| 251 | *iovec_entries = ptask->iovec_entries; | ||
| 252 | *iovec = ptask->iovec; | ||
| 253 | sock = ptask->sock; | ||
| 254 | |||
| 255 | if ((*task)->tasktype == TASK_DONE) { | ||
| 256 | *task = ptask->next; | ||
| 257 | free(ptask); | ||
| 258 | } else { | ||
| 259 | ptask->iovec_entries = 0; | ||
| 260 | ptask->iovec = NULL; | ||
| 261 | *is_partial = 1; | ||
| 262 | /* Prevent task from showing up immediately again unless new data was added */ | ||
| 263 | (*task)->tasktype = TASK_FULLSCRAPE; | ||
| 264 | } | ||
| 265 | break; | ||
| 266 | } | ||
| 267 | |||
| 268 | /* Release lock */ | ||
| 269 | pthread_mutex_unlock(&tasklist_mutex); | ||
| 270 | return sock; | ||
| 271 | } | ||
| 272 | |||
| 273 | void mutex_init(void) { | 62 | void mutex_init(void) { |
| 274 | size_t i; | 63 | size_t i; |
| 275 | pthread_mutex_init(&tasklist_mutex, NULL); | ||
| 276 | pthread_cond_init(&tasklist_being_filled, NULL); | ||
| 277 | for (i = 0; i < OT_BUCKET_COUNT; ++i) | 64 | for (i = 0; i < OT_BUCKET_COUNT; ++i) |
| 278 | pthread_mutex_init(bucket_mutex + i, NULL); | 65 | pthread_mutex_init(bucket_mutex + i, NULL); |
| 279 | memset(all_torrents, 0, sizeof(all_torrents)); | 66 | memset(all_torrents, 0, sizeof(all_torrents)); |
| 280 | } | 67 | } |
| 281 | 68 | ||
| 282 | /* Let every worker parked on the queue see that the tracker is stopping */ | 69 | /* Let every worker parked on the queue see that the tracker is stopping */ |
| 283 | void mutex_workqueue_wakeall(void) { | ||
| 284 | pthread_mutex_lock(&tasklist_mutex); | ||
| 285 | pthread_cond_broadcast(&tasklist_being_filled); | ||
| 286 | pthread_mutex_unlock(&tasklist_mutex); | ||
| 287 | } | ||
| 288 | |||
| 289 | void mutex_deinit(void) { | 70 | void mutex_deinit(void) { |
| 290 | size_t i; | 71 | size_t i; |
| 291 | for (i = 0; i < OT_BUCKET_COUNT; ++i) | 72 | for (i = 0; i < OT_BUCKET_COUNT; ++i) |
| 292 | pthread_mutex_destroy(bucket_mutex + i); | 73 | pthread_mutex_destroy(bucket_mutex + i); |
| 293 | pthread_mutex_destroy(&tasklist_mutex); | ||
| 294 | pthread_cond_destroy(&tasklist_being_filled); | ||
| 295 | memset(all_torrents, 0, sizeof(all_torrents)); | 74 | memset(all_torrents, 0, sizeof(all_torrents)); |
| 296 | } | 75 | } |
