xnu-3247.1.106.tar.gz

[apple/xnu.git] / bsd / kern / kpi_socketfilter.c

diff --git a/bsd/kern/kpi_socketfilter.c b/bsd/kern/kpi_socketfilter.c
index ac9dfcb1c5c5f41b02f97446b16fb3339dbd6a4b..bdebcc1825d875e628e53c11d7f8d3f90144d8a4 100644 (file)
--- a/bsd/kern/kpi_socketfilter.c
+++ b/bsd/kern/kpi_socketfilter.c
@@ -1256,12 +1256,16 @@ sflt_register(const struct sflt_filter *filter, int domain, int type,
        struct socket_filter *sock_filt = NULL;
        struct socket_filter *match = NULL;
        int error = 0;
-       struct protosw *pr = pffindproto(domain, protocol, type);
+       struct protosw *pr;
        unsigned int len;
        struct socket *so;
        struct inpcb *inp;
        struct solist *solisthead = NULL, *solist = NULL;
 
+       if ((domain != PF_INET) && (domain != PF_INET6))
+               return (ENOTSUP);
+
+       pr = pffindproto(domain, protocol, type);
        if (pr == NULL)
                return (ENOENT);