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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
/*
* Copyright (c) 1996, 1998 University of Utah and the Flux Group.
* All rights reserved.
*
* This file is part of the Flux OSKit. The OSKit is free software, also known
* as "open source;" you can redistribute it and/or modify it under the terms
* of the GNU General Public License (GPL), version 2, as published by the Free
* Software Foundation (FSF). To explore alternate licensing terms, contact
* the University of Utah at csl-dist@cs.utah.edu or +1-801-585-3271.
*
* The OSKit is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GPL for more details. You should have
* received a copy of the GPL along with the OSKit; see the file COPYING. If
* not, write to the FSF, 59 Temple Place #330, Boston, MA 02111-1307, USA.
*/
#include <threads/pthread_internal.h>
#include <oskit/com/wrapper.h>
/***************************************************************************/
/*
* methods of socket_factories.
*/
#include <oskit/com.h>
#include <oskit/com/services.h>
#include <oskit/net/socket.h>
static struct oskit_socket_factory *pthread_old_sf;
static OSKIT_COMDECL
socket_factory_query(oskit_socket_factory_t *b,
const struct oskit_guid *iid, void **out_ihandle)
{
if (memcmp(iid, &oskit_iunknown_iid, sizeof(*iid)) == 0 ||
memcmp(iid, &oskit_socket_factory_iid, sizeof(*iid)) == 0) {
*out_ihandle = b;
return 0;
}
*out_ihandle = NULL;
return OSKIT_E_NOINTERFACE;
}
static OSKIT_COMDECL_U
socket_factory_addref(oskit_socket_factory_t *b)
{
return 1;
}
static OSKIT_COMDECL_U
socket_factory_release(oskit_socket_factory_t *b)
{
return 1;
}
/*
* function to create a wrapped socket.
*/
static OSKIT_COMDECL
socket_factory_create(
oskit_socket_factory_t *factory,
oskit_u32_t domain,
oskit_u32_t type, oskit_u32_t protocol, oskit_socket_t **aso)
{
oskit_error_t rc;
struct oskit_socket *sock;
/*
* Use the original socket factory to create a socket, then
* wrap it up with a thread safe wrapper.
*/
osenv_process_lock();
rc = oskit_socket_factory_create(pthread_old_sf,
domain, type, protocol, &sock);
osenv_process_unlock();
if (rc)
return rc;
rc = oskit_wrap_sockio(sock,
(void (*)(void *))osenv_process_lock,
(void (*)(void *))osenv_process_unlock,
0, aso);
oskit_socket_release(sock);
return rc;
}
static struct oskit_socket_factory_ops sf_ops = {
socket_factory_query,
socket_factory_addref,
socket_factory_release,
socket_factory_create
};
static struct oskit_socket_factory pthread_new_sf = { &sf_ops };
/*
* Stash the unsafe socket factory, and install a safe version built on it
* in the registry.
*/
void
pthread_init_socketfactory(oskit_socket_factory_t *sf)
{
pthread_old_sf = sf;
/*
* And register the new socket factory.
*/
if (oskit_register(&oskit_socket_factory_iid,
(void *) &pthread_new_sf))
panic("pthread_init_socketfactory: Register failed");
}
|
