libgo/runtime/go-fieldtrack.c

   1 /* go-fieldtrack.c -- structure field data analysis.
   2
   3    Copyright 2012 The Go Authors. All rights reserved.
   4    Use of this source code is governed by a BSD-style
   5    license that can be found in the LICENSE file.  */
   6
   7 #include "runtime.h"
   8 #include "go-type.h"
   9
  10 /* The compiler will track fields that have the tag go:"track".  Any
  11    function that refers to such a field will call this function with a
  12    string
  13        fieldtrack "package.type.field"
  14
  15    This function does not actually do anything.  Instead, we gather
  16    the field tracking information by looking for strings of that form
  17    in the read-only data section.  This is, of course, a horrible
  18    hack, but it's good enough for now.  We can improve it, e.g., by a
  19    linker plugin, if this turns out to be useful.  */
  20
  21 void
  22 __go_fieldtrack (byte *p __attribute__ ((unused)))
  23 {
  24 }
  25
  26 /* A runtime function to add all the tracked fields to a
  27    map[string]bool.  */
  28
  29 extern const char _etext[] __attribute__ ((weak));
  30 extern const char _edata[] __attribute__ ((weak));
  31 #ifdef _AIX
  32 // Following symbols do not exist on AIX
  33 const char *__etext = NULL;
  34 const char *__data_start = NULL;
  35 const char *__edata = NULL;
  36 const char *__bss_start = NULL;
  37 #else
  38 extern const char __etext[] __attribute__ ((weak));
  39 extern const char __data_start[] __attribute__ ((weak));
  40 extern const char __edata[] __attribute__ ((weak));
  41 extern const char __bss_start[] __attribute__ ((weak));
  42 #endif
  43
  44 extern void *mapassign (const struct __go_map_type *, void *hmap,
  45                         const void *key)
  46   __asm__ (GOSYM_PREFIX "runtime.mapassign");
  47
  48 // The type descriptor for map[string] bool.  */
  49 extern const char map_string_bool[] __attribute__ ((weak));
  50 extern const char map_string_bool[]
  51   __asm__ (GOSYM_PREFIX "type..map.6string.7bool");
  52
  53 void runtime_Fieldtrack (void *) __asm__ (GOSYM_PREFIX "runtime.Fieldtrack");
  54
  55 void
  56 runtime_Fieldtrack (void *m)
  57 {
  58   const char *p;
  59   const char *pend;
  60   const char *prefix;
  61   size_t prefix_len;
  62
  63   if (map_string_bool == NULL)
  64     return;
  65
  66   p = __data_start;
  67   if (p == NULL)
  68     p = __etext;
  69   if (p == NULL)
  70     p = _etext;
  71   if (p == NULL)
  72     return;
  73
  74   pend = __edata;
  75   if (pend == NULL)
  76     pend = _edata;
  77   if (pend == NULL)
  78     pend = __bss_start;
  79   if (pend == NULL)
  80     return;
  81
  82   prefix = "fieldtrack ";
  83   prefix_len = __builtin_strlen (prefix);
  84
  85   while (p < pend)
  86     {
  87       const char *q1;
  88       const char *q2;
  89
  90       q1 = __builtin_memchr (p + prefix_len, '"', pend - (p + prefix_len));
  91       if (q1 == NULL)
  92         break;
  93
  94       if (__builtin_memcmp (q1 - prefix_len, prefix, prefix_len) != 0)
  95         {
  96           p = q1 + 1;
  97           continue;
  98         }
  99
 100       q1++;
 101       q2 = __builtin_memchr (q1, '"', pend - q1);
 102       if (q2 == NULL)
 103         break;
 104
 105       if (__builtin_memchr (q1, '\0', q2 - q1) == NULL)
 106         {
 107           String s;
 108           void *p;
 109
 110           s.str = (const byte *) q1;
 111           s.len = q2 - q1;
 112           p = mapassign((const void*) map_string_bool, m, &s);
 113           *(_Bool*)p = 1;
 114         }
 115
 116       p = q2;
 117     }
 118 }