Whamcloud - gitweb
Land b_smallfix onto HEAD (20040416_1638) (more 2.6 build fixes)
[fs/lustre-release.git] / lnet / ulnds / socklnd / procapi.c
1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2  * vim:expandtab:shiftwidth=8:tabstop=8:
3  *
4  *  Copyright (c) 2002 Cray Inc.
5  *  Copyright (c) 2003 Cluster File Systems, Inc.
6  *
7  *   This file is part of Lustre, http://www.lustre.org.
8  *
9  *   Lustre is free software; you can redistribute it and/or
10  *   modify it under the terms of version 2 of the GNU General Public
11  *   License as published by the Free Software Foundation.
12  *
13  *   Lustre is distributed in the hope that it will be useful,
14  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *   GNU General Public License for more details.
17  *
18  *   You should have received a copy of the GNU General Public License
19  *   along with Lustre; if not, write to the Free Software
20  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21  */
22
23 /* api.c:
24  *  This file provides the 'api' side for the process-based nals.
25  *  it is responsible for creating the 'library' side thread,
26  *  and passing wrapped portals transactions to it.
27  *
28  *  Along with initialization, shutdown, and transport to the library
29  *  side, this file contains some stubs to satisfy the nal definition.
30  */
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <unistd.h>
34 #include <string.h>
35 #ifndef __CYGWIN__
36 #include <syscall.h>
37 #endif
38 #include <sys/socket.h>
39 #include <procbridge.h>
40 #include <pqtimer.h>
41 #include <dispatch.h>
42 #include <errno.h>
43
44
45 /* XXX CFS workaround, to give a chance to let nal thread wake up
46  * from waiting in select
47  */
48 static int procbridge_notifier_handler(void *arg)
49 {
50     static char buf[8];
51     procbridge p = (procbridge) arg;
52
53     syscall(SYS_read, p->notifier[1], buf, sizeof(buf));
54     return 1;
55 }
56
57 void procbridge_wakeup_nal(procbridge p)
58 {
59     static char buf[8];
60     syscall(SYS_write, p->notifier[0], buf, sizeof(buf));
61 }
62
63 /* Function: forward
64  * Arguments: nal_t *nal: pointer to my top-side nal structure
65  *            id: the command to pass to the lower layer
66  *            args, args_len:pointer to and length of the request
67  *            ret, ret_len:  pointer to and size of the result
68  * Returns: a portals status code
69  *
70  * forwards a packaged api call from the 'api' side to the 'library'
71  *   side, and collects the result
72  */
73 static int procbridge_forward(nal_t *n, int id, void *args, size_t args_len,
74                               void *ret, size_t ret_len)
75 {
76     bridge b = (bridge) n->nal_data;
77
78     if (id == PTL_FINI) {
79             lib_fini(b->nal_cb);
80
81             if (b->shutdown)
82                 (*b->shutdown)(b);
83     }
84
85     lib_dispatch(b->nal_cb, NULL, id, args, ret);
86
87     return (PTL_OK);
88 }
89
90
91 /* Function: shutdown
92  * Arguments: nal: a pointer to my top side nal structure
93  *            ni: my network interface index
94  *
95  * cleanup nal state, reclaim the lower side thread and
96  *   its state using PTL_FINI codepoint
97  */
98 static void procbridge_shutdown(nal_t *n)
99 {
100     bridge b=(bridge)n->nal_data;
101     procbridge p=(procbridge)b->local;
102
103     p->nal_flags |= NAL_FLAG_STOPPING;
104     procbridge_wakeup_nal(p);
105
106     do {
107         pthread_mutex_lock(&p->mutex);
108         if (p->nal_flags & NAL_FLAG_STOPPED) {
109                 pthread_mutex_unlock(&p->mutex);
110                 break;
111         }
112         pthread_cond_wait(&p->cond, &p->mutex);
113         pthread_mutex_unlock(&p->mutex);
114     } while (1);
115
116     free(p);
117 }
118
119
120 static void procbridge_lock(nal_t * n, unsigned long *flags)
121 {
122     bridge b=(bridge)n->nal_data;
123     procbridge p=(procbridge)b->local;
124
125     pthread_mutex_lock(&p->mutex);
126 }
127
128 static void procbridge_unlock(nal_t * n, unsigned long *flags)
129 {
130     bridge b=(bridge)n->nal_data;
131     procbridge p=(procbridge)b->local;
132
133     pthread_mutex_unlock(&p->mutex);
134 }
135
136 /* Function: yield
137  * Arguments:  pid:
138  *
139  *  this function was originally intended to allow the
140  *   lower half thread to be scheduled to allow progress. we
141  *   overload it to explicitly block until signalled by the
142  *   lower half.
143  */
144 static int procbridge_yield(nal_t *n, unsigned long *flags, int milliseconds)
145 {
146     bridge b=(bridge)n->nal_data;
147     procbridge p=(procbridge)b->local;
148
149     if (milliseconds == 0)
150             return 0;
151             
152     if (milliseconds < 0) {
153         pthread_cond_wait(&p->cond,&p->mutex);
154     } else {
155         struct timeval then;
156         struct timeval now;
157         struct timespec timeout;
158
159         gettimeofday(&then, NULL);
160         timeout.tv_sec = then.tv_sec + milliseconds/1000;
161         timeout.tv_nsec = then.tv_usec * 1000 + milliseconds % 1000 * 1000000;
162         if (timeout.tv_nsec >= 1000000000) {
163                 timeout.tv_sec++;
164                 timeout.tv_nsec -= 1000000000;
165         }
166
167         pthread_cond_timedwait(&p->cond, &p->mutex, &timeout);
168
169         gettimeofday(&now, NULL);
170         milliseconds -= (now.tv_sec - then.tv_sec) * 1000 + 
171                         (now.tv_usec - then.tv_usec) / 1000;
172         
173         if (milliseconds < 0)
174                 milliseconds = 0;
175     }
176
177     return (milliseconds);
178 }
179
180 /* forward decl */
181 extern int procbridge_startup (nal_t *, ptl_pid_t,
182                                ptl_ni_limits_t *, ptl_ni_limits_t *);
183
184 /* api_nal
185  *  the interface vector to allow the generic code to access
186  *  this nal. this is seperate from the library side nal_cb.
187  *  TODO: should be dyanmically allocated
188  */
189 nal_t procapi_nal = {
190     nal_data: NULL,
191     startup:  procbridge_startup,
192     shutdown: procbridge_shutdown,
193     forward:  procbridge_forward,
194     yield:    procbridge_yield,
195     lock:     procbridge_lock,
196     unlock:   procbridge_unlock
197 };
198
199 ptl_nid_t tcpnal_mynid;
200
201 /* Function: procbridge_startup
202  *
203  * Arguments:  pid: requested process id (port offset)
204  *                  PTL_ID_ANY not supported.
205  *             desired: limits passed from the application
206  *                      and effectively ignored
207  *             actual:  limits actually allocated and returned
208  *
209  * Returns: portals rc
210  *
211  * initializes the tcp nal. we define unix_failure as an
212  * error wrapper to cut down clutter.
213  */
214 int procbridge_startup (nal_t *nal, ptl_pid_t requested_pid,
215                         ptl_ni_limits_t *requested_limits,
216                         ptl_ni_limits_t *actual_limits)
217 {
218     nal_init_args_t args;
219
220     procbridge p;
221     bridge b;
222     /* XXX nal_type is purely private to tcpnal here */
223     int nal_type = PTL_IFACE_TCP;/* PTL_IFACE_DEFAULT FIXME hack */
224
225     LASSERT(nal == &procapi_nal);
226
227     init_unix_timer();
228
229     b=(bridge)malloc(sizeof(struct bridge));
230     p=(procbridge)malloc(sizeof(struct procbridge));
231     nal->nal_data=b;
232     b->local=p;
233
234     args.nia_requested_pid = requested_pid;
235     args.nia_requested_limits = requested_limits;
236     args.nia_actual_limits = actual_limits;
237     args.nia_nal_type = nal_type;
238     args.nia_bridge = b;
239
240     /* init procbridge */
241     pthread_mutex_init(&p->mutex,0);
242     pthread_cond_init(&p->cond, 0);
243     p->nal_flags = 0;
244
245     /* initialize notifier */
246     if (socketpair(AF_UNIX, SOCK_STREAM, 0, p->notifier)) {
247         perror("socketpair failed");
248         return PTL_FAIL;
249     }
250
251     if (!register_io_handler(p->notifier[1], READ_HANDLER,
252                 procbridge_notifier_handler, p)) {
253         perror("fail to register notifier handler");
254         return PTL_FAIL;
255     }
256
257     /* create nal thread */
258     if (pthread_create(&p->t, NULL, nal_thread, &args)) {
259         perror("nal_init: pthread_create");
260         return PTL_FAIL;
261     }
262
263     do {
264         pthread_mutex_lock(&p->mutex);
265         if (p->nal_flags & (NAL_FLAG_RUNNING | NAL_FLAG_STOPPED)) {
266                 pthread_mutex_unlock(&p->mutex);
267                 break;
268         }
269         pthread_cond_wait(&p->cond, &p->mutex);
270         pthread_mutex_unlock(&p->mutex);
271     } while (1);
272
273     if (p->nal_flags & NAL_FLAG_STOPPED)
274         return PTL_FAIL;
275
276     b->nal_cb->ni.nid = tcpnal_mynid;
277
278     return PTL_OK;
279 }