X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/43866e378188c25dd1e2208016ab3cbeb086ae6c..5eebf7385fedb1517b66b53c28e5aa6bb0a2be50:/bsd/netinet/raw_ip.c

diff --git a/bsd/netinet/raw_ip.c b/bsd/netinet/raw_ip.c
index f524dd41a..60f2ce01b 100644
--- a/bsd/netinet/raw_ip.c
+++ b/bsd/netinet/raw_ip.c
@@ -3,22 +3,19 @@
  *
  * @APPLE_LICENSE_HEADER_START@
  * 
- * Copyright (c) 1999-2003 Apple Computer, Inc.  All Rights Reserved.
+ * The contents of this file constitute Original Code as defined in and
+ * are subject to the Apple Public Source License Version 1.1 (the
+ * "License").  You may not use this file except in compliance with the
+ * License.  Please obtain a copy of the License at
+ * http://www.apple.com/publicsource and read it before using this file.
  * 
- * This file contains Original Code and/or Modifications of Original Code
- * as defined in and that are subject to the Apple Public Source License
- * Version 2.0 (the 'License'). You may not use this file except in
- * compliance with the License. Please obtain a copy of the License at
- * http://www.opensource.apple.com/apsl/ and read it before using this
- * file.
- * 
- * The Original Code and all software distributed under the License are
- * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * This Original Code and all software distributed under the License are
+ * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
- * Please see the License for the specific language governing rights and
- * limitations under the License.
+ * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
+ * License for the specific language governing rights and limitations
+ * under the License.
  * 
  * @APPLE_LICENSE_HEADER_END@
  */
@@ -300,6 +297,23 @@ rip_output(m, so, dst)
 			  inp->inp_moptions));
 }
 
+int
+load_ipfw()
+{
+	kern_return_t	err;
+	
+	/* Load the kext by the identifier */
+	err = kmod_load_extension("com.apple.nke.IPFirewall");
+	if (err) return err;
+	
+	if (ip_fw_ctl_ptr == NULL) {
+		/* Wait for the kext to finish loading */
+		err = tsleep(&ip_fw_ctl_ptr, PWAIT | PCATCH, "load_ipfw_kext", 5 * 60 /* 5 seconds */);
+	}
+	
+	return err == 0 && ip_fw_ctl_ptr == NULL ? -1 : err;
+}
+
 /*
  * Raw IP socket option processing.
  */
@@ -334,9 +348,11 @@ rip_ctloutput(so, sopt)
 		case IP_OLD_FW_ADD:
 		case IP_OLD_FW_GET:
 			if (ip_fw_ctl_ptr == 0)
-				error = ENOPROTOOPT;
-			else
+				error = load_ipfw();
+			if (ip_fw_ctl_ptr && error == 0)
 				error = ip_fw_ctl_ptr(sopt);
+			else
+				error = ENOPROTOOPT;
 			break;
 
 #if DUMMYNET
@@ -401,9 +417,11 @@ rip_ctloutput(so, sopt)
 		case IP_OLD_FW_ZERO:
 		case IP_OLD_FW_RESETLOG:
 			if (ip_fw_ctl_ptr == 0)
-				error = ENOPROTOOPT;
-			else
+				error = load_ipfw();
+			if (ip_fw_ctl_ptr && error == 0)
 				error = ip_fw_ctl_ptr(sopt);
+			else
+				error = ENOPROTOOPT;
 			break;
 
 #if DUMMYNET