Package list golang-github-farsightsec-go-nmsg / 4bbb937e-4955-45e6-9c4e-0d9331b45441/main cgo-nmsg / callbacks.go

Tree @4bbb937e-4955-45e6-9c4e-0d9331b45441/main (Download .tar.gz)

callbacks.go @4bbb937e-4955-45e6-9c4e-0d9331b45441/mainraw · history · blame

 * Copyright (c) 2017 by Farsight Security, Inc.
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at

package nmsg

#cgo pkg-config: libnmsg
#cgo LDFLAGS: -lnmsg
#include <nmsg.h>
#include <stdlib.h>
import "C"
import (

type outCbEntry struct {
	index int

type inCbEntry struct {
	index int

var cbm sync.Mutex
var outCbTable []Output
var inCbTable []Input

// The C library may not hold a pointer to a Go variable, but we
// need to store enough context in the callback user data to find
// the go object which registered the callback. We solve this by
// allocating memory on the C side (with C.malloc, C.calloc) and
// storing a value in that memory which we can use to look up the
// Go value on the Go side.
// The approach we take here is to have a package-global list of
// Output and Input, and store the index in the list as a
// in C-allocated memory. The location of this memory is returned
// as an unsafe.Pointer suitable for passing to the (void *user)
// argument of libnmsg callback registration functions.

func registerOutput(o Output) unsafe.Pointer {
	defer cbm.Unlock()
	idx := len(outCbTable)
	outCbTable = append(outCbTable, o)
	idxptr := C.calloc(C.size_t(1), C.size_t(unsafe.Sizeof(
	*(* =
	return idxptr

func registerInput(i Input) unsafe.Pointer {
	defer cbm.Unlock()
	idx := len(inCbTable)
	inCbTable = append(inCbTable, i)
	idxptr := C.calloc(C.size_t(1), C.size_t(unsafe.Sizeof(
	*(* =
	return idxptr

//export outputCallback
func outputCallback(msg C.nmsg_message_t, user unsafe.Pointer) {
	idx := int(*(*
	if idx < len(outCbTable) {
		o := outCbTable[idx]
	panic(fmt.Sprintf("outputCallback: invalid index %d", idx))

//export inputCallback
func inputCallback(msg, user unsafe.Pointer) C.nmsg_res {
	idx := int(*(*
	if idx < len(inCbTable) {
		i := inCbTable[idx]
		for {
			m, err := i.Read()

			if ErrorRetry(err) {
			if err != nil {
				*(*C.nmsg_message_t)(msg) = nil
				if e, ok := err.(nmsgResError); ok {
					return C.nmsg_res(e)
				return C.nmsg_res_failure
			runtime.SetFinalizer(m, nil)
			*(*C.nmsg_message_t)(msg) = m.message
			return C.nmsg_res_success
	panic(fmt.Sprintf("inputCallback: invalid index %d", idx))