From f52da1cac583c31a7f0ca5530466dce2bede598e Mon Sep 17 00:00:00 2001 From: Luke Shumaker Date: Tue, 25 Aug 2015 14:19:33 -0600 Subject: initial commit --- sd-daemon/listen_fds.go | 51 ++++++++++++++++++++++++++++++++++++++++++++++ sd-daemon/logger/logger.go | 20 ++++++++++++++++++ 2 files changed, 71 insertions(+) create mode 100644 sd-daemon/listen_fds.go create mode 100644 sd-daemon/logger/logger.go diff --git a/sd-daemon/listen_fds.go b/sd-daemon/listen_fds.go new file mode 100644 index 0000000..c90d373 --- /dev/null +++ b/sd-daemon/listen_fds.go @@ -0,0 +1,51 @@ +// Copyright 2015 CoreOS, Inc. +// Copyright 2015 Luke Shumaker +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +// Package activation implements primitives for systemd socket activation. +package listen + +import ( + "os" + "strconv" + "syscall" +) + +//#include +import "C" + +func ListenFds(unsetEnv bool) []*os.File { + if unsetEnv { + defer os.Unsetenv("LISTEN_PID") + defer os.Unsetenv("LISTEN_FDS") + } + + pid, err := strconv.Atoi(os.Getenv("LISTEN_PID")) + if err != nil || pid != os.Getpid() { + return nil + } + + nfds, err := strconv.Atoi(os.Getenv("LISTEN_FDS")) + if err != nil || nfds == 0 { + return nil + } + + files := make([]*os.File, 0, nfds) + for fd := C.SD_LISTEN_FDS_START; fd < C.SD_LISTEN_FDS_START+nfds; fd++ { + syscall.CloseOnExec(fd) + files = append(files, os.NewFile(uintptr(fd), "LISTEN_FD_"+strconv.Itoa(fd))) + } + + return files +} diff --git a/sd-daemon/logger/logger.go b/sd-daemon/logger/logger.go new file mode 100644 index 0000000..cfabe9f --- /dev/null +++ b/sd-daemon/logger/logger.go @@ -0,0 +1,20 @@ +package logger + +import ("fmt"; "os") + +//#include +import "C" + +func log(level string, format string, a ...interface{}) { + f := level + format + "\n"; + fmt.Fprintf(os.Stderr, f, a); +} + +/* system is unusable */ func Emerg( format string, a ...interface{}) { log(C.SD_EMERG , format, a); } +/* action must be taken immediately */ func Alert( format string, a ...interface{}) { log(C.SD_ALERT , format, a); } +/* critical conditions */ func Crit( format string, a ...interface{}) { log(C.SD_CRIT , format, a); } +/* error conditions */ func Err( format string, a ...interface{}) { log(C.SD_ERR , format, a); } +/* warning conditions */ func Warning(format string, a ...interface{}) { log(C.SD_WARNING, format, a); } +/* normal but significant condition */ func Notice( format string, a ...interface{}) { log(C.SD_NOTICE , format, a); } +/* informational */ func Info( format string, a ...interface{}) { log(C.SD_INFO , format, a); } +/* debug-level messages */ func Debug( format string, a ...interface{}) { log(C.SD_DEBUG , format, a); } -- cgit v1.2.3