4 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2 only,
8 * as published by the Free Software Foundation.
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License version 2 for more details (a copy is included
14 * in the LICENSE file that accompanied this code).
16 * You should have received a copy of the GNU General Public License
17 * version 2 along with this program; If not, see
18 * http://www.gnu.org/licenses/gpl-2.0.html
23 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
24 * Use is subject to license terms.
26 * Copyright (c) 2011, 2015, Intel Corporation.
29 * This file is part of Lustre, http://www.lustre.org/
30 * Lustre is a trademark of Sun Microsystems, Inc.
32 * lustre/ptlrpc/sec_plain.c
34 * Author: Eric Mei <ericm@clusterfs.com>
37 #define DEBUG_SUBSYSTEM S_SEC
39 #include "../include/obd_support.h"
40 #include "../include/obd_cksum.h"
41 #include "../include/obd_class.h"
42 #include "../include/lustre_net.h"
43 #include "../include/lustre_sec.h"
44 #include "ptlrpc_internal.h"
47 struct ptlrpc_sec pls_base;
49 struct ptlrpc_cli_ctx *pls_ctx;
52 static inline struct plain_sec *sec2plsec(struct ptlrpc_sec *sec)
54 return container_of(sec, struct plain_sec, pls_base);
57 static struct ptlrpc_sec_policy plain_policy;
58 static struct ptlrpc_ctx_ops plain_ctx_ops;
59 static struct ptlrpc_svc_ctx plain_svc_ctx;
61 static unsigned int plain_at_offset;
64 * for simplicity, plain policy rpc use fixed layout.
66 #define PLAIN_PACK_SEGMENTS (4)
68 #define PLAIN_PACK_HDR_OFF (0)
69 #define PLAIN_PACK_MSG_OFF (1)
70 #define PLAIN_PACK_USER_OFF (2)
71 #define PLAIN_PACK_BULK_OFF (3)
73 #define PLAIN_FL_USER (0x01)
74 #define PLAIN_FL_BULK (0x02)
79 __u8 ph_sp; /* source */
80 __u8 ph_bulk_hash_alg; /* complete flavor desc */
84 struct plain_bulk_token {
88 #define PLAIN_BSD_SIZE \
89 (sizeof(struct ptlrpc_bulk_sec_desc) + sizeof(struct plain_bulk_token))
91 /****************************************
92 * bulk checksum helpers *
93 ****************************************/
95 static int plain_unpack_bsd(struct lustre_msg *msg, int swabbed)
97 struct ptlrpc_bulk_sec_desc *bsd;
99 if (bulk_sec_desc_unpack(msg, PLAIN_PACK_BULK_OFF, swabbed))
102 bsd = lustre_msg_buf(msg, PLAIN_PACK_BULK_OFF, PLAIN_BSD_SIZE);
104 CERROR("bulk sec desc has short size %d\n",
105 lustre_msg_buflen(msg, PLAIN_PACK_BULK_OFF));
109 if (bsd->bsd_svc != SPTLRPC_BULK_SVC_NULL &&
110 bsd->bsd_svc != SPTLRPC_BULK_SVC_INTG) {
111 CERROR("invalid bulk svc %u\n", bsd->bsd_svc);
118 static int plain_generate_bulk_csum(struct ptlrpc_bulk_desc *desc,
120 struct plain_bulk_token *token)
122 if (hash_alg == BULK_HASH_ALG_NULL)
125 memset(token->pbt_hash, 0, sizeof(token->pbt_hash));
126 return sptlrpc_get_bulk_checksum(desc, hash_alg, token->pbt_hash,
127 sizeof(token->pbt_hash));
130 static int plain_verify_bulk_csum(struct ptlrpc_bulk_desc *desc,
132 struct plain_bulk_token *tokenr)
134 struct plain_bulk_token tokenv;
137 if (hash_alg == BULK_HASH_ALG_NULL)
140 memset(&tokenv.pbt_hash, 0, sizeof(tokenv.pbt_hash));
141 rc = sptlrpc_get_bulk_checksum(desc, hash_alg, tokenv.pbt_hash,
142 sizeof(tokenv.pbt_hash));
146 if (memcmp(tokenr->pbt_hash, tokenv.pbt_hash, sizeof(tokenr->pbt_hash)))
151 static void corrupt_bulk_data(struct ptlrpc_bulk_desc *desc)
156 for (i = 0; i < desc->bd_iov_count; i++) {
157 if (desc->bd_iov[i].bv_len == 0)
160 ptr = kmap(desc->bd_iov[i].bv_page);
161 off = desc->bd_iov[i].bv_offset & ~PAGE_MASK;
163 kunmap(desc->bd_iov[i].bv_page);
168 /****************************************
170 ****************************************/
173 int plain_ctx_refresh(struct ptlrpc_cli_ctx *ctx)
175 /* should never reach here */
181 int plain_ctx_validate(struct ptlrpc_cli_ctx *ctx)
187 int plain_ctx_sign(struct ptlrpc_cli_ctx *ctx, struct ptlrpc_request *req)
189 struct lustre_msg *msg = req->rq_reqbuf;
190 struct plain_header *phdr;
192 msg->lm_secflvr = req->rq_flvr.sf_rpc;
194 phdr = lustre_msg_buf(msg, PLAIN_PACK_HDR_OFF, 0);
197 phdr->ph_sp = ctx->cc_sec->ps_part;
198 phdr->ph_bulk_hash_alg = req->rq_flvr.u_bulk.hash.hash_alg;
200 if (req->rq_pack_udesc)
201 phdr->ph_flags |= PLAIN_FL_USER;
202 if (req->rq_pack_bulk)
203 phdr->ph_flags |= PLAIN_FL_BULK;
205 req->rq_reqdata_len = lustre_msg_size_v2(msg->lm_bufcount,
211 int plain_ctx_verify(struct ptlrpc_cli_ctx *ctx, struct ptlrpc_request *req)
213 struct lustre_msg *msg = req->rq_repdata;
214 struct plain_header *phdr;
218 if (msg->lm_bufcount != PLAIN_PACK_SEGMENTS) {
219 CERROR("unexpected reply buf count %u\n", msg->lm_bufcount);
223 swabbed = ptlrpc_rep_need_swab(req);
225 phdr = lustre_msg_buf(msg, PLAIN_PACK_HDR_OFF, sizeof(*phdr));
227 CERROR("missing plain header\n");
231 if (phdr->ph_ver != 0) {
232 CERROR("Invalid header version\n");
236 /* expect no user desc in reply */
237 if (phdr->ph_flags & PLAIN_FL_USER) {
238 CERROR("Unexpected udesc flag in reply\n");
242 if (phdr->ph_bulk_hash_alg != req->rq_flvr.u_bulk.hash.hash_alg) {
243 CERROR("reply bulk flavor %u != %u\n", phdr->ph_bulk_hash_alg,
244 req->rq_flvr.u_bulk.hash.hash_alg);
248 if (unlikely(req->rq_early)) {
249 unsigned int hsize = 4;
251 cfs_crypto_hash_digest(CFS_HASH_ALG_CRC32,
252 lustre_msg_buf(msg, PLAIN_PACK_MSG_OFF,
254 lustre_msg_buflen(msg,
256 NULL, 0, (unsigned char *)&cksum,
258 if (cksum != msg->lm_cksum) {
260 "early reply checksum mismatch: %08x != %08x\n",
261 cpu_to_le32(cksum), msg->lm_cksum);
265 /* whether we sent with bulk or not, we expect the same
266 * in reply, except for early reply
268 if (!req->rq_early &&
269 !equi(req->rq_pack_bulk == 1,
270 phdr->ph_flags & PLAIN_FL_BULK)) {
271 CERROR("%s bulk checksum in reply\n",
272 req->rq_pack_bulk ? "Missing" : "Unexpected");
276 if (phdr->ph_flags & PLAIN_FL_BULK) {
277 if (plain_unpack_bsd(msg, swabbed))
282 req->rq_repmsg = lustre_msg_buf(msg, PLAIN_PACK_MSG_OFF, 0);
283 req->rq_replen = lustre_msg_buflen(msg, PLAIN_PACK_MSG_OFF);
288 int plain_cli_wrap_bulk(struct ptlrpc_cli_ctx *ctx,
289 struct ptlrpc_request *req,
290 struct ptlrpc_bulk_desc *desc)
292 struct ptlrpc_bulk_sec_desc *bsd;
293 struct plain_bulk_token *token;
296 LASSERT(req->rq_pack_bulk);
297 LASSERT(req->rq_reqbuf->lm_bufcount == PLAIN_PACK_SEGMENTS);
299 bsd = lustre_msg_buf(req->rq_reqbuf, PLAIN_PACK_BULK_OFF, 0);
300 token = (struct plain_bulk_token *)bsd->bsd_data;
302 bsd->bsd_version = 0;
304 bsd->bsd_type = SPTLRPC_BULK_DEFAULT;
305 bsd->bsd_svc = SPTLRPC_FLVR_BULK_SVC(req->rq_flvr.sf_rpc);
307 if (bsd->bsd_svc == SPTLRPC_BULK_SVC_NULL)
310 if (req->rq_bulk_read)
313 rc = plain_generate_bulk_csum(desc, req->rq_flvr.u_bulk.hash.hash_alg,
316 CERROR("bulk write: failed to compute checksum: %d\n", rc);
319 * for sending we only compute the wrong checksum instead
320 * of corrupting the data so it is still correct on a redo
322 if (OBD_FAIL_CHECK(OBD_FAIL_OSC_CHECKSUM_SEND) &&
323 req->rq_flvr.u_bulk.hash.hash_alg != BULK_HASH_ALG_NULL)
324 token->pbt_hash[0] ^= 0x1;
331 int plain_cli_unwrap_bulk(struct ptlrpc_cli_ctx *ctx,
332 struct ptlrpc_request *req,
333 struct ptlrpc_bulk_desc *desc)
335 struct ptlrpc_bulk_sec_desc *bsdv;
336 struct plain_bulk_token *tokenv;
340 LASSERT(req->rq_pack_bulk);
341 LASSERT(req->rq_reqbuf->lm_bufcount == PLAIN_PACK_SEGMENTS);
342 LASSERT(req->rq_repdata->lm_bufcount == PLAIN_PACK_SEGMENTS);
344 bsdv = lustre_msg_buf(req->rq_repdata, PLAIN_PACK_BULK_OFF, 0);
345 tokenv = (struct plain_bulk_token *)bsdv->bsd_data;
347 if (req->rq_bulk_write) {
348 if (bsdv->bsd_flags & BSD_FL_ERR)
353 /* fix the actual data size */
354 for (i = 0, nob = 0; i < desc->bd_iov_count; i++) {
355 if (desc->bd_iov[i].bv_len + nob > desc->bd_nob_transferred) {
356 desc->bd_iov[i].bv_len =
357 desc->bd_nob_transferred - nob;
359 nob += desc->bd_iov[i].bv_len;
362 rc = plain_verify_bulk_csum(desc, req->rq_flvr.u_bulk.hash.hash_alg,
365 CERROR("bulk read: client verify failed: %d\n", rc);
370 /****************************************
372 ****************************************/
375 struct ptlrpc_cli_ctx *plain_sec_install_ctx(struct plain_sec *plsec)
377 struct ptlrpc_cli_ctx *ctx, *ctx_new;
379 ctx_new = kzalloc(sizeof(*ctx_new), GFP_NOFS);
381 write_lock(&plsec->pls_lock);
383 ctx = plsec->pls_ctx;
385 atomic_inc(&ctx->cc_refcount);
388 } else if (ctx_new) {
391 atomic_set(&ctx->cc_refcount, 1); /* for cache */
392 ctx->cc_sec = &plsec->pls_base;
393 ctx->cc_ops = &plain_ctx_ops;
395 ctx->cc_flags = PTLRPC_CTX_CACHED | PTLRPC_CTX_UPTODATE;
396 ctx->cc_vcred.vc_uid = 0;
397 spin_lock_init(&ctx->cc_lock);
398 INIT_LIST_HEAD(&ctx->cc_req_list);
399 INIT_LIST_HEAD(&ctx->cc_gc_chain);
401 plsec->pls_ctx = ctx;
402 atomic_inc(&plsec->pls_base.ps_nctx);
403 atomic_inc(&plsec->pls_base.ps_refcount);
405 atomic_inc(&ctx->cc_refcount); /* for caller */
408 write_unlock(&plsec->pls_lock);
414 void plain_destroy_sec(struct ptlrpc_sec *sec)
416 struct plain_sec *plsec = sec2plsec(sec);
418 LASSERT(sec->ps_policy == &plain_policy);
419 LASSERT(sec->ps_import);
420 LASSERT(atomic_read(&sec->ps_refcount) == 0);
421 LASSERT(atomic_read(&sec->ps_nctx) == 0);
422 LASSERT(!plsec->pls_ctx);
424 class_import_put(sec->ps_import);
430 void plain_kill_sec(struct ptlrpc_sec *sec)
436 struct ptlrpc_sec *plain_create_sec(struct obd_import *imp,
437 struct ptlrpc_svc_ctx *svc_ctx,
438 struct sptlrpc_flavor *sf)
440 struct plain_sec *plsec;
441 struct ptlrpc_sec *sec;
442 struct ptlrpc_cli_ctx *ctx;
444 LASSERT(SPTLRPC_FLVR_POLICY(sf->sf_rpc) == SPTLRPC_POLICY_PLAIN);
446 plsec = kzalloc(sizeof(*plsec), GFP_NOFS);
451 * initialize plain_sec
453 rwlock_init(&plsec->pls_lock);
454 plsec->pls_ctx = NULL;
456 sec = &plsec->pls_base;
457 sec->ps_policy = &plain_policy;
458 atomic_set(&sec->ps_refcount, 0);
459 atomic_set(&sec->ps_nctx, 0);
460 sec->ps_id = sptlrpc_get_next_secid();
461 sec->ps_import = class_import_get(imp);
463 spin_lock_init(&sec->ps_lock);
464 INIT_LIST_HEAD(&sec->ps_gc_list);
465 sec->ps_gc_interval = 0;
468 /* install ctx immediately if this is a reverse sec */
470 ctx = plain_sec_install_ctx(plsec);
472 plain_destroy_sec(sec);
475 sptlrpc_cli_ctx_put(ctx, 1);
482 struct ptlrpc_cli_ctx *plain_lookup_ctx(struct ptlrpc_sec *sec,
483 struct vfs_cred *vcred,
484 int create, int remove_dead)
486 struct plain_sec *plsec = sec2plsec(sec);
487 struct ptlrpc_cli_ctx *ctx;
489 read_lock(&plsec->pls_lock);
490 ctx = plsec->pls_ctx;
492 atomic_inc(&ctx->cc_refcount);
493 read_unlock(&plsec->pls_lock);
496 ctx = plain_sec_install_ctx(plsec);
502 void plain_release_ctx(struct ptlrpc_sec *sec,
503 struct ptlrpc_cli_ctx *ctx, int sync)
505 LASSERT(atomic_read(&sec->ps_refcount) > 0);
506 LASSERT(atomic_read(&sec->ps_nctx) > 0);
507 LASSERT(atomic_read(&ctx->cc_refcount) == 0);
508 LASSERT(ctx->cc_sec == sec);
512 atomic_dec(&sec->ps_nctx);
513 sptlrpc_sec_put(sec);
517 int plain_flush_ctx_cache(struct ptlrpc_sec *sec,
518 uid_t uid, int grace, int force)
520 struct plain_sec *plsec = sec2plsec(sec);
521 struct ptlrpc_cli_ctx *ctx;
523 /* do nothing unless caller want to flush for 'all' */
527 write_lock(&plsec->pls_lock);
528 ctx = plsec->pls_ctx;
529 plsec->pls_ctx = NULL;
530 write_unlock(&plsec->pls_lock);
533 sptlrpc_cli_ctx_put(ctx, 1);
538 int plain_alloc_reqbuf(struct ptlrpc_sec *sec,
539 struct ptlrpc_request *req,
542 __u32 buflens[PLAIN_PACK_SEGMENTS] = { 0, };
545 buflens[PLAIN_PACK_HDR_OFF] = sizeof(struct plain_header);
546 buflens[PLAIN_PACK_MSG_OFF] = msgsize;
548 if (req->rq_pack_udesc)
549 buflens[PLAIN_PACK_USER_OFF] = sptlrpc_current_user_desc_size();
551 if (req->rq_pack_bulk) {
552 LASSERT(req->rq_bulk_read || req->rq_bulk_write);
553 buflens[PLAIN_PACK_BULK_OFF] = PLAIN_BSD_SIZE;
556 alloc_len = lustre_msg_size_v2(PLAIN_PACK_SEGMENTS, buflens);
558 if (!req->rq_reqbuf) {
559 LASSERT(!req->rq_pool);
561 alloc_len = size_roundup_power2(alloc_len);
562 req->rq_reqbuf = libcfs_kvzalloc(alloc_len, GFP_NOFS);
566 req->rq_reqbuf_len = alloc_len;
568 LASSERT(req->rq_pool);
569 LASSERT(req->rq_reqbuf_len >= alloc_len);
570 memset(req->rq_reqbuf, 0, alloc_len);
573 lustre_init_msg_v2(req->rq_reqbuf, PLAIN_PACK_SEGMENTS, buflens, NULL);
574 req->rq_reqmsg = lustre_msg_buf(req->rq_reqbuf, PLAIN_PACK_MSG_OFF, 0);
576 if (req->rq_pack_udesc) {
577 int rc = sptlrpc_pack_user_desc(req->rq_reqbuf,
578 PLAIN_PACK_USER_OFF);
587 void plain_free_reqbuf(struct ptlrpc_sec *sec,
588 struct ptlrpc_request *req)
591 kvfree(req->rq_reqbuf);
592 req->rq_reqbuf = NULL;
593 req->rq_reqbuf_len = 0;
598 int plain_alloc_repbuf(struct ptlrpc_sec *sec,
599 struct ptlrpc_request *req,
602 __u32 buflens[PLAIN_PACK_SEGMENTS] = { 0, };
605 buflens[PLAIN_PACK_HDR_OFF] = sizeof(struct plain_header);
606 buflens[PLAIN_PACK_MSG_OFF] = msgsize;
608 if (req->rq_pack_bulk) {
609 LASSERT(req->rq_bulk_read || req->rq_bulk_write);
610 buflens[PLAIN_PACK_BULK_OFF] = PLAIN_BSD_SIZE;
613 alloc_len = lustre_msg_size_v2(PLAIN_PACK_SEGMENTS, buflens);
615 /* add space for early reply */
616 alloc_len += plain_at_offset;
618 alloc_len = size_roundup_power2(alloc_len);
620 req->rq_repbuf = libcfs_kvzalloc(alloc_len, GFP_NOFS);
624 req->rq_repbuf_len = alloc_len;
629 void plain_free_repbuf(struct ptlrpc_sec *sec,
630 struct ptlrpc_request *req)
632 kvfree(req->rq_repbuf);
633 req->rq_repbuf = NULL;
634 req->rq_repbuf_len = 0;
638 int plain_enlarge_reqbuf(struct ptlrpc_sec *sec,
639 struct ptlrpc_request *req,
640 int segment, int newsize)
642 struct lustre_msg *newbuf;
644 int newmsg_size, newbuf_size;
646 LASSERT(req->rq_reqbuf);
647 LASSERT(req->rq_reqbuf_len >= req->rq_reqlen);
648 LASSERT(lustre_msg_buf(req->rq_reqbuf, PLAIN_PACK_MSG_OFF, 0) ==
651 /* compute new embedded msg size. */
652 oldsize = req->rq_reqmsg->lm_buflens[segment];
653 req->rq_reqmsg->lm_buflens[segment] = newsize;
654 newmsg_size = lustre_msg_size_v2(req->rq_reqmsg->lm_bufcount,
655 req->rq_reqmsg->lm_buflens);
656 req->rq_reqmsg->lm_buflens[segment] = oldsize;
658 /* compute new wrapper msg size. */
659 oldsize = req->rq_reqbuf->lm_buflens[PLAIN_PACK_MSG_OFF];
660 req->rq_reqbuf->lm_buflens[PLAIN_PACK_MSG_OFF] = newmsg_size;
661 newbuf_size = lustre_msg_size_v2(req->rq_reqbuf->lm_bufcount,
662 req->rq_reqbuf->lm_buflens);
663 req->rq_reqbuf->lm_buflens[PLAIN_PACK_MSG_OFF] = oldsize;
665 /* request from pool should always have enough buffer */
666 LASSERT(!req->rq_pool || req->rq_reqbuf_len >= newbuf_size);
668 if (req->rq_reqbuf_len < newbuf_size) {
669 newbuf_size = size_roundup_power2(newbuf_size);
671 newbuf = libcfs_kvzalloc(newbuf_size, GFP_NOFS);
675 /* Must lock this, so that otherwise unprotected change of
676 * rq_reqmsg is not racing with parallel processing of
677 * imp_replay_list traversing threads. See LU-3333
678 * This is a bandaid at best, we really need to deal with this
679 * in request enlarging code before unpacking that's already
683 spin_lock(&req->rq_import->imp_lock);
685 memcpy(newbuf, req->rq_reqbuf, req->rq_reqbuf_len);
687 kvfree(req->rq_reqbuf);
688 req->rq_reqbuf = newbuf;
689 req->rq_reqbuf_len = newbuf_size;
690 req->rq_reqmsg = lustre_msg_buf(req->rq_reqbuf,
691 PLAIN_PACK_MSG_OFF, 0);
694 spin_unlock(&req->rq_import->imp_lock);
697 _sptlrpc_enlarge_msg_inplace(req->rq_reqbuf, PLAIN_PACK_MSG_OFF,
699 _sptlrpc_enlarge_msg_inplace(req->rq_reqmsg, segment, newsize);
701 req->rq_reqlen = newmsg_size;
705 /****************************************
707 ****************************************/
709 static struct ptlrpc_svc_ctx plain_svc_ctx = {
710 .sc_refcount = ATOMIC_INIT(1),
711 .sc_policy = &plain_policy,
715 int plain_accept(struct ptlrpc_request *req)
717 struct lustre_msg *msg = req->rq_reqbuf;
718 struct plain_header *phdr;
721 LASSERT(SPTLRPC_FLVR_POLICY(req->rq_flvr.sf_rpc) ==
722 SPTLRPC_POLICY_PLAIN);
724 if (SPTLRPC_FLVR_BASE(req->rq_flvr.sf_rpc) !=
725 SPTLRPC_FLVR_BASE(SPTLRPC_FLVR_PLAIN) ||
726 SPTLRPC_FLVR_BULK_TYPE(req->rq_flvr.sf_rpc) !=
727 SPTLRPC_FLVR_BULK_TYPE(SPTLRPC_FLVR_PLAIN)) {
728 CERROR("Invalid rpc flavor %x\n", req->rq_flvr.sf_rpc);
732 if (msg->lm_bufcount < PLAIN_PACK_SEGMENTS) {
733 CERROR("unexpected request buf count %u\n", msg->lm_bufcount);
737 swabbed = ptlrpc_req_need_swab(req);
739 phdr = lustre_msg_buf(msg, PLAIN_PACK_HDR_OFF, sizeof(*phdr));
741 CERROR("missing plain header\n");
745 if (phdr->ph_ver != 0) {
746 CERROR("Invalid header version\n");
750 if (phdr->ph_bulk_hash_alg >= BULK_HASH_ALG_MAX) {
751 CERROR("invalid hash algorithm: %u\n", phdr->ph_bulk_hash_alg);
755 req->rq_sp_from = phdr->ph_sp;
756 req->rq_flvr.u_bulk.hash.hash_alg = phdr->ph_bulk_hash_alg;
758 if (phdr->ph_flags & PLAIN_FL_USER) {
759 if (sptlrpc_unpack_user_desc(msg, PLAIN_PACK_USER_OFF,
761 CERROR("Mal-formed user descriptor\n");
765 req->rq_pack_udesc = 1;
766 req->rq_user_desc = lustre_msg_buf(msg, PLAIN_PACK_USER_OFF, 0);
769 if (phdr->ph_flags & PLAIN_FL_BULK) {
770 if (plain_unpack_bsd(msg, swabbed))
773 req->rq_pack_bulk = 1;
776 req->rq_reqmsg = lustre_msg_buf(msg, PLAIN_PACK_MSG_OFF, 0);
777 req->rq_reqlen = msg->lm_buflens[PLAIN_PACK_MSG_OFF];
779 req->rq_svc_ctx = &plain_svc_ctx;
780 atomic_inc(&req->rq_svc_ctx->sc_refcount);
786 int plain_alloc_rs(struct ptlrpc_request *req, int msgsize)
788 struct ptlrpc_reply_state *rs;
789 __u32 buflens[PLAIN_PACK_SEGMENTS] = { 0, };
790 int rs_size = sizeof(*rs);
792 LASSERT(msgsize % 8 == 0);
794 buflens[PLAIN_PACK_HDR_OFF] = sizeof(struct plain_header);
795 buflens[PLAIN_PACK_MSG_OFF] = msgsize;
797 if (req->rq_pack_bulk && (req->rq_bulk_read || req->rq_bulk_write))
798 buflens[PLAIN_PACK_BULK_OFF] = PLAIN_BSD_SIZE;
800 rs_size += lustre_msg_size_v2(PLAIN_PACK_SEGMENTS, buflens);
802 rs = req->rq_reply_state;
806 LASSERT(rs->rs_size >= rs_size);
808 rs = libcfs_kvzalloc(rs_size, GFP_NOFS);
812 rs->rs_size = rs_size;
815 rs->rs_svc_ctx = req->rq_svc_ctx;
816 atomic_inc(&req->rq_svc_ctx->sc_refcount);
817 rs->rs_repbuf = (struct lustre_msg *)(rs + 1);
818 rs->rs_repbuf_len = rs_size - sizeof(*rs);
820 lustre_init_msg_v2(rs->rs_repbuf, PLAIN_PACK_SEGMENTS, buflens, NULL);
821 rs->rs_msg = lustre_msg_buf_v2(rs->rs_repbuf, PLAIN_PACK_MSG_OFF, 0);
823 req->rq_reply_state = rs;
828 void plain_free_rs(struct ptlrpc_reply_state *rs)
830 LASSERT(atomic_read(&rs->rs_svc_ctx->sc_refcount) > 1);
831 atomic_dec(&rs->rs_svc_ctx->sc_refcount);
833 if (!rs->rs_prealloc)
838 int plain_authorize(struct ptlrpc_request *req)
840 struct ptlrpc_reply_state *rs = req->rq_reply_state;
841 struct lustre_msg_v2 *msg = rs->rs_repbuf;
842 struct plain_header *phdr;
848 if (req->rq_replen != msg->lm_buflens[PLAIN_PACK_MSG_OFF])
849 len = lustre_shrink_msg(msg, PLAIN_PACK_MSG_OFF,
852 len = lustre_msg_size_v2(msg->lm_bufcount, msg->lm_buflens);
854 msg->lm_secflvr = req->rq_flvr.sf_rpc;
856 phdr = lustre_msg_buf(msg, PLAIN_PACK_HDR_OFF, 0);
859 phdr->ph_bulk_hash_alg = req->rq_flvr.u_bulk.hash.hash_alg;
861 if (req->rq_pack_bulk)
862 phdr->ph_flags |= PLAIN_FL_BULK;
864 rs->rs_repdata_len = len;
866 if (likely(req->rq_packed_final)) {
867 if (lustre_msghdr_get_flags(req->rq_reqmsg) & MSGHDR_AT_SUPPORT)
868 req->rq_reply_off = plain_at_offset;
870 req->rq_reply_off = 0;
872 unsigned int hsize = 4;
874 cfs_crypto_hash_digest(CFS_HASH_ALG_CRC32,
875 lustre_msg_buf(msg, PLAIN_PACK_MSG_OFF,
877 lustre_msg_buflen(msg,
879 NULL, 0, (unsigned char *)&msg->lm_cksum,
881 req->rq_reply_off = 0;
888 int plain_svc_unwrap_bulk(struct ptlrpc_request *req,
889 struct ptlrpc_bulk_desc *desc)
891 struct ptlrpc_reply_state *rs = req->rq_reply_state;
892 struct ptlrpc_bulk_sec_desc *bsdr, *bsdv;
893 struct plain_bulk_token *tokenr;
896 LASSERT(req->rq_bulk_write);
897 LASSERT(req->rq_pack_bulk);
899 bsdr = lustre_msg_buf(req->rq_reqbuf, PLAIN_PACK_BULK_OFF, 0);
900 tokenr = (struct plain_bulk_token *)bsdr->bsd_data;
901 bsdv = lustre_msg_buf(rs->rs_repbuf, PLAIN_PACK_BULK_OFF, 0);
903 bsdv->bsd_version = 0;
904 bsdv->bsd_type = SPTLRPC_BULK_DEFAULT;
905 bsdv->bsd_svc = bsdr->bsd_svc;
908 if (bsdr->bsd_svc == SPTLRPC_BULK_SVC_NULL)
911 rc = plain_verify_bulk_csum(desc, req->rq_flvr.u_bulk.hash.hash_alg,
914 bsdv->bsd_flags |= BSD_FL_ERR;
915 CERROR("bulk write: server verify failed: %d\n", rc);
922 int plain_svc_wrap_bulk(struct ptlrpc_request *req,
923 struct ptlrpc_bulk_desc *desc)
925 struct ptlrpc_reply_state *rs = req->rq_reply_state;
926 struct ptlrpc_bulk_sec_desc *bsdr, *bsdv;
927 struct plain_bulk_token *tokenv;
930 LASSERT(req->rq_bulk_read);
931 LASSERT(req->rq_pack_bulk);
933 bsdr = lustre_msg_buf(req->rq_reqbuf, PLAIN_PACK_BULK_OFF, 0);
934 bsdv = lustre_msg_buf(rs->rs_repbuf, PLAIN_PACK_BULK_OFF, 0);
935 tokenv = (struct plain_bulk_token *)bsdv->bsd_data;
937 bsdv->bsd_version = 0;
938 bsdv->bsd_type = SPTLRPC_BULK_DEFAULT;
939 bsdv->bsd_svc = bsdr->bsd_svc;
942 if (bsdr->bsd_svc == SPTLRPC_BULK_SVC_NULL)
945 rc = plain_generate_bulk_csum(desc, req->rq_flvr.u_bulk.hash.hash_alg,
948 CERROR("bulk read: server failed to compute checksum: %d\n",
951 if (OBD_FAIL_CHECK(OBD_FAIL_OSC_CHECKSUM_RECEIVE))
952 corrupt_bulk_data(desc);
958 static struct ptlrpc_ctx_ops plain_ctx_ops = {
959 .refresh = plain_ctx_refresh,
960 .validate = plain_ctx_validate,
961 .sign = plain_ctx_sign,
962 .verify = plain_ctx_verify,
963 .wrap_bulk = plain_cli_wrap_bulk,
964 .unwrap_bulk = plain_cli_unwrap_bulk,
967 static struct ptlrpc_sec_cops plain_sec_cops = {
968 .create_sec = plain_create_sec,
969 .destroy_sec = plain_destroy_sec,
970 .kill_sec = plain_kill_sec,
971 .lookup_ctx = plain_lookup_ctx,
972 .release_ctx = plain_release_ctx,
973 .flush_ctx_cache = plain_flush_ctx_cache,
974 .alloc_reqbuf = plain_alloc_reqbuf,
975 .free_reqbuf = plain_free_reqbuf,
976 .alloc_repbuf = plain_alloc_repbuf,
977 .free_repbuf = plain_free_repbuf,
978 .enlarge_reqbuf = plain_enlarge_reqbuf,
981 static struct ptlrpc_sec_sops plain_sec_sops = {
982 .accept = plain_accept,
983 .alloc_rs = plain_alloc_rs,
984 .authorize = plain_authorize,
985 .free_rs = plain_free_rs,
986 .unwrap_bulk = plain_svc_unwrap_bulk,
987 .wrap_bulk = plain_svc_wrap_bulk,
990 static struct ptlrpc_sec_policy plain_policy = {
991 .sp_owner = THIS_MODULE,
993 .sp_policy = SPTLRPC_POLICY_PLAIN,
994 .sp_cops = &plain_sec_cops,
995 .sp_sops = &plain_sec_sops,
998 int sptlrpc_plain_init(void)
1000 __u32 buflens[PLAIN_PACK_SEGMENTS] = { 0, };
1003 buflens[PLAIN_PACK_MSG_OFF] = lustre_msg_early_size();
1004 plain_at_offset = lustre_msg_size_v2(PLAIN_PACK_SEGMENTS, buflens);
1006 rc = sptlrpc_register_policy(&plain_policy);
1008 CERROR("failed to register: %d\n", rc);
1013 void sptlrpc_plain_fini(void)
1017 rc = sptlrpc_unregister_policy(&plain_policy);
1019 CERROR("cannot unregister: %d\n", rc);