src/ucspi.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         "fmt"
  20         "os"
  21         "time"
  22 )
  23
  24 const UCSPITCPClient = "UCSPI-TCP-CLIENT"
  25
  26 type UCSPIConn struct {
  27         R *os.File
  28         W *os.File
  29 }
  30
  31 func (c UCSPIConn) Read(p []byte) (n int, err error) {
  32         return c.R.Read(p)
  33 }
  34
  35 func (c UCSPIConn) Write(p []byte) (n int, err error) {
  36         return c.W.Write(p)
  37 }
  38
  39 func (c UCSPIConn) SetReadDeadline(t time.Time) error {
  40         return c.R.SetReadDeadline(t)
  41 }
  42
  43 func (c UCSPIConn) SetWriteDeadline(t time.Time) error {
  44         return c.W.SetWriteDeadline(t)
  45 }
  46
  47 func (c UCSPIConn) Close() error {
  48         if err := c.R.Close(); err != nil {
  49                 c.W.Close()
  50                 return err
  51         }
  52         return c.W.Close()
  53 }
  54
  55 func UCSPITCPRemoteAddr() (addr string) {
  56         if proto := os.Getenv("PROTO"); proto == "TCP" {
  57                 port := os.Getenv("TCPREMOTEPORT")
  58                 if host := os.Getenv("TCPREMOTEHOST"); host == "" {
  59                         addr = fmt.Sprintf("[%s]:%s", os.Getenv("TCPREMOTEIP"), port)
  60                 } else {
  61                         addr = fmt.Sprintf("%s:%s", host, port)
  62                 }
  63         }
  64         return
  65 }