Commit 4d2fba3a authored by Christoph Hellwig's avatar Christoph Hellwig Committed by David S. Miller
Browse files

sctp: pass a kernel pointer to sctp_setsockopt_scheduler



Use the kernel pointer that sctp_setsockopt has available instead of
directly handling the user pointer.

Signed-off-by: default avatarChristoph Hellwig <hch@lst.de>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent 4d6fb260
Loading
Loading
Loading
Loading
+14 −19
Original line number Diff line number Diff line
@@ -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;
@@ -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);