Add files to generate Go bindings using the c-for-go tool

Signed-off-by: Kevin Klues <[email protected]>

Author: klueska

Makefile

@@ -0,0 +1,34 @@
+GEN_DIR = $(PWD)/gen
+PKG_DIR = $(PWD)/pkg
+GEN_BINDINGS_DIR = $(GEN_DIR)/nvml
+PKG_BINDINGS_DIR = $(PKG_DIR)/nvml
+
+SOURCES = $(shell find $(GEN_BINDINGS_DIR) -type f)
+
+.PHONY: all test clean
+.PHONY: bindings test-bindings clean-bindings
+
+all: bindings
+test: test-bindings
+clean: clean-bindings
+
+$(PKG_BINDINGS_DIR): $(SOURCES)
+	c-for-go -out $(PKG_DIR) $(GEN_BINDINGS_DIR)/nvml.yml
+	cp $(GEN_BINDINGS_DIR)/*.go $(PKG_BINDINGS_DIR)
+	cp $(GEN_BINDINGS_DIR)/nvml.h $(PKG_BINDINGS_DIR)
+	patch $(PKG_BINDINGS_DIR)/nvml.h $(GEN_BINDINGS_DIR)/nvml.h.patch
+	cd $(PKG_BINDINGS_DIR); \
+		go tool cgo -godefs types.go > types_gen.go; \
+		go fmt types_gen.go; \
+	cd -> /dev/null
+	rm -rf $(PKG_BINDINGS_DIR)/types.go $(PKG_BINDINGS_DIR)/_obj
+
+bindings: $(PKG_BINDINGS_DIR)
+
+test-bindings: $(PKG_BINDINGS_DIR)
+	cd $(PKG_BINDINGS_DIR); \
+		go test -v .; \
+	cd -> /dev/null
+
+clean-bindings:
+	rm -rf $(PKG_BINDINGS_DIR)

gen/nvml/cgo_helpers.go

@@ -0,0 +1,35 @@
+// Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
+
+package nvml
+
+import (
+	"unsafe"
+)
+
+import "C"
+
+var cgoAllocsUnknown = new(struct{})
+
+type stringHeader struct {
+	Data unsafe.Pointer
+	Len  int
+}
+
+// packPCharString creates a Go string backed by *C.char and avoids copying.
+func packPCharString(p *C.char) (raw string) {
+	if p != nil && *p != 0 {
+		h := (*stringHeader)(unsafe.Pointer(&raw))
+		h.Data = unsafe.Pointer(p)
+		for *p != 0 {
+			p = (*C.char)(unsafe.Pointer(uintptr(unsafe.Pointer(p)) + 1)) // p++
+		}
+		h.Len = int(uintptr(unsafe.Pointer(p)) - uintptr(h.Data))
+	}
+	return
+}
+
+// unpackPCharString represents the data from Go string as *C.char and avoids copying.
+func unpackPCharString(str string) (*C.char, *struct{}) {
+	h := (*stringHeader)(unsafe.Pointer(&str))
+	return (*C.char)(h.Data), cgoAllocsUnknown
+}