bgpd: optimize aspath string representation and assegments handling
* bgp_aspath.h: Add str_len to struct aspath.
* bgp_aspath.c: Save the aspath string representation length and use it
instead of strlen().
(aspath_make_str_count) assign the string buffer directly for
consistency with the string length and change the return type to void.
(aspath_dup) use str_len and copy the string instead of calling
aspath_make_str_count().
(assegment_data_new) change from XCALLOC to XMALLOC. All users initialize
the memory before use.
(assegment_data_free) unused, removed.
(aspath_intern) check that there's always a ->str pointer.
(aspath_hash_alloc) reuse assegments and string representation instead of
copying them.
(aspath_parse) now aspath_hash_alloc does not dupes memory, free the
temporary structures only if the aspath it is in the hash.
(aspath_cmp_left) remove useless NULL initialization.
Signed-off-by: Jorge Boncompte [DTI2] <jorge@dti2.net>
Signed-off-by: David Lamparter <equinox@opensourcerouting.org>
diff --git a/bgpd/bgp_aspath.c b/bgpd/bgp_aspath.c
index 64b3677..c37a889 100644
--- a/bgpd/bgp_aspath.c
+++ b/bgpd/bgp_aspath.c
@@ -91,16 +91,11 @@
/* Stream for SNMP. See aspath_snmp_pathseg */
static struct stream *snmp_stream;
+/* Callers are required to initialize the memory */
static as_t *
assegment_data_new (int num)
{
- return (XCALLOC (MTYPE_AS_SEG_DATA, ASSEGMENT_DATA_SIZE (num, 1)));
-}
-
-static void
-assegment_data_free (as_t *asdata)
-{
- XFREE (MTYPE_AS_SEG_DATA,asdata);
+ return (XMALLOC (MTYPE_AS_SEG_DATA, ASSEGMENT_DATA_SIZE (num, 1)));
}
/* Get a new segment. Note that 0 is an allowed length,
@@ -191,6 +186,7 @@
assegment_prepend_asns (struct assegment *seg, as_t asnum, int num)
{
as_t *newas;
+ int i;
if (!num)
return seg;
@@ -199,22 +195,16 @@
return seg; /* we don't do huge prepends */
newas = assegment_data_new (seg->length + num);
-
- if (newas)
- {
- int i;
- for (i = 0; i < num; i++)
- newas[i] = asnum;
-
- memcpy (newas + num, seg->as, ASSEGMENT_DATA_SIZE (seg->length, 1));
- XFREE (MTYPE_AS_SEG_DATA, seg->as);
- seg->as = newas;
- seg->length += num;
- return seg;
- }
- assegment_free_all (seg);
- return NULL;
+ for (i = 0; i < num; i++)
+ newas[i] = asnum;
+
+ memcpy (newas + num, seg->as, ASSEGMENT_DATA_SIZE (seg->length, 1));
+ XFREE (MTYPE_AS_SEG_DATA, seg->as);
+ seg->as = newas;
+ seg->length += num;
+
+ return seg;
}
/* append given array of as numbers to the segment */
@@ -504,7 +494,7 @@
}
/* Convert aspath structure to string expression. */
-static char *
+static void
aspath_make_str_count (struct aspath *as)
{
struct assegment *seg;
@@ -515,9 +505,10 @@
/* Empty aspath. */
if (!as->segments)
{
- str_buf = XMALLOC (MTYPE_AS_STR, 1);
- str_buf[0] = '\0';
- return str_buf;
+ as->str = XMALLOC (MTYPE_AS_STR, 1);
+ as->str[0] = '\0';
+ as->str_len = 0;
+ return;
}
seg = as->segments;
@@ -554,7 +545,9 @@
break;
default:
XFREE (MTYPE_AS_STR, str_buf);
- return NULL;
+ as->str = NULL;
+ as->str_len = 0;
+ return;
}
/* We might need to increase str_buf, particularly if path has
@@ -601,8 +594,10 @@
assert (len < str_size);
str_buf[len] = '\0';
+ as->str = str_buf;
+ as->str_len = len;
- return str_buf;
+ return;
}
static void
@@ -610,7 +605,7 @@
{
if (as->str)
XFREE (MTYPE_AS_STR, as->str);
- as->str = aspath_make_str_count (as);
+ aspath_make_str_count (as);
}
/* Intern allocated AS path. */
@@ -618,21 +613,19 @@
aspath_intern (struct aspath *aspath)
{
struct aspath *find;
-
- /* Assert this AS path structure is not interned. */
+
+ /* Assert this AS path structure is not interned and has the string
+ representation built. */
assert (aspath->refcnt == 0);
+ assert (aspath->str);
/* Check AS path hash. */
find = hash_get (ashash, aspath, hash_alloc_intern);
-
if (find != aspath)
aspath_free (aspath);
find->refcnt++;
- if (! find->str)
- find->str = aspath_make_str_count (find);
-
return find;
}
@@ -641,16 +634,25 @@
struct aspath *
aspath_dup (struct aspath *aspath)
{
+ unsigned short buflen = aspath->str_len + 1;
struct aspath *new;
new = XCALLOC (MTYPE_AS_PATH, sizeof (struct aspath));
if (aspath->segments)
new->segments = assegment_dup_all (aspath->segments);
- else
- new->segments = NULL;
- new->str = aspath_make_str_count (aspath);
+ if (!aspath->str)
+ return new;
+
+ new->str = XMALLOC (MTYPE_AS_STR, buflen);
+ new->str_len = aspath->str_len;
+
+ /* copy the string data */
+ if (aspath->str_len > 0)
+ memcpy (new->str, aspath->str, buflen);
+ else
+ new->str[0] = '\0';
return new;
}
@@ -658,19 +660,24 @@
static void *
aspath_hash_alloc (void *arg)
{
- struct aspath *aspath;
+ const struct aspath *aspath = arg;
+ struct aspath *new;
+
+ /* Malformed AS path value. */
+ assert (aspath->str);
+ if (! aspath->str)
+ return NULL;
/* New aspath structure is needed. */
- aspath = aspath_dup (arg);
-
- /* Malformed AS path value. */
- if (! aspath->str)
- {
- aspath_free (aspath);
- return NULL;
- }
+ new = XMALLOC (MTYPE_AS_PATH, sizeof (struct aspath));
- return aspath;
+ /* Reuse segments and string representation */
+ new->refcnt = 0;
+ new->segments = aspath->segments;
+ new->str = aspath->str;
+ new->str_len = aspath->str_len;
+
+ return new;
}
/* parse as-segment byte stream in struct assegment */
@@ -794,19 +801,21 @@
memset (&as, 0, sizeof (struct aspath));
if (assegments_parse (s, length, &as.segments, use32bit) < 0)
return NULL;
-
+
/* If already same aspath exist then return it. */
find = hash_get (ashash, &as, aspath_hash_alloc);
-
- /* aspath_hash_alloc dupes segments too. that probably could be
- * optimised out.
- */
- assegment_free_all (as.segments);
- if (as.str)
- XFREE (MTYPE_AS_STR, as.str);
-
- if (! find)
- return NULL;
+
+ /* bug! should not happen, let the daemon crash below */
+ assert (find);
+
+ /* if the aspath was already hashed free temporary memory. */
+ if (find->refcnt)
+ {
+ assegment_free_all (as.segments);
+ /* aspath_key_make() always updates the string */
+ XFREE (MTYPE_AS_STR, as.str);
+ }
+
find->refcnt++;
return find;
@@ -1400,8 +1409,8 @@
int
aspath_cmp_left (const struct aspath *aspath1, const struct aspath *aspath2)
{
- const struct assegment *seg1 = NULL;
- const struct assegment *seg2 = NULL;
+ const struct assegment *seg1;
+ const struct assegment *seg2;
if (!(aspath1 && aspath2))
return 0;
@@ -1639,7 +1648,7 @@
struct aspath *aspath;
aspath = aspath_new ();
- aspath->str = aspath_make_str_count (aspath);
+ aspath_make_str_count (aspath);
return aspath;
}
@@ -1798,7 +1807,7 @@
}
}
- aspath->str = aspath_make_str_count (aspath);
+ aspath_make_str_count (aspath);
return aspath;
}
@@ -1807,13 +1816,13 @@
unsigned int
aspath_key_make (void *p)
{
- struct aspath * aspath = (struct aspath *) p;
+ struct aspath *aspath = (struct aspath *) p;
unsigned int key = 0;
if (!aspath->str)
aspath_str_update (aspath);
-
- key = jhash (aspath->str, strlen(aspath->str), 2334325);
+
+ key = jhash (aspath->str, aspath->str_len, 2334325);
return key;
}
@@ -1876,7 +1885,7 @@
{
assert (format);
vty_out (vty, format, as->str);
- if (strlen (as->str) && strlen (suffix))
+ if (as->str_len && strlen (suffix))
vty_out (vty, "%s", suffix);
}