1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * Copyright (c) 2021, 2022 Oracle. All rights reserved.
4 *
5 * The AUTH_TLS credential is used only to probe a remote peer
6 * for RPC-over-TLS support.
7 */
8
9#include <linux/types.h>
10#include <linux/module.h>
11#include <linux/sunrpc/clnt.h>
12
13static const char *starttls_token = "STARTTLS";
14static const size_t starttls_len = 8;
15
16static struct rpc_auth tls_auth;
17static struct rpc_cred tls_cred;
18
19static void tls_encode_probe(struct rpc_rqst *rqstp, struct xdr_stream *xdr,
20 const void *obj)
21{
22}
23
24static int tls_decode_probe(struct rpc_rqst *rqstp, struct xdr_stream *xdr,
25 void *obj)
26{
27 return 0;
28}
29
30static const struct rpc_procinfo rpcproc_tls_probe = {
31 .p_encode = tls_encode_probe,
32 .p_decode = tls_decode_probe,
33};
34
35static void rpc_tls_probe_call_prepare(struct rpc_task *task, void *data)
36{
37 task->tk_flags &= ~RPC_TASK_NO_RETRANS_TIMEOUT;
38 rpc_call_start(task);
39}
40
41static void rpc_tls_probe_call_done(struct rpc_task *task, void *data)
42{
43}
44
45static const struct rpc_call_ops rpc_tls_probe_ops = {
46 .rpc_call_prepare = rpc_tls_probe_call_prepare,
47 .rpc_call_done = rpc_tls_probe_call_done,
48};
49
50static int tls_probe(struct rpc_clnt *clnt)
51{
52 struct rpc_message msg = {
53 .rpc_proc = &rpcproc_tls_probe,
54 };
55 struct rpc_task_setup task_setup_data = {
56 .rpc_client = clnt,
57 .rpc_message = &msg,
58 .rpc_op_cred = &tls_cred,
59 .callback_ops = &rpc_tls_probe_ops,
60 .flags = RPC_TASK_SOFT | RPC_TASK_SOFTCONN,
61 };
62 struct rpc_task *task;
63 int status;
64
65 task = rpc_run_task(&task_setup_data);
66 if (IS_ERR(ptr: task))
67 return PTR_ERR(ptr: task);
68 status = task->tk_status;
69 rpc_put_task(task);
70 return status;
71}
72
73static struct rpc_auth *tls_create(const struct rpc_auth_create_args *args,
74 struct rpc_clnt *clnt)
75{
76 refcount_inc(r: &tls_auth.au_count);
77 return &tls_auth;
78}
79
80static void tls_destroy(struct rpc_auth *auth)
81{
82}
83
84static struct rpc_cred *tls_lookup_cred(struct rpc_auth *auth,
85 struct auth_cred *acred, int flags)
86{
87 return get_rpccred(cred: &tls_cred);
88}
89
90static void tls_destroy_cred(struct rpc_cred *cred)
91{
92}
93
94static int tls_match(struct auth_cred *acred, struct rpc_cred *cred, int taskflags)
95{
96 return 1;
97}
98
99static int tls_marshal(struct rpc_task *task, struct xdr_stream *xdr)
100{
101 __be32 *p;
102
103 p = xdr_reserve_space(xdr, nbytes: 4 * XDR_UNIT);
104 if (!p)
105 return -EMSGSIZE;
106 /* Credential */
107 *p++ = rpc_auth_tls;
108 *p++ = xdr_zero;
109 /* Verifier */
110 *p++ = rpc_auth_null;
111 *p = xdr_zero;
112 return 0;
113}
114
115static int tls_refresh(struct rpc_task *task)
116{
117 set_bit(RPCAUTH_CRED_UPTODATE, addr: &task->tk_rqstp->rq_cred->cr_flags);
118 return 0;
119}
120
121static int tls_validate(struct rpc_task *task, struct xdr_stream *xdr)
122{
123 __be32 *p;
124 void *str;
125
126 p = xdr_inline_decode(xdr, XDR_UNIT);
127 if (!p)
128 return -EIO;
129 if (*p != rpc_auth_null)
130 return -EIO;
131 if (xdr_stream_decode_opaque_inline(xdr, ptr: &str, maxlen: starttls_len) != starttls_len)
132 return -EPROTONOSUPPORT;
133 if (memcmp(p: str, q: starttls_token, size: starttls_len))
134 return -EPROTONOSUPPORT;
135 return 0;
136}
137
138const struct rpc_authops authtls_ops = {
139 .owner = THIS_MODULE,
140 .au_flavor = RPC_AUTH_TLS,
141 .au_name = "NULL",
142 .create = tls_create,
143 .destroy = tls_destroy,
144 .lookup_cred = tls_lookup_cred,
145 .ping = tls_probe,
146};
147
148static struct rpc_auth tls_auth = {
149 .au_cslack = NUL_CALLSLACK,
150 .au_rslack = NUL_REPLYSLACK,
151 .au_verfsize = NUL_REPLYSLACK,
152 .au_ralign = NUL_REPLYSLACK,
153 .au_ops = &authtls_ops,
154 .au_flavor = RPC_AUTH_TLS,
155 .au_count = REFCOUNT_INIT(1),
156};
157
158static const struct rpc_credops tls_credops = {
159 .cr_name = "AUTH_TLS",
160 .crdestroy = tls_destroy_cred,
161 .crmatch = tls_match,
162 .crmarshal = tls_marshal,
163 .crwrap_req = rpcauth_wrap_req_encode,
164 .crrefresh = tls_refresh,
165 .crvalidate = tls_validate,
166 .crunwrap_resp = rpcauth_unwrap_resp_decode,
167};
168
169static struct rpc_cred tls_cred = {
170 .cr_lru = LIST_HEAD_INIT(tls_cred.cr_lru),
171 .cr_auth = &tls_auth,
172 .cr_ops = &tls_credops,
173 .cr_count = REFCOUNT_INIT(2),
174 .cr_flags = 1UL << RPCAUTH_CRED_UPTODATE,
175};
176

source code of linux/net/sunrpc/auth_tls.c