forgot to pull the publish_no_reverse change to the example.

[catta] / avahi-common / alternative.c

diff --git a/avahi-common/alternative.c b/avahi-common/alternative.c
index 77470baa405009105be92b0cb9a0db89b17f575f..b3d39f0ed90484d23c9fd8a41ce48ec83fefa1c2 100644 (file)
--- a/avahi-common/alternative.c
+++ b/avahi-common/alternative.c
@@ -1,18 +1,16 @@
-/* $Id$ */
-
 /***
   This file is part of avahi.
- 
+
   avahi is free software; you can redistribute it and/or modify it
   under the terms of the GNU Lesser General Public License as
   published by the Free Software Foundation; either version 2.1 of the
   License, or (at your option) any later version.
- 
+
   avahi is distributed in the hope that it will be useful, but WITHOUT
   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
   Public License for more details.
- 
+
   You should have received a copy of the GNU Lesser General Public
   License along with avahi; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
@@ -30,55 +28,105 @@
 
 #include "alternative.h"
 #include "malloc.h"
+#include "domain.h"
+#include "utf8.h"
+
+static void drop_incomplete_utf8(char *c) {
+    char *e;
+
+    e = strchr(c, 0) - 1;
+
+    while (e >= c) {
 
-char * avahi_alternative_host_name(const char *s) {
+        if (avahi_utf8_valid(c))
+            break;
+
+        assert(*e & 128);
+        *e = 0;
+
+        e--;
+    }
+}
+
+char *avahi_alternative_host_name(const char *s) {
     const char *e;
     char *r;
 
     assert(s);
 
-    e = strrchr(s, '-');
+    if (!avahi_is_valid_host_name(s))
+        return NULL;
 
-    if (e) {
+    if ((e = strrchr(s, '-'))) {
         const char *p;
-        
-        for (p = e+1; *p; p++)
+
+        e++;
+
+        for (p = e; *p; p++)
             if (!isdigit(*p)) {
                 e = NULL;
                 break;
             }
 
-        if (e && (*(e+1) == '0' || (*(e+1) == 0)))
+        if (e && (*e == '0' || *e == 0))
             e = NULL;
     }
 
     if (e) {
+        char *c, *m;
+        size_t l;
+        int n;
+
+        n = atoi(e)+1;
+        if (!(m = avahi_strdup_printf("%i", n)))
+            return NULL;
+
+        l = e-s-1;
+
+        if (l >= AVAHI_LABEL_MAX-1-strlen(m)-1)
+            l = AVAHI_LABEL_MAX-1-strlen(m)-1;
+
+        if (!(c = avahi_strndup(s, l))) {
+            avahi_free(m);
+            return NULL;
+        }
+
+        drop_incomplete_utf8(c);
+
+        r = avahi_strdup_printf("%s-%s", c, m);
+        avahi_free(c);
+        avahi_free(m);
+
+    } else {
         char *c;
 
-        e++;
-        
-        if (!(c = avahi_strndup(s, e-s)))
+        if (!(c = avahi_strndup(s, AVAHI_LABEL_MAX-1-2)))
             return NULL;
 
-        r = avahi_strdup_printf("%s%i", c, atoi(e)+1);
+        drop_incomplete_utf8(c);
+
+        r = avahi_strdup_printf("%s-2", c);
         avahi_free(c);
-        
-    } else
-        r = avahi_strdup_printf("%s-2", s);
-    
+    }
+
+    assert(avahi_is_valid_host_name(r));
+
     return r;
 }
 
 char *avahi_alternative_service_name(const char *s) {
     const char *e;
     char *r;
-    
+
     assert(s);
 
+    if (!avahi_is_valid_service_name(s))
+        return NULL;
+
     if ((e = strstr(s, " #"))) {
         const char *n, *p;
         e += 2;
-    
+
         while ((n = strstr(e, " #")))
             e = n + 2;
 
@@ -91,17 +139,44 @@ char *avahi_alternative_service_name(const char *s) {
         if (e && (*e == '0' || *e == 0))
             e = NULL;
     }
-    
+
     if (e) {
+        char *c, *m;
+        size_t l;
+        int n;
+
+        n = atoi(e)+1;
+        if (!(m = avahi_strdup_printf("%i", n)))
+            return NULL;
+
+        l = e-s-2;
+
+        if (l >= AVAHI_LABEL_MAX-1-strlen(m)-2)
+            l = AVAHI_LABEL_MAX-1-strlen(m)-2;
+
+        if (!(c = avahi_strndup(s, l))) {
+            avahi_free(m);
+            return NULL;
+        }
+
+        drop_incomplete_utf8(c);
+
+        r = avahi_strdup_printf("%s #%s", c, m);
+        avahi_free(c);
+        avahi_free(m);
+    } else {
         char *c;
 
-        if (!(c = avahi_strndup(s, e-s)))
+        if (!(c = avahi_strndup(s, AVAHI_LABEL_MAX-1-3)))
             return NULL;
-        
-        r = avahi_strdup_printf("%s%i", c, atoi(e)+1);
+
+        drop_incomplete_utf8(c);
+
+        r = avahi_strdup_printf("%s #2", c);
         avahi_free(c);
-    } else
-        r = avahi_strdup_printf("%s #2", s);
+    }
+
+    assert(avahi_is_valid_service_name(r));
 
     return r;
 }