sctp: fix a potential OOB access in sctp_sched_set_sched()

[platform/kernel/linux-rpi.git] / net / sctp / stream_sched.c

diff --git a/net/sctp/stream_sched.c b/net/sctp/stream_sched.c
index e843760..54afbe4 100644 (file)
--- a/net/sctp/stream_sched.c
+++ b/net/sctp/stream_sched.c
@@ -148,18 +148,19 @@ static void sctp_sched_free_sched(struct sctp_stream *stream)
 int sctp_sched_set_sched(struct sctp_association *asoc,
                         enum sctp_sched_type sched)
 {
-       struct sctp_sched_ops *n = sctp_sched_ops[sched];
        struct sctp_sched_ops *old = asoc->outqueue.sched;
        struct sctp_datamsg *msg = NULL;
+       struct sctp_sched_ops *n;
        struct sctp_chunk *ch;
        int i, ret = 0;
 
-       if (old == n)
-               return ret;
-
        if (sched > SCTP_SS_MAX)
                return -EINVAL;
 
+       n = sctp_sched_ops[sched];
+       if (old == n)
+               return ret;
+
        if (old)
                sctp_sched_free_sched(&asoc->stream);