basic.go 2.53 KB
Newer Older
Russ Cox's avatar
Russ Cox committed
1 2 3 4
// Copyright 2010 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.

5
// Basic test cases for cgo.
Russ Cox's avatar
Russ Cox committed
6

7
package cgotest
Russ Cox's avatar
Russ Cox committed
8 9 10 11 12 13

/*
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <errno.h>
14
#include <unistd.h>
Russ Cox's avatar
Russ Cox committed
15 16 17 18

#define SHIFT(x, y)  ((x)<<(y))
#define KILO SHIFT(1, 10)

19
enum E {
Russ Cox's avatar
Russ Cox committed
20 21 22
	Enum1 = 1,
	Enum2 = 2,
};
Russ Cox's avatar
Russ Cox committed
23 24 25 26 27 28 29

typedef unsigned char uuid_t[20];

void uuid_generate(uuid_t x) {
	x[0] = 0;
}

30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
struct S {
	int x;
};

extern enum E myConstFunc(struct S* const ctx, int const id, struct S **const filter);

enum E myConstFunc(struct S *const ctx, int const id, struct S **const filter) { return 0; }

// issue 1222
typedef union {
	long align;
} xxpthread_mutex_t;

struct ibv_async_event {
	union {
		int x;
	} element;
};

struct ibv_context {
	xxpthread_mutex_t mutex;
};
Alex Brainman's avatar
Alex Brainman committed
52 53 54 55

int add(int x, int y) {
	return x+y;
};
Russ Cox's avatar
Russ Cox committed
56 57 58
*/
import "C"
import (
Russ Cox's avatar
Russ Cox committed
59
	"syscall"
60
	"testing"
61
	"time"
Russ Cox's avatar
Russ Cox committed
62 63 64 65 66 67 68
	"unsafe"
)

const EINVAL = C.EINVAL /* test #define */

var KILO = C.KILO

Russ Cox's avatar
Russ Cox committed
69 70 71 72 73
func uuidgen() {
	var uuid C.uuid_t
	C.uuid_generate(&uuid[0])
}

74
func Strtol(s string, base int) (int, error) {
Russ Cox's avatar
Russ Cox committed
75 76 77 78 79 80 81 82 83 84 85 86 87
	p := C.CString(s)
	n, err := C.strtol(p, nil, C.int(base))
	C.free(unsafe.Pointer(p))
	return int(n), err
}

func Atol(s string) int {
	p := C.CString(s)
	n := C.atol(p)
	C.free(unsafe.Pointer(p))
	return int(n)
}

Russ Cox's avatar
Russ Cox committed
88
func testConst(t *testing.T) {
89 90 91
	C.myConstFunc(nil, 0, nil)
}

Russ Cox's avatar
Russ Cox committed
92
func testEnum(t *testing.T) {
Russ Cox's avatar
Russ Cox committed
93
	if C.Enum1 != 1 || C.Enum2 != 2 {
94
		t.Error("bad enum", C.Enum1, C.Enum2)
Russ Cox's avatar
Russ Cox committed
95 96 97
	}
}

Russ Cox's avatar
Russ Cox committed
98
func testAtol(t *testing.T) {
Russ Cox's avatar
Russ Cox committed
99 100
	l := Atol("123")
	if l != 123 {
101
		t.Error("Atol 123: ", l)
Russ Cox's avatar
Russ Cox committed
102 103 104
	}
}

Russ Cox's avatar
Russ Cox committed
105
func testErrno(t *testing.T) {
106 107 108 109 110 111 112 113 114
	p := C.CString("no-such-file")
	m := C.CString("r")
	f, err := C.fopen(p, m)
	C.free(unsafe.Pointer(p))
	C.free(unsafe.Pointer(m))
	if err == nil {
		C.fclose(f)
		t.Fatalf("C.fopen: should fail")
	}
Russ Cox's avatar
Russ Cox committed
115
	if err != syscall.ENOENT {
Rob Pike's avatar
Rob Pike committed
116
		t.Fatalf("C.fopen: unexpected error: %v", err)
Russ Cox's avatar
Russ Cox committed
117 118 119
	}
}

Russ Cox's avatar
Russ Cox committed
120
func testMultipleAssign(t *testing.T) {
121
	p := C.CString("234")
Russ Cox's avatar
Russ Cox committed
122 123
	n, m := C.strtol(p, nil, 345), C.strtol(p, nil, 10)
	if n != 0 || m != 234 {
124
		t.Fatal("Strtol x2: ", n, m)
Russ Cox's avatar
Russ Cox committed
125 126 127 128
	}
	C.free(unsafe.Pointer(p))
}

129 130 131 132 133 134 135 136 137 138 139 140 141 142
func testSetgid(t *testing.T) {
	// Issue 3871.
	c := make(chan bool)
	go func() {
		C.setgid(0)
		c <- true
	}()
	select {
	case <-c:
	case <-time.After(5 * time.Second):
		t.Error("setgid hung")
	}
}

Russ Cox's avatar
Russ Cox committed
143
var (
144 145 146
	cuint  = (C.uint)(0)
	culong C.ulong
	cchar  C.char
Russ Cox's avatar
Russ Cox committed
147 148
)

149 150 151
type Context struct {
	ctx *C.struct_ibv_context
}
Alex Brainman's avatar
Alex Brainman committed
152 153 154 155 156 157 158 159

func benchCgoCall(b *testing.B) {
	const x = C.int(2)
	const y = C.int(3)
	for i := 0; i < b.N; i++ {
		C.add(x, y)
	}
}