Codebase list golang-github-farsightsec-go-nmsg / fa6df4be-b667-498a-8af3-52e1bb06183d/main cgo-nmsg / io.go
fa6df4be-b667-498a-8af3-52e1bb06183d/main

Tree @fa6df4be-b667-498a-8af3-52e1bb06183d/main (Download .tar.gz)

io.go @fa6df4be-b667-498a-8af3-52e1bb06183d/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 http://mozilla.org/MPL/2.0/.
 */

package nmsg

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

// IO is a handle to a libnmsg io loop connecting one or more Inputs
// with one ore more Outputs.
type IO struct {
	nmsgIO C.nmsg_io_t
}

// NewIO creates and returns a new IO
func NewIO() *IO {
	io := C.nmsg_io_init()
	if io != nil {
		return &IO{io}
	}
	return nil
}

// AddInputChannel opens an NMSG channel and adds it as an Input to the
// IO.
func (io *IO) AddInputChannel(channel string) error {
	cchan := C.CString(channel)
	res := C.nmsg_io_add_input_channel(io.nmsgIO, cchan, nil)
	C.free(unsafe.Pointer(cchan))
	return nmsgError(res)
}

// AddInputSockSpec opens one or more sockets based on the sockspec
// (add/port ,or addr/lowport..highport) and adds it to the IO
// as an input.
func (io *IO) AddInputSockSpec(sockspec string) error {
	css := C.CString(sockspec)
	res := C.nmsg_io_add_input_sockspec(io.nmsgIO, css, nil)
	C.free(unsafe.Pointer(css))
	return nmsgError(res)
}

// AddInput adds a separately created Input to the IO as an input.
func (io *IO) AddInput(i Input) error {
	ni, ok := i.(*nmsgInput)
	if !ok {
		ni = NewCallbackInput(i.Read).(*nmsgInput)
	}
	return nmsgError(C.nmsg_io_add_input(io.nmsgIO, ni.input, nil))
}

// AddOutput adds a separately created Output to the IO as an output.
func (io *IO) AddOutput(o Output) error {
	nout, ok := o.(*nmsgOutput)
	if !ok {
		nout = NewCallbackOutput(o.Write).(*nmsgOutput)
	}
	return nmsgError(C.nmsg_io_add_output(io.nmsgIO, nout.output, nil))
}

// SetMirrored controls whether the IO mirrors output to all outputs
// (mirrored = true) or stripes its output across all outputs.
func (io *IO) SetMirrored(mirrored bool) {
	if mirrored {
		C.nmsg_io_set_output_mode(io.nmsgIO, C.nmsg_io_output_mode_mirror)
		return
	}
	C.nmsg_io_set_output_mode(io.nmsgIO, C.nmsg_io_output_mode_stripe)
}

// SetDebug sets the debug print level of the underlying io.
// Larger numbers are more verbose.
func (io *IO) SetDebug(debug int) {
	C.nmsg_io_set_debug(io.nmsgIO, C.int(debug))
}

// Run starts the IO loop, returning when it is finished or broken
// with Break()
func (io *IO) Run() error {
	return nmsgError(C.nmsg_io_loop(io.nmsgIO))
}

// Break stops the IO main loop.
func (io *IO) Break() {
	C.nmsg_io_breakloop(io.nmsgIO)
}