src/pipe.go

   1 /*
   2 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
   3 Copyright (C) 2016-2022 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, version 3 of the License.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 package nncp
  19
  20 import (
  21         "os"
  22         "os/exec"
  23         "time"
  24 )
  25
  26 type PipeConn struct {
  27         cmd *exec.Cmd
  28         r   *os.File
  29         w   *os.File
  30 }
  31
  32 func NewPipeConn(command string) (ConnDeadlined, error) {
  33         cmd := exec.Command("/bin/sh", "-c", command)
  34         stdinR, stdinW, err := os.Pipe()
  35         if err != nil {
  36                 return nil, err
  37         }
  38         cmd.Stdin = stdinR
  39         stdoutR, stdoutW, err := os.Pipe()
  40         if err != nil {
  41                 return nil, err
  42         }
  43         cmd.Stdout = stdoutW
  44         err = cmd.Start()
  45         if err != nil {
  46                 return nil, err
  47         }
  48         return &PipeConn{cmd, stdoutR, stdinW}, nil
  49 }
  50
  51 func (c PipeConn) Read(p []byte) (n int, err error) {
  52         return c.r.Read(p)
  53 }
  54
  55 func (c PipeConn) Write(p []byte) (n int, err error) {
  56         return c.w.Write(p)
  57 }
  58
  59 func (c PipeConn) SetReadDeadline(t time.Time) error {
  60         return c.r.SetReadDeadline(t)
  61 }
  62
  63 func (c PipeConn) SetWriteDeadline(t time.Time) error {
  64         return c.w.SetWriteDeadline(t)
  65 }
  66
  67 func (c PipeConn) Close() (err error) {
  68         err = c.r.Close()
  69         err = c.w.Close()
  70         go c.cmd.Wait()
  71         time.AfterFunc(time.Duration(10*time.Second), func() {
  72                 c.cmd.Process.Kill()
  73         })
  74         return
  75 }