Author: kp
Date: Sat Apr 14 00:20:47 2018
New Revision: 332497
URL: https://svnweb.freebsd.org/changeset/base/332497

Log:
  MFC r332142:
  
  pf: Improve ioctl validation
  
  Ensure that multiplications for memory allocations cannot overflow, and
  that we'll not try to allocate M_WAITOK for potentially overly large
  allocations.

Modified:
  stable/10/sys/netpfil/pf/pf_ioctl.c
Directory Properties:
  stable/10/   (props changed)

Modified: stable/10/sys/netpfil/pf/pf_ioctl.c
==============================================================================
--- stable/10/sys/netpfil/pf/pf_ioctl.c Sat Apr 14 00:12:16 2018        
(r332496)
+++ stable/10/sys/netpfil/pf/pf_ioctl.c Sat Apr 14 00:20:47 2018        
(r332497)
@@ -2723,9 +2723,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->pfrio_size < 0 ||
+                   WOULD_OVERFLOW(io->pfrio_size, sizeof(struct pfr_addr))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = io->pfrio_size * sizeof(struct pfr_addr);
                pfras = mallocarray(io->pfrio_size, sizeof(struct pfr_addr),
-                   M_TEMP, M_WAITOK);
+                   M_TEMP, M_NOWAIT);
                if (! pfras) {
                        error = ENOMEM;
                        break;
@@ -2755,9 +2760,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->pfrio_size < 0 ||
+                   WOULD_OVERFLOW(io->pfrio_size, sizeof(struct pfr_addr))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = io->pfrio_size * sizeof(struct pfr_addr);
                pfras = mallocarray(io->pfrio_size, sizeof(struct pfr_addr),
-                   M_TEMP, M_WAITOK);
+                   M_TEMP, M_NOWAIT);
                if (! pfras) {
                        error = ENOMEM;
                        break;
@@ -2787,10 +2797,18 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->pfrio_size < 0 || io->pfrio_size2 < 0) {
+                       error = EINVAL;
+                       break;
+               }
                count = max(io->pfrio_size, io->pfrio_size2);
+               if (WOULD_OVERFLOW(count, sizeof(struct pfr_addr))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = count * sizeof(struct pfr_addr);
                pfras = mallocarray(count, sizeof(struct pfr_addr), M_TEMP,
-                   M_WAITOK);
+                   M_NOWAIT);
                if (! pfras) {
                        error = ENOMEM;
                        break;
@@ -2821,9 +2839,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->pfrio_size < 0 ||
+                   WOULD_OVERFLOW(io->pfrio_size, sizeof(struct pfr_addr))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = io->pfrio_size * sizeof(struct pfr_addr);
                pfras = mallocarray(io->pfrio_size, sizeof(struct pfr_addr),
-                   M_TEMP, M_WAITOK);
+                   M_TEMP, M_NOWAIT);
                if (! pfras) {
                        error = ENOMEM;
                        break;
@@ -2847,9 +2870,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->pfrio_size < 0 ||
+                   WOULD_OVERFLOW(io->pfrio_size, sizeof(struct pfr_astats))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = io->pfrio_size * sizeof(struct pfr_astats);
                pfrastats = mallocarray(io->pfrio_size,
-                   sizeof(struct pfr_astats), M_TEMP, M_WAITOK);
+                   sizeof(struct pfr_astats), M_TEMP, M_NOWAIT);
                if (! pfrastats) {
                        error = ENOMEM;
                        break;
@@ -2873,9 +2901,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->pfrio_size < 0 ||
+                   WOULD_OVERFLOW(io->pfrio_size, sizeof(struct pfr_addr))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = io->pfrio_size * sizeof(struct pfr_addr);
                pfras = mallocarray(io->pfrio_size, sizeof(struct pfr_addr),
-                   M_TEMP, M_WAITOK);
+                   M_TEMP, M_NOWAIT);
                if (! pfras) {
                        error = ENOMEM;
                        break;
@@ -2905,9 +2938,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->pfrio_size < 0 ||
+                   WOULD_OVERFLOW(io->pfrio_size, sizeof(struct pfr_addr))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = io->pfrio_size * sizeof(struct pfr_addr);
                pfras = mallocarray(io->pfrio_size, sizeof(struct pfr_addr),
-                   M_TEMP, M_WAITOK);
+                   M_TEMP, M_NOWAIT);
                if (! pfras) {
                        error = ENOMEM;
                        break;
@@ -2937,9 +2975,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->pfrio_size < 0 ||
+                   WOULD_OVERFLOW(io->pfrio_size, sizeof(struct pfr_addr))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = io->pfrio_size * sizeof(struct pfr_addr);
                pfras = mallocarray(io->pfrio_size, sizeof(struct pfr_addr),
-                   M_TEMP, M_WAITOK);
+                   M_TEMP, M_NOWAIT);
                if (! pfras) {
                        error = ENOMEM;
                        break;
@@ -2984,9 +3027,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->size < 0 ||
+                   WOULD_OVERFLOW(io->size, sizeof(struct pfioc_trans_e))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = sizeof(struct pfioc_trans_e) * io->size;
                ioes = mallocarray(io->size, sizeof(struct pfioc_trans_e),
-                   M_TEMP, M_WAITOK);
+                   M_TEMP, M_NOWAIT);
                if (! ioes) {
                        error = ENOMEM;
                        break;
@@ -3055,9 +3103,14 @@ DIOCCHANGEADDR_error:
                        error = ENODEV;
                        break;
                }
+               if (io->size < 0 ||
+                   WOULD_OVERFLOW(io->size, sizeof(struct pfioc_trans_e))) {
+                       error = EINVAL;
+                       break;
+               }
                totlen = sizeof(struct pfioc_trans_e) * io->size;
                ioes = mallocarray(io->size, sizeof(struct pfioc_trans_e),
-                   M_TEMP, M_WAITOK);
+                   M_TEMP, M_NOWAIT);
                if (! ioes) {
                        error = ENOMEM;
                        break;
_______________________________________________
svn-src-all@freebsd.org mailing list
https://lists.freebsd.org/mailman/listinfo/svn-src-all
To unsubscribe, send any mail to "svn-src-all-unsubscr...@freebsd.org"

Reply via email to