Skip to content

Commit

Permalink
sctp: pass a kernel pointer to sctp_setsockopt_scheduler
Browse files Browse the repository at this point in the history
Use the kernel pointer that sctp_setsockopt has available instead of
directly handling the user pointer.

Signed-off-by: Christoph Hellwig <hch@lst.de>
Signed-off-by: David S. Miller <davem@davemloft.net>
  • Loading branch information
Christoph Hellwig authored and David S. Miller committed Jul 20, 2020
1 parent 4d6fb26 commit 4d2fba3
Showing 1 changed file with 14 additions and 19 deletions.
33 changes: 14 additions & 19 deletions net/sctp/socket.c
Original file line number Diff line number Diff line change
Expand Up @@ -4115,44 +4115,39 @@ static int sctp_setsockopt_add_streams(struct sock *sk,
}

static int sctp_setsockopt_scheduler(struct sock *sk,
char __user *optval,
struct sctp_assoc_value *params,
unsigned int optlen)
{
struct sctp_sock *sp = sctp_sk(sk);
struct sctp_association *asoc;
struct sctp_assoc_value params;
int retval = 0;

if (optlen < sizeof(params))
if (optlen < sizeof(*params))
return -EINVAL;

optlen = sizeof(params);
if (copy_from_user(&params, optval, optlen))
return -EFAULT;

if (params.assoc_value > SCTP_SS_MAX)
if (params->assoc_value > SCTP_SS_MAX)
return -EINVAL;

asoc = sctp_id2assoc(sk, params.assoc_id);
if (!asoc && params.assoc_id > SCTP_ALL_ASSOC &&
asoc = sctp_id2assoc(sk, params->assoc_id);
if (!asoc && params->assoc_id > SCTP_ALL_ASSOC &&
sctp_style(sk, UDP))
return -EINVAL;

if (asoc)
return sctp_sched_set_sched(asoc, params.assoc_value);
return sctp_sched_set_sched(asoc, params->assoc_value);

if (sctp_style(sk, TCP))
params.assoc_id = SCTP_FUTURE_ASSOC;
params->assoc_id = SCTP_FUTURE_ASSOC;

if (params.assoc_id == SCTP_FUTURE_ASSOC ||
params.assoc_id == SCTP_ALL_ASSOC)
sp->default_ss = params.assoc_value;
if (params->assoc_id == SCTP_FUTURE_ASSOC ||
params->assoc_id == SCTP_ALL_ASSOC)
sp->default_ss = params->assoc_value;

if (params.assoc_id == SCTP_CURRENT_ASSOC ||
params.assoc_id == SCTP_ALL_ASSOC) {
if (params->assoc_id == SCTP_CURRENT_ASSOC ||
params->assoc_id == SCTP_ALL_ASSOC) {
list_for_each_entry(asoc, &sp->ep->asocs, asocs) {
int ret = sctp_sched_set_sched(asoc,
params.assoc_value);
params->assoc_value);

if (ret && !retval)
retval = ret;
Expand Down Expand Up @@ -4660,7 +4655,7 @@ static int sctp_setsockopt(struct sock *sk, int level, int optname,
retval = sctp_setsockopt_add_streams(sk, kopt, optlen);
break;
case SCTP_STREAM_SCHEDULER:
retval = sctp_setsockopt_scheduler(sk, optval, optlen);
retval = sctp_setsockopt_scheduler(sk, kopt, optlen);
break;
case SCTP_STREAM_SCHEDULER_VALUE:
retval = sctp_setsockopt_scheduler_value(sk, optval, optlen);
Expand Down

0 comments on commit 4d2fba3

Please sign in to comment.