src/pipe.go

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