]> git.saurik.com Git - apple/mdnsresponder.git/blobdiff - mDNSShared/dnssd_ipc.c
mDNSResponder-161.1.tar.gz
[apple/mdnsresponder.git] / mDNSShared / dnssd_ipc.c
index 6a0236df98a2c982bdc18d7ba2aa3aa815b6659a..2e167471c6dbfbf125da3ce49349639c03e53552 100644 (file)
@@ -1,32 +1,48 @@
-/*
+/* -*- Mode: C; tab-width: 4 -*-
+ *
  * Copyright (c) 2003-2004, Apple Computer, Inc. All rights reserved.
  *
- * Redistribution and use in source and binary forms, with or without 
+ * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are met:
  *
- * 1.  Redistributions of source code must retain the above copyright notice, 
- *     this list of conditions and the following disclaimer. 
- * 2.  Redistributions in binary form must reproduce the above copyright notice, 
- *     this list of conditions and the following disclaimer in the documentation 
- *     and/or other materials provided with the distribution. 
- * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of its 
- *     contributors may be used to endorse or promote products derived from this 
- *     software without specific prior written permission. 
+ * 1.  Redistributions of source code must retain the above copyright notice,
+ *     this list of conditions and the following disclaimer.
+ * 2.  Redistributions in binary form must reproduce the above copyright notice,
+ *     this list of conditions and the following disclaimer in the documentation
+ *     and/or other materials provided with the distribution.
+ * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of its
+ *     contributors may be used to endorse or promote products derived from this
+ *     software without specific prior written permission.
  *
- * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 
- * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 
- * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 
- * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 
- * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 
+ * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
        Change History (most recent first):
 
 $Log: dnssd_ipc.c,v $
+Revision 1.20  2007/07/23 22:12:53  cheshire
+<rdar://problem/5352299> Make mDNSResponder more defensive against malicious local clients
+
+Revision 1.19  2007/05/16 01:06:52  cheshire
+<rdar://problem/4471320> Improve reliability of kDNSServiceFlagsMoreComing flag on multiprocessor machines
+
+Revision 1.18  2007/03/21 19:01:57  cheshire
+<rdar://problem/5078494> IPC code not 64-bit-savvy: assumes long=32bits, and short=16bits
+
+Revision 1.17  2006/10/27 00:38:22  cheshire
+Strip accidental trailing whitespace from lines
+
+Revision 1.16  2006/08/14 23:05:53  cheshire
+Added "tab-width" emacs header line
+
 Revision 1.15  2005/01/27 22:57:56  cheshire
 Fix compile errors on gcc4
 
@@ -59,7 +75,7 @@ Update to APSL 2.0
 
 #include "dnssd_ipc.h"
 
-void put_long(const uint32_t l, char **ptr)
+void put_uint32(const uint32_t l, char **ptr)
        {
        (*ptr)[0] = (char)((l >> 24) &  0xFF);
        (*ptr)[1] = (char)((l >> 16) &  0xFF);
@@ -68,25 +84,41 @@ void put_long(const uint32_t l, char **ptr)
        *ptr += sizeof(uint32_t);
        }
 
-uint32_t get_long(char **ptr)
+uint32_t get_uint32(char **ptr, char *end)
        {
-       uint8_t *p = (uint8_t*) *ptr;
-       *ptr += sizeof(uint32_t);
-       return((uint32_t) ((uint32_t)p[0] << 24 | (uint32_t)p[1] << 16 | (uint32_t)p[2] << 8 | p[3]));
+       if (!*ptr || *ptr + sizeof(uint32_t) > end)
+               {
+               *ptr = NULL;
+               return(0);
+               }
+       else
+               {
+               uint8_t *p = (uint8_t*) *ptr;
+               *ptr += sizeof(uint32_t);
+               return((uint32_t) ((uint32_t)p[0] << 24 | (uint32_t)p[1] << 16 | (uint32_t)p[2] << 8 | p[3]));
+               }
        }
 
-void put_short(uint16_t s, char **ptr)
+void put_uint16(uint16_t s, char **ptr)
        {
        (*ptr)[0] = (char)((s >>  8) &  0xFF);
        (*ptr)[1] = (char)((s      ) &  0xFF);
        *ptr += sizeof(uint16_t);
        }
 
-uint16_t get_short(char **ptr)
+uint16_t get_uint16(char **ptr, char *end)
        {
-       uint8_t *p = (uint8_t*) *ptr;
-       *ptr += sizeof(uint16_t);
-       return((uint16_t) ((uint16_t)p[0] << 8 | p[1]));
+       if (!*ptr || *ptr + sizeof(uint16_t) > end)
+               {
+               *ptr = NULL;
+               return(0);
+               }
+       else
+               {
+               uint8_t *p = (uint8_t*) *ptr;
+               *ptr += sizeof(uint16_t);
+               return((uint16_t) ((uint16_t)p[0] << 8 | p[1]));
+               }
        }
 
 int put_string(const char *str, char **ptr)
@@ -97,13 +129,26 @@ int put_string(const char *str, char **ptr)
        return 0;
        }
 
-int get_string(char **ptr, char *buffer, int buflen)
+int get_string(char **ptr, char *end, char *buffer, int buflen)
        {
-       int overrun = (int)strlen(*ptr) <  buflen ? 0 : -1;
-       strncpy(buffer, *ptr,  buflen - 1);
-       buffer[buflen - 1] = '\0';
-       *ptr += strlen(buffer) + 1;
-       return overrun;
+       if (!*ptr)
+               {
+               *buffer = 0;
+               return(-1);
+               }
+       else
+               {
+               char *lim = buffer + buflen;    // Calculate limit
+               while (*ptr < end && buffer < lim)
+                       {
+                       char c = *buffer++ = *(*ptr)++;
+                       if (c == 0) return(0);          // Success
+                       }
+               if (buffer == lim) buffer--;
+               *buffer = 0;                                    // Failed, so terminate string,
+               *ptr = NULL;                                    // clear pointer,
+               return(-1);                                             // and return failure indication
+               }
        }
 
 void put_rdata(const int rdlen, const unsigned char *rdata, char **ptr)
@@ -112,18 +157,26 @@ void put_rdata(const int rdlen, const unsigned char *rdata, char **ptr)
        *ptr += rdlen;
        }
 
-char *get_rdata(char **ptr, int rdlen)
+char *get_rdata(char **ptr, char *end, int rdlen)
        {
-       char *rd = *ptr;
-       *ptr += rdlen;
-       return rd;
+       if (!*ptr || *ptr + rdlen > end)
+               {
+               *ptr = NULL;
+               return(0);
+               }
+       else
+               {
+               char *rd = *ptr;
+               *ptr += rdlen;
+               return rd;
+               }
        }
 
 void ConvertHeaderBytes(ipc_msg_hdr *hdr)
        {
        hdr->version   = htonl(hdr->version);
        hdr->datalen   = htonl(hdr->datalen);
-       hdr->flags     = htonl(hdr->flags);
+       hdr->ipc_flags = htonl(hdr->ipc_flags);
        hdr->op        = htonl(hdr->op );
        hdr->reg_index = htonl(hdr->reg_index);
        }