aboutsummaryrefslogtreecommitdiff
path: root/libgo/runtime/go-setenv.c
blob: a62796b3de2fbc979dba42544f0a1fef332fa2a8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
/* go-setenv.c -- set the C environment from Go.

   Copyright 2011 The Go Authors. All rights reserved.
   Use of this source code is governed by a BSD-style
   license that can be found in the LICENSE file.  */

#include "config.h"

#include <stddef.h>
#include <stdlib.h>

#include "go-alloc.h"
#include "go-string.h"

/* Set the C environment from Go.  This is called by syscall.Setenv.  */

void setenv_c (struct __go_string, struct __go_string)
  __asm__ ("libgo_syscall.syscall.setenv_c");

void
setenv_c (struct __go_string k, struct __go_string v)
{
  const unsigned char *ks;
  unsigned char *kn;
  const unsigned char *vs;
  unsigned char *vn;

  ks = k.__data;
  kn = NULL;
  vs = v.__data;
  vn = NULL;

#ifdef HAVE_SETENV

  if (ks[k.__length] != 0)
    {
      kn = __go_alloc (k.__length + 1);
      __builtin_memcpy (kn, ks, k.__length);
      ks = kn;
    }

  if (vs[v.__length] != 0)
    {
      vn = __go_alloc (v.__length + 1);
      __builtin_memcpy (vn, vs, v.__length);
      vs = vn;
    }

  setenv ((const char *) ks, (const char *) vs, 1);

#else /* !defined(HAVE_SETENV) */

  kn = __go_alloc (k.__length + v.__length + 2);
  __builtin_memcpy (kn, ks, k.__length);
  kn[k.__length] = '=';
  __builtin_memcpy (kn + k.__length + 1, vs, v.__length);
  kn[k.__length + v.__length + 1] = '\0';
  putenv ((char *) kn);

#endif /* !defined(HAVE_SETENV) */

  if (kn != NULL)
    __go_free (kn);
  if (vn != NULL)
    __go_free (vn);
}