forked from ssbc/go-secretstream
-
Notifications
You must be signed in to change notification settings - Fork 0
/
conn.go
81 lines (65 loc) · 2.09 KB
/
conn.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
/*
This file is part of secretstream.
secretstream is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
secretstream is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with secretstream. If not, see <http://www.gnu.org/licenses/>.
*/
package secretstream
import (
"io"
"net"
"time"
)
// Addr wrapps a net.Addr and adds the public key
type Addr struct {
net.Addr
pubKey []byte
}
// Network returns the network type of the net.Addr and appends /secret to it
// TODO(cryptix): the appended string might interfer with callers expecting "tcp"?
func (a Addr) Network() string {
return a.Addr.Network() + "/secret"
}
// PubKey returns the corrosponding public key for this connection
func (a Addr) PubKey() []byte {
return a.pubKey
}
// Conn is a boxstream wrapped net.Conn
type Conn struct {
io.Reader
io.Writer
conn net.Conn
// public keys
local, remote []byte
}
// Close closes the underlying net.Conn
func (c Conn) Close() error {
return c.conn.Close()
}
// LocalAddr returns the local net.Addr with the local public key
func (c Conn) LocalAddr() net.Addr {
return Addr{c.conn.LocalAddr(), c.local}
}
// RemoteAddr returns the remote net.Addr with the remote public key
func (c Conn) RemoteAddr() net.Addr {
return Addr{c.conn.RemoteAddr(), c.remote}
}
// SetDeadline passes the call to the underlying net.Conn
func (c Conn) SetDeadline(t time.Time) error {
return c.conn.SetDeadline(t)
}
// SetReadDeadline passes the call to the underlying net.Conn
func (c Conn) SetReadDeadline(t time.Time) error {
return c.conn.SetReadDeadline(t)
}
// SetWriteDeadline passes the call to the underlying net.Conn
func (c Conn) SetWriteDeadline(t time.Time) error {
return c.conn.SetWriteDeadline(t)
}