Whamcloud - gitweb
- tiny fix to ost_request.c
[fs/lustre-release.git] / lustre / osc / osc_request.c
1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2  * vim:expandtab:shiftwidth=8:tabstop=8:
3  *
4  * Copryright (C) 2001 Cluster File Systems, Inc.
5  *
6  *  This code is issued under the GNU General Public License.
7  *  See the file COPYING in this distribution
8  *
9  *  Author Peter Braam <braam@clusterfs.com>
10  * 
11  *  This server is single threaded at present (but can easily be multi
12  *  threaded). For testing and management it is treated as an
13  *  obd_device, although it does not export a full OBD method table
14  *  (the requests are coming in over the wire, so object target
15  *  modules do not have a full method table.)
16  * 
17  */
18
19 #define EXPORT_SYMTAB
20
21 #include <linux/config.h>
22 #include <linux/module.h>
23 #include <linux/kernel.h>
24 #include <linux/mm.h>
25 #include <linux/string.h>
26 #include <linux/stat.h>
27 #include <linux/errno.h>
28 #include <linux/locks.h>
29 #include <linux/unistd.h>
30
31 #include <asm/system.h>
32 #include <asm/uaccess.h>
33
34 #include <linux/fs.h>
35 #include <linux/stat.h>
36 #include <asm/uaccess.h>
37 #include <asm/segment.h>
38 #include <linux/miscdevice.h>
39
40 #define DEBUG_SUBSYSTEM S_OSC
41
42 #include <linux/obd_support.h>
43 #include <linux/obd_class.h>
44 #include <linux/lustre_lib.h>
45 #include <linux/lustre_idl.h>
46
47 struct ptlrpc_client *osc_con2cl(struct obd_conn *conn)
48 {
49         struct osc_obd *osc = &conn->oc_dev->u.osc;
50         return &osc->osc_peer;
51
52 }
53
54 static int osc_connect(struct obd_conn *conn)
55 {
56         struct ptlrpc_request *request;
57         struct ptlrpc_client *peer = osc_con2cl(conn);
58         int rc; 
59         ENTRY;
60         
61         request = ptlrpc_prep_req(peer, OST_CONNECT, 0, NULL, 0, NULL);
62         if (!request) { 
63                 CERROR("cannot pack req!\n"); 
64                 return -ENOMEM;
65         }
66
67         request->rq_replen = 
68                 sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
69
70         rc = ptlrpc_queue_wait(peer, request);
71         if (rc) { 
72                 EXIT;
73                 goto out;
74         }
75       
76         CDEBUG(D_INODE, "received connid %d\n", request->rq_rep.ost->connid); 
77
78         conn->oc_id = request->rq_rep.ost->connid;
79  out:
80         ptlrpc_free_req(request);
81         EXIT;
82         return rc;
83 }
84
85 static int osc_disconnect(struct obd_conn *conn)
86 {
87         struct ptlrpc_request *request;
88         struct ptlrpc_client *peer = osc_con2cl(conn);
89         int rc; 
90         ENTRY;
91         
92         request = ptlrpc_prep_req(peer, OST_DISCONNECT, 0, NULL, 0, NULL);
93         if (!request) { 
94                 CERROR("cannot pack req!\n"); 
95                 return -ENOMEM;
96         }
97         request->rq_req.ost->connid = conn->oc_id;
98         request->rq_replen = 
99                 sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
100
101         rc = ptlrpc_queue_wait(peer, request);
102         if (rc) { 
103                 EXIT;
104                 goto out;
105         }
106  out:
107         ptlrpc_free_req(request);
108         EXIT;
109         return rc;
110 }
111
112
113 static int osc_getattr(struct obd_conn *conn, struct obdo *oa)
114 {
115         struct ptlrpc_request *request;
116         struct ptlrpc_client *peer = osc_con2cl(conn);
117         int rc; 
118
119         request = ptlrpc_prep_req(peer, OST_GETATTR, 0, NULL, 0, NULL);
120         if (!request) { 
121                 CERROR("cannot pack req!\n"); 
122                 return -ENOMEM;
123         }
124         
125         memcpy(&request->rq_req.ost->oa, oa, sizeof(*oa));
126         request->rq_req.ost->oa.o_valid = ~0;
127         request->rq_replen = 
128                 sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
129         
130         rc = ptlrpc_queue_wait(peer, request);
131         if (rc) { 
132                 EXIT;
133                 goto out;
134         }
135
136         CDEBUG(D_INODE, "mode: %o\n", request->rq_rep.ost->oa.o_mode); 
137         if (oa) { 
138                 memcpy(oa, &request->rq_rep.ost->oa, sizeof(*oa));
139         }
140
141  out:
142         ptlrpc_free_req(request);
143         return 0;
144 }
145
146 static int osc_setattr(struct obd_conn *conn, struct obdo *oa)
147 {
148         struct ptlrpc_request *request;
149         struct ptlrpc_client *peer = osc_con2cl(conn);
150         int rc; 
151
152         request = ptlrpc_prep_req(peer, OST_SETATTR, 0, NULL, 0, NULL);
153         if (!request) { 
154                 CERROR("cannot pack req!\n"); 
155                 return -ENOMEM;
156         }
157         
158         memcpy(&request->rq_req.ost->oa, oa, sizeof(*oa));
159         request->rq_replen = 
160                 sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
161         
162         rc = ptlrpc_queue_wait(peer, request);
163         if (rc) { 
164                 EXIT;
165                 goto out;
166         }
167
168  out:
169         ptlrpc_free_req(request);
170         return 0;
171 }
172
173 static int osc_create(struct obd_conn *conn, struct obdo *oa)
174 {
175         struct ptlrpc_request *request;
176         struct ptlrpc_client *peer = osc_con2cl(conn);
177         int rc; 
178
179         if (!oa) { 
180                 CERROR("oa NULL\n"); 
181         }
182         request = ptlrpc_prep_req(peer, OST_CREATE, 0, NULL, 0, NULL);
183         if (!request) { 
184                 CERROR("cannot pack req!\n"); 
185                 return -ENOMEM;
186         }
187         
188         memcpy(&request->rq_req.ost->oa, oa, sizeof(*oa));
189         request->rq_req.ost->connid = conn->oc_id;
190         request->rq_req.ost->oa.o_valid = ~0;
191         request->rq_replen = 
192                 sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
193         
194         rc = ptlrpc_queue_wait(peer, request);
195         if (rc) { 
196                 EXIT;
197                 goto out;
198         }
199         memcpy(oa, &request->rq_rep.ost->oa, sizeof(*oa));
200
201  out:
202         ptlrpc_free_req(request);
203         return 0;
204 }
205
206 static int osc_punch(struct obd_conn *conn, struct obdo *oa, obd_size count,
207                      obd_off offset)
208 {
209         struct ptlrpc_request *request;
210         struct ptlrpc_client *peer = osc_con2cl(conn);
211         int rc; 
212
213         if (!oa) { 
214                 CERROR("oa NULL\n"); 
215         }
216         request = ptlrpc_prep_req(peer, OST_PUNCH, 0, NULL, 0, NULL);
217         if (!request) { 
218                 CERROR("cannot pack req!\n"); 
219                 return -ENOMEM;
220         }
221         
222         memcpy(&request->rq_req.ost->oa, oa, sizeof(*oa));
223         request->rq_req.ost->oa.o_valid = ~0;
224         request->rq_req.ost->oa.o_size = offset;
225         request->rq_req.ost->oa.o_blocks = count;
226         request->rq_replen = 
227                 sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
228         
229         rc = ptlrpc_queue_wait(peer, request);
230         if (rc) { 
231                 EXIT;
232                 goto out;
233         }
234         memcpy(oa, &request->rq_rep.ost->oa, sizeof(*oa));
235
236  out:
237         ptlrpc_free_req(request);
238         return 0;
239 }
240
241 static int osc_destroy(struct obd_conn *conn, struct obdo *oa)
242 {
243         struct ptlrpc_request *request;
244         struct ptlrpc_client *peer = osc_con2cl(conn);
245         int rc; 
246
247         if (!oa) { 
248                 CERROR("oa NULL\n"); 
249         }
250         request = ptlrpc_prep_req(peer, OST_DESTROY, 0, NULL, 0, NULL);
251         if (!request) { 
252                 CERROR("cannot pack req!\n"); 
253                 return -ENOMEM;
254         }
255         
256         memcpy(&request->rq_req.ost->oa, oa, sizeof(*oa));
257         request->rq_req.ost->oa.o_valid = ~0;
258         request->rq_replen = 
259                 sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
260         
261         rc = ptlrpc_queue_wait(peer, request);
262         if (rc) { 
263                 EXIT;
264                 goto out;
265         }
266         memcpy(oa, &request->rq_rep.ost->oa, sizeof(*oa));
267
268  out:
269         ptlrpc_free_req(request);
270         return 0;
271 }
272
273 int osc_sendpage(struct obd_conn *conn, struct ptlrpc_request *req,
274                  struct niobuf *dst, struct niobuf *src)
275 {
276         if (conn->oc_id != -1) {
277                 /* local sendpage */
278                 memcpy((char *)(unsigned long)dst->addr,
279                        (char *)(unsigned long)src->addr, src->len);
280         } else {
281                 struct ptlrpc_client *cl = osc_con2cl(conn);
282                 struct ptlrpc_bulk_desc *bulk;
283                 char *buf;
284                 int rc;
285
286                 bulk = ptlrpc_prep_bulk(&cl->cli_server);
287                 if (bulk == NULL)
288                         return -ENOMEM;
289
290                 spin_lock(&cl->cli_lock);
291                 bulk->b_xid = cl->cli_xid++;
292                 spin_unlock(&cl->cli_lock);
293
294                 OBD_ALLOC(buf, src->len);
295                 if (!buf) {
296                         OBD_FREE(bulk, sizeof(*bulk));
297                         return -ENOMEM;
298                 }
299
300                 memcpy(buf, (char *)(unsigned long)src->addr, src->len);
301
302                 bulk->b_buf = buf;
303                 bulk->b_buflen = src->len;
304                 /* FIXME: maybe we should add an XID to struct niobuf? */
305                 bulk->b_xid = (__u32)(unsigned long)src->page;
306
307                 rc = ptlrpc_send_bulk(bulk, OSC_BULK_PORTAL);
308                 if (rc != 0) {
309                         CERROR("send_bulk failed: %d\n", rc);
310                         BUG();
311                         return rc;
312                 }
313                 wait_event_interruptible(bulk->b_waitq,
314                                          ptlrpc_check_bulk_sent(bulk));
315
316                 if (bulk->b_flags == PTL_RPC_INTR) {
317                         EXIT;
318                         /* FIXME: hey hey, we leak here. */
319                         return -EINTR;
320                 }
321
322                 OBD_FREE(bulk, sizeof(*bulk));
323                 OBD_FREE(buf, src->len);
324         }
325
326         return 0;
327 }
328
329 int osc_brw_read(struct obd_conn *conn, obd_count num_oa, struct obdo **oa,
330                  obd_count *oa_bufs, struct page **buf, obd_size *count,
331                  obd_off *offset, obd_flag *flags)
332 {
333         struct ptlrpc_client *cl = osc_con2cl(conn);
334         struct ptlrpc_request *request;
335         int pages;
336         int rc; 
337         struct obd_ioobj ioo;
338         struct niobuf src;
339         int size1, size2 = 0; 
340         void *ptr1, *ptr2;
341         int i, j, n;
342         struct ptlrpc_bulk_desc **bulk;
343
344         size1 = num_oa * sizeof(ioo); 
345         pages = 0;
346         for (i = 0; i < num_oa; i++) { 
347                 size2 += oa_bufs[i] * sizeof(src);
348                 pages += oa_bufs[i];
349         }
350
351         /* We actually pack a _third_ buffer, with XIDs for bulk pages */
352         size2 += pages * sizeof(__u32);
353         request = ptlrpc_prep_req(cl, OST_BRW, size1, NULL, size2, NULL);
354         if (!request) { 
355                 CERROR("cannot pack req!\n"); 
356                 return -ENOMEM;
357         }
358         request->rq_req.ost->cmd = OBD_BRW_READ;
359
360         OBD_ALLOC(bulk, pages * sizeof(struct ptlrpc_bulk_desc *));
361         if (bulk == NULL) {
362                 CERROR("cannot alloc bulk desc vector\n");
363                 return -ENOMEM;
364         }
365         memset(bulk, 0, pages * sizeof(struct ptlrpc_bulk_desc *));
366
367         n = 0;
368         ptr1 = ost_req_buf1(request->rq_req.ost);
369         ptr2 = ost_req_buf2(request->rq_req.ost);
370         for (i = 0; i < num_oa; i++) {
371                 ost_pack_ioo(&ptr1, oa[i], oa_bufs[i]); 
372                 for (j = 0; j < oa_bufs[i]; j++) {
373                         bulk[n] = ptlrpc_prep_bulk(&cl->cli_server);
374                         if (bulk[n] == NULL) {
375                                 CERROR("cannot alloc bulk desc\n");
376                                 rc = -ENOMEM;
377                                 goto out;
378                         }
379
380                         spin_lock(&cl->cli_lock);
381                         bulk[n]->b_xid = cl->cli_xid++;
382                         spin_unlock(&cl->cli_lock);
383                         bulk[n]->b_buf = kmap(buf[n]);
384                         bulk[n]->b_buflen = PAGE_SIZE;
385                         bulk[n]->b_portal = OST_BULK_PORTAL;
386                         ost_pack_niobuf(&ptr2, bulk[n]->b_buf, offset[n],
387                                         count[n], flags[n]);
388                         n++;
389                 }
390         }
391
392         /* This is kinda silly--put the XIDs in the "third" buffer. */
393         for (n = 0; n < pages; n++) {
394                 *(__u32 *)ptr2 = bulk[n]->b_xid;
395                 ptr2 = (char *)ptr2 + sizeof(__u32);
396
397                 rc = ptlrpc_register_bulk(bulk[n]);
398                 if (rc)
399                         goto out;
400         }
401
402         request->rq_replen = sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
403         rc = ptlrpc_queue_wait(cl, request);
404
405  out:
406         /* FIXME: if we've called ptlrpc_wait_bulk but rc != 0, we need to
407          * abort those bulk listeners. */
408
409         if (request->rq_rephdr)
410                 OBD_FREE(request->rq_rephdr, request->rq_replen);
411         n = 0;
412         for (i = 0; i < num_oa; i++) {
413                 for (j = 0; j < oa_bufs[i]; j++) {
414                         if (bulk[n] == NULL)
415                                 continue;
416                         kunmap(bulk[n]->b_buf);
417                         OBD_FREE(bulk[n], sizeof(struct ptlrpc_bulk_desc));
418                         n++;
419                 }
420         }
421
422         OBD_FREE(bulk, pages * sizeof(struct ptlrpc_bulk_desc *));
423         ptlrpc_free_req(request);
424         return rc;
425 }
426
427 int osc_brw_write(struct obd_conn *conn, obd_count num_oa, struct obdo **oa,
428                   obd_count *oa_bufs, struct page **buf, obd_size *count,
429                   obd_off *offset, obd_flag *flags)
430 {
431         struct ptlrpc_client *cl = osc_con2cl(conn);
432         struct ptlrpc_request *request;
433         struct obd_ioobj ioo;
434         struct niobuf src;
435         int pages, rc, i, j, n, size1, size2 = 0; 
436         void *ptr1, *ptr2;
437
438         size1 = num_oa * sizeof(ioo); 
439         pages = 0;
440         for (i = 0; i < num_oa; i++) { 
441                 size2 += oa_bufs[i] * sizeof(src);
442                 pages += oa_bufs[i];
443         }
444
445         request = ptlrpc_prep_req(cl, OST_BRW, size1, NULL, size2, NULL);
446         if (!request) { 
447                 CERROR("cannot pack req!\n"); 
448                 return -ENOMEM;
449         }
450         request->rq_req.ost->cmd = OBD_BRW_WRITE;
451
452         n = 0;
453         ptr1 = ost_req_buf1(request->rq_req.ost);
454         ptr2 = ost_req_buf2(request->rq_req.ost);
455         for (i = 0; i < num_oa; i++) {
456                 ost_pack_ioo(&ptr1, oa[i], oa_bufs[i]); 
457                 for (j = 0; j < oa_bufs[i]; j++) {
458                         ost_pack_niobuf(&ptr2, kmap(buf[n]), offset[n],
459                                         count[n], flags[n]);
460                         n++;
461                 }
462         }
463
464         request->rq_replen = sizeof(struct ptlrep_hdr) +
465                 sizeof(struct ost_rep) + pages * sizeof(struct niobuf);
466         rc = ptlrpc_queue_wait(cl, request);
467         if (rc) { 
468                 EXIT;
469                 goto out;
470         }
471
472         ptr2 = ost_rep_buf2(request->rq_rep.ost);
473         if (request->rq_rep.ost->buflen2 != n * sizeof(struct niobuf)) {
474                 CERROR("buffer length wrong (%d vs. %d)\n",
475                        request->rq_rep.ost->buflen2, n * sizeof(struct niobuf));
476                 EXIT;
477                 goto out;
478         }
479
480         for (i = 0; i < num_oa; i++) {
481                 for (j = 0; j < oa_bufs[i]; j++) {
482                         struct niobuf *dst;
483                         src.addr = (__u64)(unsigned long)buf[n];
484                         src.len = count[n];
485                         ost_unpack_niobuf(&ptr2, &dst);
486                         osc_sendpage(conn, request, dst, &src);
487                         n++;
488                 }
489         }
490
491         /* Reuse the request structure for the completion request. */
492         OBD_FREE(request->rq_rephdr, request->rq_replen);
493         request->rq_rephdr = NULL;
494         request->rq_repbuf = NULL;
495         request->rq_reqhdr->opc = OST_BRW_COMPLETE;
496         request->rq_replen = sizeof(struct ptlrep_hdr) + sizeof(struct ost_rep);
497         rc = ptlrpc_queue_wait(cl, request);
498         if (rc) { 
499                 EXIT;
500                 goto out;
501         }
502
503  out:
504         if (request->rq_rephdr)
505                 OBD_FREE(request->rq_rephdr, request->rq_replen);
506         n = 0;
507         for (i = 0; i < num_oa; i++) {
508                 for (j = 0; j < oa_bufs[i]; j++) {
509                         kunmap(buf[n]);
510                         n++;
511                 }
512         }
513
514         ptlrpc_free_req(request);
515         return 0;
516 }
517
518 int osc_brw(int rw, struct obd_conn *conn, obd_count num_oa,
519               struct obdo **oa, obd_count *oa_bufs, struct page **buf,
520               obd_size *count, obd_off *offset, obd_flag *flags)
521 {
522         if (rw == OBD_BRW_READ) {
523                 return osc_brw_read(conn, num_oa, oa, oa_bufs, buf, count,
524                                     offset, flags);
525         } else {
526                 return osc_brw_write(conn, num_oa, oa, oa_bufs, buf, count,
527                                      offset, flags);
528         }
529 }
530
531 /* mount the file system (secretly) */
532 static int osc_setup(struct obd_device *obddev, obd_count len,
533                         void *buf)
534                         
535 {
536         struct osc_obd *osc = &obddev->u.osc;
537         struct obd_ioctl_data *data = (struct obd_ioctl_data *)buf;
538         int rc;
539         int dev = data->ioc_dev;
540         ENTRY;
541
542         rc = ptlrpc_connect_client(dev, "ost", 
543                                    OST_REQUEST_PORTAL, 
544                                    OSC_REPLY_PORTAL,    
545                                    ost_pack_req, 
546                                    ost_unpack_rep,
547                                    &osc->osc_peer); 
548
549         MOD_INC_USE_COUNT;
550         EXIT;
551         return rc;
552
553
554 static int osc_cleanup(struct obd_device * obddev)
555 {
556         MOD_DEC_USE_COUNT;
557         return 0;
558 }
559
560 struct obd_ops osc_obd_ops = { 
561         o_setup:   osc_setup,
562         o_cleanup: osc_cleanup, 
563         o_create: osc_create,
564         o_destroy: osc_destroy,
565         o_getattr: osc_getattr,
566         o_setattr: osc_setattr,
567         o_connect: osc_connect,
568         o_disconnect: osc_disconnect,
569         o_brw: osc_brw,
570         o_punch: osc_punch
571 };
572
573 static int __init osc_init(void)
574 {
575         obd_register_type(&osc_obd_ops, LUSTRE_OSC_NAME);
576         return 0;
577 }
578
579 static void __exit osc_exit(void)
580 {
581         obd_unregister_type(LUSTRE_OSC_NAME);
582 }
583
584 MODULE_AUTHOR("Peter J. Braam <braam@clusterfs.com>");
585 MODULE_DESCRIPTION("Lustre Object Storage Client (OSC) v1.0");
586 MODULE_LICENSE("GPL"); 
587
588 module_init(osc_init);
589 module_exit(osc_exit);