1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
/*
* Copyright (c) 2014 Philip Guenther <guenther@openbsd.org>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <pthread.h>
#include <signal.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include "thread_private.h"
#include "rthread_cb.h"
static __dead void
_thread_canceled(void)
{
pthread_exit(PTHREAD_CANCELED);
}
void
_thread_set_callbacks(const struct thread_callbacks *cb, size_t len)
{
sigset_t allmask, omask;
if (sizeof(*cb) != len) {
fprintf(stderr, "library mismatch: libc expected %zu but"
" libpthread gave %zu\n", sizeof(*cb), len);
fflush(stderr);
_exit(44);
}
sigfillset(&allmask);
if (sigprocmask(SIG_BLOCK, &allmask, &omask) == 0) {
/* mprotect RW */
memcpy(&_thread_cb, cb, sizeof(_thread_cb));
/*
* These are supplied by libc, but only enabled
* here when we actually need to prep for doing MT.
*/
_thread_cb.tc_canceled = _thread_canceled;
_thread_cb.tc_flockfile = _thread_flockfile;
_thread_cb.tc_ftrylockfile = _thread_ftrylockfile;
_thread_cb.tc_funlockfile = _thread_funlockfile;
_thread_cb.tc_malloc_lock = _thread_malloc_lock;
_thread_cb.tc_malloc_unlock = _thread_malloc_unlock;
_thread_cb.tc_atexit_lock = _thread_atexit_lock;
_thread_cb.tc_atexit_unlock = _thread_atexit_unlock;
_thread_cb.tc_atfork_lock = _thread_atfork_lock;
_thread_cb.tc_atfork_unlock = _thread_atfork_unlock;
_thread_cb.tc_arc4_lock = _thread_arc4_lock;
_thread_cb.tc_arc4_unlock = _thread_arc4_unlock;
_thread_cb.tc_mutex_lock = _thread_mutex_lock;
_thread_cb.tc_mutex_unlock = _thread_mutex_unlock;
_thread_cb.tc_mutex_destroy = _thread_mutex_destroy;
_thread_cb.tc_tag_lock = _thread_tag_lock;
_thread_cb.tc_tag_unlock = _thread_tag_unlock;
_thread_cb.tc_tag_storage = _thread_tag_storage;
/* mprotect RO | LOCKPERM | NOUNMAP */
sigprocmask(SIG_SETMASK, &omask, NULL);
}
}
|
