libgo/runtime/go-setenv.c

   1 /* go-setenv.c -- set the C environment from Go.
   2
   3    Copyright 2011 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 "config.h"
   8
   9 #include <stddef.h>
  10 #include <stdlib.h>
  11
  12 #include "go-alloc.h"
  13 #include "runtime.h"
  14
  15 /* Set the C environment from Go.  This is called by syscall.Setenv.  */
  16
  17 void setenv_c (String, String) __asm__ ("syscall.setenv_c");
  18
  19 void
  20 setenv_c (String k, String v)
  21 {
  22   const byte *ks;
  23   unsigned char *kn;
  24   const byte *vs;
  25   unsigned char *vn;
  26
  27   ks = k.str;
  28   kn = NULL;
  29   vs = v.str;
  30   vn = NULL;
  31
  32 #ifdef HAVE_SETENV
  33
  34   if (ks[k.len] != 0)
  35     {
  36       kn = __go_alloc (k.len + 1);
  37       __builtin_memcpy (kn, ks, k.len);
  38       ks = kn;
  39     }
  40
  41   if (vs[v.len] != 0)
  42     {
  43       vn = __go_alloc (v.len + 1);
  44       __builtin_memcpy (vn, vs, v.len);
  45       vs = vn;
  46     }
  47
  48   setenv ((const char *) ks, (const char *) vs, 1);
  49
  50 #else /* !defined(HAVE_SETENV) */
  51
  52   kn = __go_alloc (k.len + v.len + 2);
  53   __builtin_memcpy (kn, ks, k.len);
  54   kn[k.len] = '=';
  55   __builtin_memcpy (kn + k.len + 1, vs, v.len);
  56   kn[k.len + v.len + 1] = '\0';
  57   putenv ((char *) kn);
  58
  59 #endif /* !defined(HAVE_SETENV) */
  60
  61   if (kn != NULL)
  62     __go_free (kn);
  63   if (vn != NULL)
  64     __go_free (vn);
  65 }