libceph: store ceph_auth_handshake pointer in ceph_connection

We already copy authorizer_reply_buf and authorizer_reply_buf_len into
ceph_connection.  Factoring out __prepare_write_connect() requires two
more: authorizer_buf and authorizer_buf_len.  Store the pointer to the
handshake in con->auth rather than piling on.

Signed-off-by: Ilya Dryomov <idryomov@gmail.com>
Reviewed-by: Sage Weil <sage@redhat.com>
This commit is contained in:
Ilya Dryomov 2018-07-26 15:17:46 +02:00
parent 24e1dd6afd
commit 262614c429
2 changed files with 27 additions and 30 deletions

View File

@ -286,9 +286,8 @@ struct ceph_connection {
attempt for this connection, client */ attempt for this connection, client */
u32 peer_global_seq; /* peer's global seq for this connection */ u32 peer_global_seq; /* peer's global seq for this connection */
struct ceph_auth_handshake *auth;
int auth_retry; /* true if we need a newer authorizer */ int auth_retry; /* true if we need a newer authorizer */
void *auth_reply_buf; /* where to put the authorizer reply */
int auth_reply_buf_len;
struct mutex mutex; struct mutex mutex;

View File

@ -1434,24 +1434,26 @@ static void prepare_write_keepalive(struct ceph_connection *con)
* Connection negotiation. * Connection negotiation.
*/ */
static struct ceph_auth_handshake *get_connect_authorizer(struct ceph_connection *con, static int get_connect_authorizer(struct ceph_connection *con)
int *auth_proto)
{ {
struct ceph_auth_handshake *auth; struct ceph_auth_handshake *auth;
int auth_proto;
if (!con->ops->get_authorizer) { if (!con->ops->get_authorizer) {
con->auth = NULL;
con->out_connect.authorizer_protocol = CEPH_AUTH_UNKNOWN; con->out_connect.authorizer_protocol = CEPH_AUTH_UNKNOWN;
con->out_connect.authorizer_len = 0; con->out_connect.authorizer_len = 0;
return NULL; return 0;
} }
auth = con->ops->get_authorizer(con, auth_proto, con->auth_retry); auth = con->ops->get_authorizer(con, &auth_proto, con->auth_retry);
if (IS_ERR(auth)) if (IS_ERR(auth))
return auth; return PTR_ERR(auth);
con->auth_reply_buf = auth->authorizer_reply_buf; con->auth = auth;
con->auth_reply_buf_len = auth->authorizer_reply_buf_len; con->out_connect.authorizer_protocol = cpu_to_le32(auth_proto);
return auth; con->out_connect.authorizer_len = cpu_to_le32(auth->authorizer_buf_len);
return 0;
} }
/* /*
@ -1471,8 +1473,7 @@ static int prepare_write_connect(struct ceph_connection *con)
{ {
unsigned int global_seq = get_global_seq(con->msgr, 0); unsigned int global_seq = get_global_seq(con->msgr, 0);
int proto; int proto;
int auth_proto; int ret;
struct ceph_auth_handshake *auth;
switch (con->peer_name.type) { switch (con->peer_name.type) {
case CEPH_ENTITY_TYPE_MON: case CEPH_ENTITY_TYPE_MON:
@ -1499,20 +1500,15 @@ static int prepare_write_connect(struct ceph_connection *con)
con->out_connect.protocol_version = cpu_to_le32(proto); con->out_connect.protocol_version = cpu_to_le32(proto);
con->out_connect.flags = 0; con->out_connect.flags = 0;
auth_proto = CEPH_AUTH_UNKNOWN; ret = get_connect_authorizer(con);
auth = get_connect_authorizer(con, &auth_proto); if (ret)
if (IS_ERR(auth)) return ret;
return PTR_ERR(auth);
con->out_connect.authorizer_protocol = cpu_to_le32(auth_proto);
con->out_connect.authorizer_len = auth ?
cpu_to_le32(auth->authorizer_buf_len) : 0;
con_out_kvec_add(con, sizeof (con->out_connect), con_out_kvec_add(con, sizeof (con->out_connect),
&con->out_connect); &con->out_connect);
if (auth && auth->authorizer_buf_len) if (con->auth)
con_out_kvec_add(con, auth->authorizer_buf_len, con_out_kvec_add(con, con->auth->authorizer_buf_len,
auth->authorizer_buf); con->auth->authorizer_buf);
con->out_more = 0; con->out_more = 0;
con_flag_set(con, CON_FLAG_WRITE_PENDING); con_flag_set(con, CON_FLAG_WRITE_PENDING);
@ -1781,11 +1777,14 @@ static int read_partial_connect(struct ceph_connection *con)
if (ret <= 0) if (ret <= 0)
goto out; goto out;
size = le32_to_cpu(con->in_reply.authorizer_len); if (con->auth) {
end += size; size = le32_to_cpu(con->in_reply.authorizer_len);
ret = read_partial(con, end, size, con->auth_reply_buf); end += size;
if (ret <= 0) ret = read_partial(con, end, size,
goto out; con->auth->authorizer_reply_buf);
if (ret <= 0)
goto out;
}
dout("read_partial_connect %p tag %d, con_seq = %u, g_seq = %u\n", dout("read_partial_connect %p tag %d, con_seq = %u, g_seq = %u\n",
con, (int)con->in_reply.tag, con, (int)con->in_reply.tag,
@ -1793,7 +1792,6 @@ static int read_partial_connect(struct ceph_connection *con)
le32_to_cpu(con->in_reply.global_seq)); le32_to_cpu(con->in_reply.global_seq));
out: out:
return ret; return ret;
} }
/* /*
@ -2076,7 +2074,7 @@ static int process_connect(struct ceph_connection *con)
dout("process_connect on %p tag %d\n", con, (int)con->in_tag); dout("process_connect on %p tag %d\n", con, (int)con->in_tag);
if (con->auth_reply_buf) { if (con->auth) {
/* /*
* Any connection that defines ->get_authorizer() * Any connection that defines ->get_authorizer()
* should also define ->verify_authorizer_reply(). * should also define ->verify_authorizer_reply().