Compare commits
56 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 7c3069eabb | |||
| 3cc31323ad | |||
| 7f898dff32 | |||
| 58c1151278 | |||
| d1c549f1f2 | |||
| 9eb9cf0726 | |||
| f2abf8893c | |||
| 732622f4db | |||
| 95311e1479 | |||
| 084b629ad7 | |||
| 5713b58c7c | |||
| 56c6b7b352 | |||
| ea97afb01f | |||
| 482b0d3ad8 | |||
| 231bfeb247 | |||
| bd0425d6d4 | |||
| 5564eccf22 | |||
| d373e9791a | |||
| 7e249acd6c | |||
| b592e8ef7e | |||
| 11564b8c32 | |||
| 0a53963ec0 | |||
| b63a240e9b | |||
| 51b08dad3d | |||
| 1a3e4fe072 | |||
| 086281dab2 | |||
| 7392cbbea5 | |||
| 265c0a43ce | |||
| 417deff347 | |||
| a996f1668c | |||
| 9b290ebb96 | |||
| 756b347592 | |||
| d4e442bd84 | |||
| bbdc18698d | |||
| b049acad90 | |||
| df0211c67d | |||
| 861090b0e9 | |||
| cc15a571a8 | |||
| 3c4972d930 | |||
| 43db3a0d64 | |||
| 8e4a45a2b9 | |||
| f350afa0fe | |||
| dea4f996f9 | |||
| 0fee719b90 | |||
| 15455ff031 | |||
| a287b3adcc | |||
| d6d9a1711c | |||
| 1858dfa102 | |||
| 89b67f8559 | |||
| 649b3dfcab | |||
| ed64887ddc | |||
| 2113139cdc | |||
| 07441628d9 | |||
| 34dd9d515c | |||
| 32f64690c9 | |||
| 3f66cd93c7 |
9
.hgtags
9
.hgtags
@@ -5,3 +5,12 @@ d8b64d5527c2a5e4d76872e5bc3d69f7646135c6 libnmdc-r3
|
|||||||
fca41372e400853775b02e951f9db91d87f41adb nmdc-log-service-1.0.1
|
fca41372e400853775b02e951f9db91d87f41adb nmdc-log-service-1.0.1
|
||||||
050b424a7c5d5a27c9323c8810f3afbead1f5b96 libnmdc-r4
|
050b424a7c5d5a27c9323c8810f3afbead1f5b96 libnmdc-r4
|
||||||
da9f123633f9c28be6435ed7898139665d4c39d9 nmdc-log-service-1.0.2
|
da9f123633f9c28be6435ed7898139665d4c39d9 nmdc-log-service-1.0.2
|
||||||
|
75a78f6a78f249a2cd8aa3d29f7e5e6319b4e03b libnmdc-r5
|
||||||
|
4116422bb10229d887f9296970a166fa1ef8c5fd nmdc-log-service-1.0.3
|
||||||
|
cb86f3a40115cc46f450c0c83fd9b9d3b740e820 nmdc-log-service-1.0.4
|
||||||
|
cb86f3a40115cc46f450c0c83fd9b9d3b740e820 libnmdc-r6
|
||||||
|
71343a2c641a438206d30ea7e75dc89a11dbef00 libnmdc-r7
|
||||||
|
b0e57a5fcffdf4102d669db51a3648ddf66a0792 libnmdc-r8
|
||||||
|
e7c2c71ef24b386add728fad35fff4a996fccbac libnmdc-r9
|
||||||
|
3ecc037cf2d7080572fe87c2e39ecd153fb0e947 libnmdc-r10
|
||||||
|
5149ffe70ea8475e480b682345b31aa45a3352db release-0.11
|
||||||
|
|||||||
29
ConnectionMode.go
Normal file
29
ConnectionMode.go
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ConnectionMode rune
|
||||||
|
|
||||||
|
const (
|
||||||
|
CONNECTIONMODE_ACTIVE ConnectionMode = 'A' // 65
|
||||||
|
CONNECTIONMODE_PASSIVE ConnectionMode = 'P' // 49
|
||||||
|
CONNECTIONMODE_SOCKS5 ConnectionMode = '5' // 53
|
||||||
|
)
|
||||||
|
|
||||||
|
func (this ConnectionMode) String() string {
|
||||||
|
switch this {
|
||||||
|
case CONNECTIONMODE_ACTIVE:
|
||||||
|
return "Active"
|
||||||
|
|
||||||
|
case CONNECTIONMODE_PASSIVE:
|
||||||
|
return "Passive"
|
||||||
|
|
||||||
|
case CONNECTIONMODE_SOCKS5:
|
||||||
|
return "SOCKS5"
|
||||||
|
|
||||||
|
default:
|
||||||
|
return fmt.Sprintf("ConnectionMode(\"%s\")", string(this))
|
||||||
|
}
|
||||||
|
}
|
||||||
40
ConnectionState.go
Normal file
40
ConnectionState.go
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ConnectionState int
|
||||||
|
|
||||||
|
const (
|
||||||
|
CONNECTIONSTATE_DISCONNECTED = 1
|
||||||
|
CONNECTIONSTATE_CONNECTING = 2 // Handshake in progress
|
||||||
|
CONNECTIONSTATE_CONNECTED = 3
|
||||||
|
)
|
||||||
|
|
||||||
|
func (cs ConnectionState) String() string {
|
||||||
|
switch cs {
|
||||||
|
case CONNECTIONSTATE_DISCONNECTED:
|
||||||
|
return "Disconnected"
|
||||||
|
case CONNECTIONSTATE_CONNECTING:
|
||||||
|
return "Connecting"
|
||||||
|
case CONNECTIONSTATE_CONNECTED:
|
||||||
|
return "Connected"
|
||||||
|
default:
|
||||||
|
return "?"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func checkIsNetTimeout(err error) bool {
|
||||||
|
if err == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
switch err.(type) {
|
||||||
|
case net.Error:
|
||||||
|
return err.(net.Error).Timeout()
|
||||||
|
|
||||||
|
default:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
58
Example_test.go
Normal file
58
Example_test.go
Normal file
@@ -0,0 +1,58 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ExampleHubConnectionOptions_Connect() {
|
||||||
|
opts := HubConnectionOptions{
|
||||||
|
Address: "127.0.0.1",
|
||||||
|
Self: *NewUserInfo("slowpoke9"),
|
||||||
|
}
|
||||||
|
hub := opts.Connect()
|
||||||
|
|
||||||
|
for {
|
||||||
|
event := <-hub.OnEvent
|
||||||
|
switch event.EventType {
|
||||||
|
case EVENT_CONNECTION_STATE_CHANGED:
|
||||||
|
fmt.Printf("Connection -- %s (%s)\n", event.StateChange.Format(), event.Message)
|
||||||
|
|
||||||
|
case EVENT_PUBLIC:
|
||||||
|
fmt.Printf("Message from '%s': '%s'\n", event.Nick, event.Message)
|
||||||
|
if event.Message == "how are you" {
|
||||||
|
hub.SayPublic("good thanks!")
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
fmt.Printf("%+v\n", event)
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExampleHubConnectionOptions_ConnectSync() {
|
||||||
|
cb := func(hub *HubConnection, event HubEvent) {
|
||||||
|
switch event.EventType {
|
||||||
|
case EVENT_CONNECTION_STATE_CHANGED:
|
||||||
|
fmt.Printf("Connection -- %s (%s)\n", event.StateChange.Format(), event.Message)
|
||||||
|
|
||||||
|
case EVENT_PUBLIC:
|
||||||
|
fmt.Printf("Message from '%s': '%s'\n", event.Nick, event.Message)
|
||||||
|
if event.Message == "how are you" {
|
||||||
|
hub.SayPublic("good thanks!")
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
fmt.Printf("%+v\n", event)
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
opts := HubConnectionOptions{
|
||||||
|
Address: "127.0.0.1",
|
||||||
|
Self: *NewUserInfo("slowpoke9"),
|
||||||
|
OnEventSync: cb,
|
||||||
|
}
|
||||||
|
|
||||||
|
opts.ConnectSync() // blocking
|
||||||
|
}
|
||||||
35
HubAddress.go
Normal file
35
HubAddress.go
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type HubAddress string
|
||||||
|
|
||||||
|
func (this *HubAddress) parse() url.URL {
|
||||||
|
parsed, err := url.Parse(strings.ToLower(string(*this)))
|
||||||
|
if err != nil || len(parsed.Host) == 0 {
|
||||||
|
parsed = &url.URL{
|
||||||
|
Scheme: "nmdc",
|
||||||
|
Host: string(*this),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add default port if not specified
|
||||||
|
if !strings.ContainsRune(parsed.Host, ':') {
|
||||||
|
parsed.Host = parsed.Host + ":411"
|
||||||
|
}
|
||||||
|
|
||||||
|
return *parsed
|
||||||
|
}
|
||||||
|
|
||||||
|
func (this *HubAddress) IsSecure() bool {
|
||||||
|
parsed := this.parse()
|
||||||
|
|
||||||
|
return parsed.Scheme == "nmdcs" || parsed.Scheme == "dchubs"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (this *HubAddress) GetHostOnly() string {
|
||||||
|
return this.parse().Host
|
||||||
|
}
|
||||||
@@ -1,98 +1,14 @@
|
|||||||
// libnmdc project libnmdc.go
|
|
||||||
package libnmdc
|
package libnmdc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"net"
|
"net"
|
||||||
"net/url"
|
"strconv"
|
||||||
"regexp"
|
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
|
||||||
DEFAULT_CLIENT_TAG string = "libnmdc.go"
|
|
||||||
DEFAULT_HUB_NAME string = "(unknown)"
|
|
||||||
SEND_KEEPALIVE_EVERY time.Duration = 29 * time.Second
|
|
||||||
)
|
|
||||||
|
|
||||||
type ConnectionState int
|
|
||||||
type HubEventType int
|
|
||||||
|
|
||||||
type HubAddress string
|
|
||||||
|
|
||||||
func (this *HubAddress) parse() url.URL {
|
|
||||||
parsed, err := url.Parse(strings.ToLower(string(*this)))
|
|
||||||
if err != nil || len(parsed.Host) == 0 {
|
|
||||||
parsed = &url.URL{
|
|
||||||
Scheme: "nmdc",
|
|
||||||
Host: string(*this),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add default port if not specified
|
|
||||||
if !strings.ContainsRune(parsed.Host, ':') {
|
|
||||||
parsed.Host = parsed.Host + ":411"
|
|
||||||
}
|
|
||||||
|
|
||||||
return *parsed
|
|
||||||
}
|
|
||||||
|
|
||||||
func (this *HubAddress) IsSecure() bool {
|
|
||||||
parsed := this.parse()
|
|
||||||
|
|
||||||
return parsed.Scheme == "nmdcs" || parsed.Scheme == "dchubs"
|
|
||||||
}
|
|
||||||
|
|
||||||
func (this *HubAddress) GetHostOnly() string {
|
|
||||||
return this.parse().Host
|
|
||||||
}
|
|
||||||
|
|
||||||
const (
|
|
||||||
CONNECTIONSTATE_DISCONNECTED = 1
|
|
||||||
CONNECTIONSTATE_CONNECTING = 2 // Handshake in progress
|
|
||||||
CONNECTIONSTATE_CONNECTED = 3
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
EVENT_PUBLIC = 1
|
|
||||||
EVENT_PRIVATE = 2
|
|
||||||
EVENT_SYSTEM_MESSAGE_FROM_HUB = 3
|
|
||||||
EVENT_SYSTEM_MESSAGE_FROM_CONN = 4
|
|
||||||
EVENT_USER_JOINED = 5
|
|
||||||
EVENT_USER_PART = 6
|
|
||||||
EVENT_USER_UPDATED_INFO = 7
|
|
||||||
EVENT_CONNECTION_STATE_CHANGED = 8
|
|
||||||
EVENT_HUBNAME_CHANGED = 9
|
|
||||||
EVENT_DEBUG_MESSAGE = 10
|
|
||||||
)
|
|
||||||
|
|
||||||
var rx_protocolMessage *regexp.Regexp
|
|
||||||
var rx_publicChat *regexp.Regexp
|
|
||||||
var rx_incomingTo *regexp.Regexp
|
|
||||||
var ErrNotConnected error = errors.New("Not connected")
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
rx_protocolMessage = regexp.MustCompile("(?ms)^[^|]*\\|")
|
|
||||||
rx_publicChat = regexp.MustCompile("(?ms)^<([^>]*)> (.*)$")
|
|
||||||
rx_incomingTo = regexp.MustCompile("(?ms)^([^ ]+) From: ([^ ]+) \\$<([^>]*)> (.*)")
|
|
||||||
}
|
|
||||||
|
|
||||||
type HubConnectionOptions struct {
|
|
||||||
Address HubAddress
|
|
||||||
SkipVerifyTLS bool // using a negative verb, because bools default to false
|
|
||||||
Self UserInfo
|
|
||||||
NickPassword string
|
|
||||||
|
|
||||||
// Returning messages in async mode
|
|
||||||
NumEventsToBuffer uint
|
|
||||||
|
|
||||||
// Returning messages in sync mode
|
|
||||||
OnEventSync func(HubEvent)
|
|
||||||
}
|
|
||||||
|
|
||||||
type HubConnection struct {
|
type HubConnection struct {
|
||||||
// Supplied parameters
|
// Supplied parameters
|
||||||
Hco *HubConnectionOptions
|
Hco *HubConnectionOptions
|
||||||
@@ -100,7 +16,8 @@ type HubConnection struct {
|
|||||||
// Current remote status
|
// Current remote status
|
||||||
HubName string
|
HubName string
|
||||||
State ConnectionState
|
State ConnectionState
|
||||||
Users map[string]UserInfo
|
users map[string]UserInfo
|
||||||
|
userLock sync.RWMutex
|
||||||
|
|
||||||
// Streamed events
|
// Streamed events
|
||||||
processEvent func(HubEvent)
|
processEvent func(HubEvent)
|
||||||
@@ -110,68 +27,69 @@ type HubConnection struct {
|
|||||||
conn net.Conn // this is an interface
|
conn net.Conn // this is an interface
|
||||||
connValid bool
|
connValid bool
|
||||||
sentOurHello bool
|
sentOurHello bool
|
||||||
|
autoReconnect bool
|
||||||
}
|
}
|
||||||
|
|
||||||
type HubEvent struct {
|
// Thread-safe user accessor.
|
||||||
EventType HubEventType
|
func (this *HubConnection) Users(cb func(*map[string]UserInfo) error) error {
|
||||||
Nick string
|
this.userLock.Lock()
|
||||||
Message string
|
defer this.userLock.Unlock()
|
||||||
StateChange ConnectionState
|
|
||||||
}
|
|
||||||
|
|
||||||
func (cs ConnectionState) Format() string {
|
return cb(&this.users)
|
||||||
switch cs {
|
|
||||||
case CONNECTIONSTATE_DISCONNECTED:
|
|
||||||
return "Disconnected"
|
|
||||||
case CONNECTIONSTATE_CONNECTING:
|
|
||||||
return "Connecting"
|
|
||||||
case CONNECTIONSTATE_CONNECTED:
|
|
||||||
return "Connected"
|
|
||||||
default:
|
|
||||||
return "?"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func NMDCUnescape(encoded string) string {
|
|
||||||
v1 := strings.Replace(encoded, "$", "$", -1)
|
|
||||||
v2 := strings.Replace(v1, "|", "|", -1)
|
|
||||||
return strings.Replace(v2, "&", "&", -1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func NMDCEscape(plaintext string) string {
|
|
||||||
v1 := strings.Replace(plaintext, "&", "&", -1)
|
|
||||||
v2 := strings.Replace(v1, "|", "|", -1)
|
|
||||||
return strings.Replace(v2, "$", "$", -1)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (this *HubConnection) SayPublic(message string) {
|
func (this *HubConnection) SayPublic(message string) {
|
||||||
this.SayRaw("<" + this.Hco.Self.Nick + "> " + NMDCEscape(message) + "|")
|
this.SayRaw("<" + this.Hco.Self.Nick + "> " + Escape(message) + "|")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (this *HubConnection) SayPrivate(recipient string, message string) {
|
func (this *HubConnection) SayPrivate(recipient string, message string) {
|
||||||
this.SayRaw("$To: " + recipient + " From: " + this.Hco.Self.Nick + " $<" + this.Hco.Self.Nick + "> " + NMDCEscape(message) + "|")
|
this.SayRaw("$To: " + recipient + " From: " + this.Hco.Self.Nick + " $<" + this.Hco.Self.Nick + "> " + Escape(message) + "|")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (this *HubConnection) SayInfo() {
|
func (this *HubConnection) SayInfo() {
|
||||||
this.SayRaw(this.Hco.Self.toMyINFO() + "|")
|
this.SayRaw(this.Hco.Self.toMyINFO() + "|")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (this *HubConnection) UserExists(nick string) bool {
|
||||||
|
this.userLock.RLock()
|
||||||
|
defer this.userLock.RUnlock()
|
||||||
|
|
||||||
|
_, already_existed := this.users[nick]
|
||||||
|
return already_existed
|
||||||
|
}
|
||||||
|
|
||||||
|
func (this *HubConnection) UserCount() int {
|
||||||
|
this.userLock.RLock()
|
||||||
|
defer this.userLock.RUnlock()
|
||||||
|
|
||||||
|
return len(this.users)
|
||||||
|
}
|
||||||
|
|
||||||
func (this *HubConnection) userJoined_NameOnly(nick string) {
|
func (this *HubConnection) userJoined_NameOnly(nick string) {
|
||||||
_, already_existed := this.Users[nick]
|
if !this.UserExists(nick) {
|
||||||
if !already_existed {
|
|
||||||
this.Users[nick] = *NewUserInfo(nick)
|
this.userLock.Lock()
|
||||||
|
this.users[nick] = *NewUserInfo(nick)
|
||||||
|
this.userLock.Unlock() // Don't lock over a processEvent boundary
|
||||||
|
|
||||||
this.processEvent(HubEvent{EventType: EVENT_USER_JOINED, Nick: nick})
|
this.processEvent(HubEvent{EventType: EVENT_USER_JOINED, Nick: nick})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (this *HubConnection) userJoined_Full(uinf *UserInfo) {
|
func (this *HubConnection) userJoined_Full(uinf *UserInfo) {
|
||||||
_, already_existed := this.Users[uinf.Nick]
|
// n.b. also called when we get a replacement MyINFO for someone
|
||||||
if !already_existed {
|
this.userLock.Lock()
|
||||||
this.Users[uinf.Nick] = *uinf
|
_, userExisted := this.users[uinf.Nick] // don't use UserExists as it would deadlock the mutex
|
||||||
|
this.users[uinf.Nick] = *uinf
|
||||||
|
this.userLock.Unlock() // Don't lock over a processEvent boundary
|
||||||
|
|
||||||
|
if !userExisted {
|
||||||
this.processEvent(HubEvent{EventType: EVENT_USER_JOINED, Nick: uinf.Nick})
|
this.processEvent(HubEvent{EventType: EVENT_USER_JOINED, Nick: uinf.Nick})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// SayRaw sends raw bytes over the TCP socket. Callers should add the protocol
|
||||||
|
// terminating character `|` themselves.
|
||||||
// Note that protocol messages are transmitted on the caller thread, not from
|
// Note that protocol messages are transmitted on the caller thread, not from
|
||||||
// any internal libnmdc thread.
|
// any internal libnmdc thread.
|
||||||
func (this *HubConnection) SayRaw(protocolCommand string) error {
|
func (this *HubConnection) SayRaw(protocolCommand string) error {
|
||||||
@@ -183,28 +101,6 @@ func (this *HubConnection) SayRaw(protocolCommand string) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseLock(lock []byte) string {
|
|
||||||
|
|
||||||
nibble_swap := func(b byte) byte {
|
|
||||||
return ((b << 4) & 0xF0) | ((b >> 4) & 0x0F)
|
|
||||||
}
|
|
||||||
|
|
||||||
chr := func(b byte) string {
|
|
||||||
if b == 0 || b == 5 || b == 36 || b == 96 || b == 124 || b == 126 {
|
|
||||||
return fmt.Sprintf("/%%DCN%04d%%/", b)
|
|
||||||
} else {
|
|
||||||
return string(b)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
key := chr(nibble_swap(lock[0] ^ lock[len(lock)-2] ^ lock[len(lock)-3] ^ 5))
|
|
||||||
for i := 1; i < len(lock); i += 1 {
|
|
||||||
key += chr(nibble_swap(lock[i] ^ lock[i-1]))
|
|
||||||
}
|
|
||||||
|
|
||||||
return key
|
|
||||||
}
|
|
||||||
|
|
||||||
func (this *HubConnection) processProtocolMessage(message string) {
|
func (this *HubConnection) processProtocolMessage(message string) {
|
||||||
|
|
||||||
// Zero-length protocol message
|
// Zero-length protocol message
|
||||||
@@ -217,14 +113,14 @@ func (this *HubConnection) processProtocolMessage(message string) {
|
|||||||
// ```````````
|
// ```````````
|
||||||
if rx_publicChat.MatchString(message) {
|
if rx_publicChat.MatchString(message) {
|
||||||
pubchat_parts := rx_publicChat.FindStringSubmatch(message)
|
pubchat_parts := rx_publicChat.FindStringSubmatch(message)
|
||||||
this.processEvent(HubEvent{EventType: EVENT_PUBLIC, Nick: pubchat_parts[1], Message: NMDCUnescape(pubchat_parts[2])})
|
this.processEvent(HubEvent{EventType: EVENT_PUBLIC, Nick: pubchat_parts[1], Message: Unescape(pubchat_parts[2])})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// System messages
|
// System messages
|
||||||
// ```````````````
|
// ```````````````
|
||||||
if message[0] != '$' {
|
if message[0] != '$' {
|
||||||
this.processEvent(HubEvent{EventType: EVENT_SYSTEM_MESSAGE_FROM_HUB, Nick: this.HubName, Message: NMDCUnescape(message)})
|
this.processEvent(HubEvent{EventType: EVENT_SYSTEM_MESSAGE_FROM_HUB, Nick: this.HubName, Message: Unescape(message)})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -236,8 +132,8 @@ func (this *HubConnection) processProtocolMessage(message string) {
|
|||||||
|
|
||||||
case "$Lock":
|
case "$Lock":
|
||||||
this.SayRaw("$Supports NoGetINFO UserCommand UserIP2|" +
|
this.SayRaw("$Supports NoGetINFO UserCommand UserIP2|" +
|
||||||
"$Key " + parseLock([]byte(commandParts[1])) + "|" +
|
"$Key " + unlock([]byte(commandParts[1])) + "|" +
|
||||||
"$ValidateNick " + NMDCEscape(this.Hco.Self.Nick) + "|")
|
"$ValidateNick " + Escape(this.Hco.Self.Nick) + "|")
|
||||||
this.sentOurHello = false
|
this.sentOurHello = false
|
||||||
|
|
||||||
case "$Hello":
|
case "$Hello":
|
||||||
@@ -270,10 +166,20 @@ func (this *HubConnection) processProtocolMessage(message string) {
|
|||||||
this.processEvent(HubEvent{EventType: EVENT_SYSTEM_MESSAGE_FROM_CONN, Message: "Incorrect password."})
|
this.processEvent(HubEvent{EventType: EVENT_SYSTEM_MESSAGE_FROM_CONN, Message: "Incorrect password."})
|
||||||
|
|
||||||
case "$GetPass":
|
case "$GetPass":
|
||||||
this.SayRaw("$MyPass " + NMDCEscape(this.Hco.NickPassword) + "|")
|
if len(this.Hco.NickPassword) == 0 {
|
||||||
|
// We've got a problem. MyPass with no arguments is a syntax error with no message = instant close
|
||||||
|
// Just drop the connection
|
||||||
|
this.processEvent(HubEvent{EventType: EVENT_SYSTEM_MESSAGE_FROM_CONN, Message: "This account is passworded."})
|
||||||
|
this.Disconnect()
|
||||||
|
} else {
|
||||||
|
this.SayRaw("$MyPass " + Escape(this.Hco.NickPassword) + "|")
|
||||||
|
}
|
||||||
|
|
||||||
case "$Quit":
|
case "$Quit":
|
||||||
delete(this.Users, commandParts[1])
|
this.userLock.Lock()
|
||||||
|
delete(this.users, commandParts[1])
|
||||||
|
this.userLock.Unlock() // Don't lock over a processEvent boundary
|
||||||
|
|
||||||
this.processEvent(HubEvent{EventType: EVENT_USER_PART, Nick: commandParts[1]})
|
this.processEvent(HubEvent{EventType: EVENT_USER_PART, Nick: commandParts[1]})
|
||||||
|
|
||||||
case "$MyINFO":
|
case "$MyINFO":
|
||||||
@@ -293,12 +199,39 @@ func (this *HubConnection) processProtocolMessage(message string) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
case "$OpList":
|
||||||
|
oplist := strings.Split(commandParts[1], "$$")
|
||||||
|
opmap := map[string]struct{}{}
|
||||||
|
|
||||||
|
// Organise/sort the list, and ensure we're not meeting an operator for
|
||||||
|
// the first time
|
||||||
|
for _, nick := range oplist {
|
||||||
|
if len(nick) > 0 {
|
||||||
|
opmap[nick] = struct{}{}
|
||||||
|
this.userJoined_NameOnly(nick) // assert existence; noop otherwise
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mark all mentioned nicks as being operators, and all unmentioned nicks
|
||||||
|
// as being /not/ an operator. (second pass minimises RW mutex use)
|
||||||
|
func() {
|
||||||
|
this.userLock.Lock()
|
||||||
|
defer this.userLock.Unlock()
|
||||||
|
|
||||||
|
for nick, userinfo := range this.users {
|
||||||
|
_, isop := opmap[nick]
|
||||||
|
|
||||||
|
userinfo.IsOperator = isop
|
||||||
|
this.users[nick] = userinfo
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
case "$To:":
|
case "$To:":
|
||||||
valid := false
|
valid := false
|
||||||
if rx_incomingTo.MatchString(commandParts[1]) {
|
if rx_incomingTo.MatchString(commandParts[1]) {
|
||||||
txparts := rx_incomingTo.FindStringSubmatch(commandParts[1])
|
txparts := rx_incomingTo.FindStringSubmatch(commandParts[1])
|
||||||
if txparts[1] == this.Hco.Self.Nick && txparts[2] == txparts[3] {
|
if txparts[1] == this.Hco.Self.Nick && txparts[2] == txparts[3] {
|
||||||
this.processEvent(HubEvent{EventType: EVENT_PRIVATE, Nick: txparts[2], Message: txparts[4]})
|
this.processEvent(HubEvent{EventType: EVENT_PRIVATE, Nick: txparts[2], Message: Unescape(txparts[4])})
|
||||||
valid = true
|
valid = true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -319,11 +252,29 @@ func (this *HubConnection) processProtocolMessage(message string) {
|
|||||||
this.Hco.Address = HubAddress(commandParts[1])
|
this.Hco.Address = HubAddress(commandParts[1])
|
||||||
this.conn.Close() // we'll reconnect onto the new address
|
this.conn.Close() // we'll reconnect onto the new address
|
||||||
|
|
||||||
|
case "$UserCommand":
|
||||||
|
// $UserCommand 1 1 Group chat\New group chat$<%[mynick]> !groupchat_new||
|
||||||
|
if rx_userCommand.MatchString(commandParts[1]) {
|
||||||
|
usc := rx_userCommand.FindStringSubmatch(commandParts[1])
|
||||||
|
|
||||||
|
typeInt, _ := strconv.Atoi(usc[1])
|
||||||
|
contextInt, _ := strconv.Atoi(usc[2])
|
||||||
|
|
||||||
|
uscStruct := UserCommand{
|
||||||
|
Type: UserCommandType(typeInt),
|
||||||
|
Context: UserCommandContext(contextInt),
|
||||||
|
Message: usc[3],
|
||||||
|
Command: Unescape(usc[4]),
|
||||||
|
}
|
||||||
|
|
||||||
|
this.processEvent(HubEvent{EventType: EVENT_USERCOMMAND, UserCommand: &uscStruct})
|
||||||
|
|
||||||
|
} else {
|
||||||
|
this.processEvent(HubEvent{EventType: EVENT_DEBUG_MESSAGE, Message: "Malformed usercommand '" + commandParts[1] + "'"})
|
||||||
|
}
|
||||||
|
|
||||||
// IGNORABLE COMMANDS
|
// IGNORABLE COMMANDS
|
||||||
case "$Supports":
|
case "$Supports":
|
||||||
case "$UserCommand": // TODO $UserCommand 1 1 Group chat\New group chat$<%[mynick]> !groupchat_new||
|
|
||||||
case "$UserList":
|
|
||||||
case "$OpList":
|
|
||||||
case "$HubTopic":
|
case "$HubTopic":
|
||||||
case "$Search":
|
case "$Search":
|
||||||
case "$ConnectToMe":
|
case "$ConnectToMe":
|
||||||
@@ -334,6 +285,14 @@ func (this *HubConnection) processProtocolMessage(message string) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (this *HubConnection) Disconnect() {
|
||||||
|
this.autoReconnect = false
|
||||||
|
if this.conn != nil {
|
||||||
|
this.conn.Close()
|
||||||
|
}
|
||||||
|
// A CONNECTIONSTATE_DISCONNECTED message will be emitted by the worker.
|
||||||
|
}
|
||||||
|
|
||||||
func (this *HubConnection) worker() {
|
func (this *HubConnection) worker() {
|
||||||
var fullBuffer string
|
var fullBuffer string
|
||||||
var err error = nil
|
var err error = nil
|
||||||
@@ -372,7 +331,7 @@ func (this *HubConnection) worker() {
|
|||||||
this.conn.SetReadDeadline(time.Now().Add(SEND_KEEPALIVE_EVERY))
|
this.conn.SetReadDeadline(time.Now().Add(SEND_KEEPALIVE_EVERY))
|
||||||
nbytes, err = this.conn.Read(readBuff)
|
nbytes, err = this.conn.Read(readBuff)
|
||||||
|
|
||||||
if err != nil && err.(net.Error).Timeout() {
|
if checkIsNetTimeout(err) {
|
||||||
// No data before read deadline
|
// No data before read deadline
|
||||||
err = nil
|
err = nil
|
||||||
|
|
||||||
@@ -408,50 +367,14 @@ func (this *HubConnection) worker() {
|
|||||||
this.connValid = false
|
this.connValid = false
|
||||||
this.processEvent(HubEvent{EventType: EVENT_CONNECTION_STATE_CHANGED, StateChange: CONNECTIONSTATE_DISCONNECTED, Message: err.Error()})
|
this.processEvent(HubEvent{EventType: EVENT_CONNECTION_STATE_CHANGED, StateChange: CONNECTIONSTATE_DISCONNECTED, Message: err.Error()})
|
||||||
|
|
||||||
time.Sleep(30 * time.Second) // Wait before reconnect
|
if this.autoReconnect {
|
||||||
|
time.Sleep(AUTO_RECONNECT_AFTER) // Wait before reconnect
|
||||||
continue
|
continue
|
||||||
|
} else {
|
||||||
|
return // leave the worker for good
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (this *HubConnectionOptions) prepareConnection() *HubConnection {
|
|
||||||
if this.Self.ClientTag == "" {
|
|
||||||
this.Self.ClientTag = DEFAULT_CLIENT_TAG
|
|
||||||
}
|
|
||||||
|
|
||||||
hc := HubConnection{
|
|
||||||
Hco: this,
|
|
||||||
HubName: DEFAULT_HUB_NAME,
|
|
||||||
State: CONNECTIONSTATE_DISCONNECTED,
|
|
||||||
Users: make(map[string]UserInfo),
|
|
||||||
}
|
|
||||||
|
|
||||||
return &hc
|
|
||||||
}
|
|
||||||
|
|
||||||
// Connects to an NMDC server, and spawns a background goroutine to handle
|
|
||||||
// protocol messages. Client code should select on all the interface channels.
|
|
||||||
func (this *HubConnectionOptions) Connect() *HubConnection {
|
|
||||||
|
|
||||||
if this.NumEventsToBuffer < 1 {
|
|
||||||
this.NumEventsToBuffer = 1
|
|
||||||
}
|
|
||||||
|
|
||||||
hc := this.prepareConnection()
|
|
||||||
hc.OnEvent = make(chan HubEvent, this.NumEventsToBuffer)
|
|
||||||
hc.processEvent = func(ev HubEvent) {
|
|
||||||
hc.OnEvent <- ev
|
|
||||||
}
|
|
||||||
|
|
||||||
go hc.worker()
|
|
||||||
return hc
|
|
||||||
}
|
|
||||||
|
|
||||||
// Connects to an NMDC server, and blocks forever to handle protocol messages.
|
|
||||||
// Client code should supply an event handling function.
|
|
||||||
func (this *HubConnectionOptions) ConnectSync() {
|
|
||||||
hc := this.prepareConnection()
|
|
||||||
hc.worker()
|
|
||||||
}
|
|
||||||
68
HubConnectionOptions.go
Normal file
68
HubConnectionOptions.go
Normal file
@@ -0,0 +1,68 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
type HubConnectionOptions struct {
|
||||||
|
Address HubAddress
|
||||||
|
SkipVerifyTLS bool // using a negative verb, because bools default to false
|
||||||
|
SkipAutoReconnect bool // as above
|
||||||
|
Self UserInfo
|
||||||
|
NickPassword string
|
||||||
|
|
||||||
|
// Returning messages in async mode
|
||||||
|
NumEventsToBuffer uint
|
||||||
|
|
||||||
|
// Returning messages in sync mode
|
||||||
|
OnEventSync func(*HubConnection, HubEvent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (this *HubConnectionOptions) prepareConnection() *HubConnection {
|
||||||
|
if this.Self.ClientTag == "" {
|
||||||
|
this.Self.ClientTag = DEFAULT_CLIENT_TAG
|
||||||
|
this.Self.ClientVersion = DEFAULT_CLIENT_VERSION
|
||||||
|
}
|
||||||
|
|
||||||
|
// Shouldn't be blank either
|
||||||
|
if this.Self.ClientVersion == "" {
|
||||||
|
this.Self.ClientVersion = "0"
|
||||||
|
}
|
||||||
|
|
||||||
|
hc := HubConnection{
|
||||||
|
Hco: this,
|
||||||
|
HubName: DEFAULT_HUB_NAME,
|
||||||
|
State: CONNECTIONSTATE_DISCONNECTED,
|
||||||
|
users: make(map[string]UserInfo),
|
||||||
|
|
||||||
|
autoReconnect: !this.SkipAutoReconnect,
|
||||||
|
}
|
||||||
|
|
||||||
|
return &hc
|
||||||
|
}
|
||||||
|
|
||||||
|
// Connects to an NMDC server, and spawns a background goroutine to handle
|
||||||
|
// protocol messages. Events will be sent by channel to the returned hc.OnEvent,
|
||||||
|
// the client is responsible for selecting off this.
|
||||||
|
func (this *HubConnectionOptions) Connect() *HubConnection {
|
||||||
|
|
||||||
|
if this.NumEventsToBuffer < 1 {
|
||||||
|
this.NumEventsToBuffer = 1
|
||||||
|
}
|
||||||
|
|
||||||
|
hc := this.prepareConnection()
|
||||||
|
hc.OnEvent = make(chan HubEvent, this.NumEventsToBuffer)
|
||||||
|
hc.processEvent = func(ev HubEvent) {
|
||||||
|
hc.OnEvent <- ev
|
||||||
|
}
|
||||||
|
|
||||||
|
go hc.worker()
|
||||||
|
return hc
|
||||||
|
}
|
||||||
|
|
||||||
|
// Connects to an NMDC server, and blocks forever to handle protocol messages.
|
||||||
|
// Client code should supply an event handling function as hco.OnEventSync.
|
||||||
|
func (this *HubConnectionOptions) ConnectSync() {
|
||||||
|
hc := this.prepareConnection()
|
||||||
|
hc.OnEvent = nil
|
||||||
|
hc.processEvent = func(ev HubEvent) {
|
||||||
|
this.OnEventSync(hc, ev)
|
||||||
|
}
|
||||||
|
hc.worker()
|
||||||
|
}
|
||||||
25
HubEvent.go
Normal file
25
HubEvent.go
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
const (
|
||||||
|
EVENT_PUBLIC = 1
|
||||||
|
EVENT_PRIVATE = 2
|
||||||
|
EVENT_SYSTEM_MESSAGE_FROM_HUB = 3
|
||||||
|
EVENT_SYSTEM_MESSAGE_FROM_CONN = 4
|
||||||
|
EVENT_USER_JOINED = 5
|
||||||
|
EVENT_USER_PART = 6
|
||||||
|
EVENT_USER_UPDATED_INFO = 7
|
||||||
|
EVENT_CONNECTION_STATE_CHANGED = 8
|
||||||
|
EVENT_HUBNAME_CHANGED = 9
|
||||||
|
EVENT_DEBUG_MESSAGE = 10
|
||||||
|
EVENT_USERCOMMAND = 11
|
||||||
|
)
|
||||||
|
|
||||||
|
type HubEventType int
|
||||||
|
|
||||||
|
type HubEvent struct {
|
||||||
|
EventType HubEventType
|
||||||
|
Nick string
|
||||||
|
Message string
|
||||||
|
StateChange ConnectionState
|
||||||
|
UserCommand *UserCommand
|
||||||
|
}
|
||||||
26
UserCommand.go
Normal file
26
UserCommand.go
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
type UserCommandType uint8
|
||||||
|
|
||||||
|
const (
|
||||||
|
USERCOMMAND_TYPE_SEPARATOR UserCommandType = 0
|
||||||
|
USERCOMMAND_TYPE_RAW UserCommandType = 1
|
||||||
|
USERCOMMAND_TYPE_NICKLIMITED UserCommandType = 2
|
||||||
|
USERCOMMAND_TYPE_CLEARALL UserCommandType = 255
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserCommandContext uint8
|
||||||
|
|
||||||
|
const (
|
||||||
|
USERCOMMAND_CONTEXT_HUB UserCommandContext = 1
|
||||||
|
USERCOMMAND_CONTEXT_USER UserCommandContext = 2
|
||||||
|
USERCOMMAND_CONTEXT_SEARCH UserCommandContext = 4
|
||||||
|
USERCOMMAND_CONTEXT_FILELIST UserCommandContext = 8
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserCommand struct {
|
||||||
|
Type UserCommandType
|
||||||
|
Context UserCommandContext
|
||||||
|
Message string
|
||||||
|
Command string
|
||||||
|
}
|
||||||
17
UserFlag.go
Normal file
17
UserFlag.go
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
type UserFlag byte
|
||||||
|
|
||||||
|
const (
|
||||||
|
FLAG_NORMAL UserFlag = 1
|
||||||
|
FLAG_AWAY_1 UserFlag = 2
|
||||||
|
FLAG_AWAY_2 UserFlag = 3
|
||||||
|
FLAG_SERVER_1 UserFlag = 4
|
||||||
|
FLAG_SERVER_2 UserFlag = 5
|
||||||
|
FLAG_SERVER_AWAY_1 UserFlag = 6
|
||||||
|
FLAG_SERVER_AWAY_2 UserFlag = 7
|
||||||
|
FLAG_FIREBALL_1 UserFlag = 8
|
||||||
|
FLAG_FIREBALL_2 UserFlag = 9
|
||||||
|
FLAG_FIREBALL_AWAY_1 UserFlag = 10
|
||||||
|
FLAG_FIREBALL_AWAY_2 UserFlag = 11
|
||||||
|
)
|
||||||
132
UserInfo.go
Normal file
132
UserInfo.go
Normal file
@@ -0,0 +1,132 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// This structure represents a user connected to a hub.
|
||||||
|
type UserInfo struct {
|
||||||
|
Nick string
|
||||||
|
Description string
|
||||||
|
ClientTag string
|
||||||
|
ClientVersion string
|
||||||
|
Email string
|
||||||
|
ShareSize uint64
|
||||||
|
ConnectionMode ConnectionMode
|
||||||
|
Flag UserFlag
|
||||||
|
Slots uint64
|
||||||
|
Speed string
|
||||||
|
HubsUnregistered uint64
|
||||||
|
HubsRegistered uint64
|
||||||
|
HubsOperator uint64
|
||||||
|
IsOperator bool
|
||||||
|
}
|
||||||
|
|
||||||
|
var rx_myinfo *regexp.Regexp
|
||||||
|
var rx_myinfo_notag *regexp.Regexp
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
// Format: $ALL <nick> <description>$ $<connection><flag>$<e-mail>$<sharesize>$
|
||||||
|
|
||||||
|
HEAD := `(?ms)^\$ALL ([^ ]+) `
|
||||||
|
FOOT := `\$.\$([^$]+)\$([^$]*)\$([0-9]*)\$$`
|
||||||
|
|
||||||
|
rx_myinfo = regexp.MustCompile(HEAD + `([^<]*)<(.+?) V:([^,]+),M:(.),H:([0-9]+)/([0-9]+)/([0-9]+),S:([0-9]+)>` + FOOT)
|
||||||
|
rx_myinfo_notag = regexp.MustCompile(HEAD + `([^$]*)` + FOOT) // Fallback for no tag
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewUserInfo(username string) *UserInfo {
|
||||||
|
return &UserInfo{
|
||||||
|
Nick: username,
|
||||||
|
ConnectionMode: CONNECTIONMODE_PASSIVE,
|
||||||
|
HubsUnregistered: 1,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func maybeParse(str string, dest *uint64, default_val uint64) {
|
||||||
|
sz, err := strconv.ParseUint(str, 10, 64)
|
||||||
|
if err == nil {
|
||||||
|
*dest = sz
|
||||||
|
} else {
|
||||||
|
*dest = default_val
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (this *UserInfo) fromMyINFO(protomsg string) error {
|
||||||
|
|
||||||
|
// Normal format (with tag in exact V/M/H/S order)
|
||||||
|
matches := rx_myinfo.FindStringSubmatch(protomsg)
|
||||||
|
if matches != nil {
|
||||||
|
this.Nick = matches[1]
|
||||||
|
this.Description = Unescape(matches[2])
|
||||||
|
this.ClientTag = Unescape(matches[3])
|
||||||
|
this.ClientVersion = matches[4]
|
||||||
|
this.ConnectionMode = ConnectionMode(matches[5][0])
|
||||||
|
maybeParse(matches[6], &this.HubsUnregistered, 0)
|
||||||
|
maybeParse(matches[7], &this.HubsRegistered, 0)
|
||||||
|
maybeParse(matches[8], &this.HubsOperator, 0)
|
||||||
|
maybeParse(matches[9], &this.Slots, 0)
|
||||||
|
if len(matches[10]) > 1 {
|
||||||
|
this.Speed = matches[10][:len(matches[10])-2]
|
||||||
|
} else {
|
||||||
|
this.Speed = ""
|
||||||
|
}
|
||||||
|
this.Flag = UserFlag(matches[10][len(matches[10])-1])
|
||||||
|
this.Email = Unescape(matches[11])
|
||||||
|
maybeParse(matches[12], &this.ShareSize, 0)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// No-tag format, used in early connection
|
||||||
|
matches = rx_myinfo_notag.FindStringSubmatch(protomsg)
|
||||||
|
if matches != nil {
|
||||||
|
this.Nick = matches[1]
|
||||||
|
this.Description = Unescape(matches[2])
|
||||||
|
this.ClientTag = ""
|
||||||
|
this.ClientVersion = "0"
|
||||||
|
this.ConnectionMode = CONNECTIONMODE_PASSIVE
|
||||||
|
this.HubsUnregistered = 0
|
||||||
|
this.HubsRegistered = 0
|
||||||
|
this.HubsOperator = 0
|
||||||
|
this.Slots = 0
|
||||||
|
|
||||||
|
if len(matches[3]) > 1 {
|
||||||
|
this.Speed = matches[3][:len(matches[3])-2]
|
||||||
|
} else {
|
||||||
|
this.Speed = ""
|
||||||
|
}
|
||||||
|
this.Flag = UserFlag(matches[3][len(matches[3])-1])
|
||||||
|
this.Email = Unescape(matches[4])
|
||||||
|
maybeParse(matches[5], &this.ShareSize, 0)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Couldn't get anything out of it...
|
||||||
|
return errors.New("Malformed MyINFO")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns the MyINFO command, WITH leading $MyINFO, and WITHOUT trailing pipe
|
||||||
|
func (this *UserInfo) toMyINFO() string {
|
||||||
|
return fmt.Sprintf(
|
||||||
|
"$MyINFO $ALL %s %s<%s V:%s,M:%c,H:%d/%d/%d,S:%d>$ $%s%c$%s$%d$",
|
||||||
|
this.Nick,
|
||||||
|
this.Description,
|
||||||
|
this.ClientTag,
|
||||||
|
strings.Replace(this.ClientVersion, ",", "-", -1), // just in case
|
||||||
|
this.ConnectionMode,
|
||||||
|
this.HubsUnregistered,
|
||||||
|
this.HubsRegistered,
|
||||||
|
this.HubsOperator,
|
||||||
|
this.Slots,
|
||||||
|
this.Speed,
|
||||||
|
this.Flag,
|
||||||
|
this.Email,
|
||||||
|
this.ShareSize,
|
||||||
|
)
|
||||||
|
}
|
||||||
88
UserInfo_test.go
Normal file
88
UserInfo_test.go
Normal file
@@ -0,0 +1,88 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
type myInfoTestPair struct {
|
||||||
|
in string
|
||||||
|
expect UserInfo
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMyINFOParse(t *testing.T) {
|
||||||
|
|
||||||
|
cases := []myInfoTestPair{
|
||||||
|
|
||||||
|
myInfoTestPair{
|
||||||
|
in: "$ALL Bxxxy description<ApexDC++ V:1.4.3,M:P,H:9/0/2,S:1>$ $0.01\x01$xyz@example.com$53054999578$",
|
||||||
|
expect: UserInfo{
|
||||||
|
Nick: "Bxxxy",
|
||||||
|
Description: "description",
|
||||||
|
ClientTag: "ApexDC++",
|
||||||
|
ClientVersion: "1.4.3",
|
||||||
|
Email: "xyz@example.com",
|
||||||
|
ShareSize: 53054999578,
|
||||||
|
ConnectionMode: CONNECTIONMODE_PASSIVE,
|
||||||
|
Flag: FLAG_NORMAL,
|
||||||
|
Slots: 1,
|
||||||
|
Speed: "0.0",
|
||||||
|
HubsUnregistered: 9,
|
||||||
|
HubsRegistered: 0,
|
||||||
|
HubsOperator: 2,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
myInfoTestPair{
|
||||||
|
in: "$ALL ixxxxxxx0 $P$10A$$0$",
|
||||||
|
expect: UserInfo{
|
||||||
|
Nick: "ixxxxxxx0",
|
||||||
|
ClientVersion: "0", // Auto-inserted by the parser for short-format MyINFO strings
|
||||||
|
ConnectionMode: CONNECTIONMODE_PASSIVE,
|
||||||
|
Flag: UserFlag(rune('A')),
|
||||||
|
Speed: "1",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
myInfoTestPair{
|
||||||
|
in: "$ALL SXXXX_XXXXXXR <ncdc V:1.19.1-12-g5561,M:P,H:1/0/0,S:10>$ $0.005Q$$0$",
|
||||||
|
expect: UserInfo{
|
||||||
|
Nick: "SXXXX_XXXXXXR",
|
||||||
|
ClientTag: "ncdc",
|
||||||
|
ClientVersion: "1.19.1-12-g5561",
|
||||||
|
ConnectionMode: CONNECTIONMODE_PASSIVE,
|
||||||
|
Flag: UserFlag(rune('Q')),
|
||||||
|
Slots: 10,
|
||||||
|
Speed: "0.00",
|
||||||
|
HubsUnregistered: 1,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
myInfoTestPair{
|
||||||
|
in: "$ALL mxxxu desccccc<HexChat V:2.12.1,M:P,H:1/0/0,S:0>$ $p$$0$",
|
||||||
|
expect: UserInfo{
|
||||||
|
Nick: "mxxxu",
|
||||||
|
Description: "desccccc",
|
||||||
|
ClientTag: "HexChat",
|
||||||
|
ClientVersion: "2.12.1",
|
||||||
|
ConnectionMode: CONNECTIONMODE_PASSIVE,
|
||||||
|
Flag: UserFlag(rune('p')),
|
||||||
|
HubsUnregistered: 1,
|
||||||
|
Slots: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, v := range cases {
|
||||||
|
|
||||||
|
got := UserInfo{}
|
||||||
|
err := got.fromMyINFO(v.in)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("MyINFO parse warning (%s)", err.Error())
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if got != v.expect {
|
||||||
|
t.Errorf("MyINFO parse failure\nExpected:\n%+v\nGot:\n%+v\n", v.expect, got)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
66
__dist/README.txt
Normal file
66
__dist/README.txt
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
An NMDC client protocol library for Golang.
|
||||||
|
|
||||||
|
- Offers both channel-based and synchronous APIs.
|
||||||
|
- Includes a sample logging client using the channel-based API.
|
||||||
|
- This code hosting site isn't (yet) compatible with `go get`.
|
||||||
|
|
||||||
|
Written in golang
|
||||||
|
Tags: nmdc
|
||||||
|
|
||||||
|
=CHANGELOG=
|
||||||
|
|
||||||
|
2017-02-05 0.12
|
||||||
|
- Fix an issue with mutex deadlock when accessing user information from a callback
|
||||||
|
- Fix an issue with silent disconnection if a password was required but not present
|
||||||
|
|
||||||
|
2016-11-29 0.11
|
||||||
|
- BREAKING: Remove some exported methods
|
||||||
|
- BREAKING: Fix an issue with missing sufficient parameters in the synchronous API
|
||||||
|
- Enhancement: Improve output under godoc
|
||||||
|
- Fix an issue with special characters appearing in recieved private messages
|
||||||
|
- Fix an issue with parsing active/passive connection modes
|
||||||
|
- Fix an issue with errors appearing on stdout
|
||||||
|
|
||||||
|
2016-10-08 r10
|
||||||
|
- Feature: Support `$UserCommand`
|
||||||
|
|
||||||
|
2016-08-27 r9
|
||||||
|
- Fix an issue with parsing MyINFO strings with zero-length speed descriptions
|
||||||
|
- Fix an issue with not storing updated profile information
|
||||||
|
|
||||||
|
2016-05-10 r8
|
||||||
|
- Enhancement: Separate `ClientTag` and `ClientVersion` in `UserInfo` structs
|
||||||
|
|
||||||
|
2016-05-08 r7
|
||||||
|
- BREAKING: Remove direct access to `HubConnection.Users` map
|
||||||
|
- Feature: Threadsafe user map accessor
|
||||||
|
- Feature: Option to disable auto-reconnection
|
||||||
|
- Feature: New `Disconnect()`, `UserCount()`, `UserExists()` functions
|
||||||
|
- Enhancement: Support `$OpList`, add `IsOperator` member to `UserInfo` structs
|
||||||
|
- Refactor into multiple files
|
||||||
|
|
||||||
|
2016-04-16 r6
|
||||||
|
- Fix an issue with calling `panic()` on certain types of abnormal network failure
|
||||||
|
|
||||||
|
2016-04-04 r5
|
||||||
|
- Enhancement: Support protocol keepalives
|
||||||
|
- Enhancement: Support hub redirects (`$ForceMove`)
|
||||||
|
|
||||||
|
2016-04-03 r4
|
||||||
|
- Feature: Add synchronous API
|
||||||
|
- Fix an issue with reading HubConnection's state parameter
|
||||||
|
- Fix an issue with buffered protocol commands
|
||||||
|
|
||||||
|
2016-04-03 r3
|
||||||
|
- Feature: Add `SkipVerifyTLS` option
|
||||||
|
- Fix an issue with calling `panic()` if connection failed
|
||||||
|
|
||||||
|
2016-04-02 r2
|
||||||
|
- Enhancement: Support NMDC-over-TLS (NMDCS)
|
||||||
|
- Fix an issue recieving private messages
|
||||||
|
- Fix an issue with calling `panic()` if connection failed
|
||||||
|
- Fix an issue parsing URIs without a specified port
|
||||||
|
- Move sample content into directory with excluded build
|
||||||
|
|
||||||
|
2016-02-12 r1
|
||||||
|
- Initial public release
|
||||||
64
libnmdc.go
Normal file
64
libnmdc.go
Normal file
@@ -0,0 +1,64 @@
|
|||||||
|
package libnmdc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
DEFAULT_CLIENT_TAG string = "libnmdc.go"
|
||||||
|
DEFAULT_CLIENT_VERSION string = "0.11"
|
||||||
|
DEFAULT_HUB_NAME string = "(unknown)"
|
||||||
|
SEND_KEEPALIVE_EVERY time.Duration = 29 * time.Second
|
||||||
|
AUTO_RECONNECT_AFTER time.Duration = 30 * time.Second
|
||||||
|
)
|
||||||
|
|
||||||
|
var rx_protocolMessage *regexp.Regexp
|
||||||
|
var rx_publicChat *regexp.Regexp
|
||||||
|
var rx_incomingTo *regexp.Regexp
|
||||||
|
var rx_userCommand *regexp.Regexp
|
||||||
|
var ErrNotConnected error = errors.New("Not connected")
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
rx_protocolMessage = regexp.MustCompile("(?ms)^[^|]*\\|")
|
||||||
|
rx_publicChat = regexp.MustCompile("(?ms)^<([^>]*)> (.*)$")
|
||||||
|
rx_incomingTo = regexp.MustCompile("(?ms)^([^ ]+) From: ([^ ]+) \\$<([^>]*)> (.*)")
|
||||||
|
rx_userCommand = regexp.MustCompile(`(?ms)^(\d+) (\d+)\s?([^\$]*)\$?(.*)`)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Unescape(encoded string) string {
|
||||||
|
v1 := strings.Replace(encoded, "$", "$", -1)
|
||||||
|
v2 := strings.Replace(v1, "|", "|", -1)
|
||||||
|
return strings.Replace(v2, "&", "&", -1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Escape(plaintext string) string {
|
||||||
|
v1 := strings.Replace(plaintext, "&", "&", -1)
|
||||||
|
v2 := strings.Replace(v1, "|", "|", -1)
|
||||||
|
return strings.Replace(v2, "$", "$", -1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func unlock(lock []byte) string {
|
||||||
|
|
||||||
|
nibble_swap := func(b byte) byte {
|
||||||
|
return ((b << 4) & 0xF0) | ((b >> 4) & 0x0F)
|
||||||
|
}
|
||||||
|
|
||||||
|
chr := func(b byte) string {
|
||||||
|
if b == 0 || b == 5 || b == 36 || b == 96 || b == 124 || b == 126 {
|
||||||
|
return fmt.Sprintf("/%%DCN%04d%%/", b)
|
||||||
|
} else {
|
||||||
|
return string(b)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
key := chr(nibble_swap(lock[0] ^ lock[len(lock)-2] ^ lock[len(lock)-3] ^ 5))
|
||||||
|
for i := 1; i < len(lock); i += 1 {
|
||||||
|
key += chr(nibble_swap(lock[i] ^ lock[i-1]))
|
||||||
|
}
|
||||||
|
|
||||||
|
return key
|
||||||
|
}
|
||||||
@@ -1,152 +0,0 @@
|
|||||||
// libnmdc project UserInfo.go
|
|
||||||
package libnmdc
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"regexp"
|
|
||||||
"strconv"
|
|
||||||
)
|
|
||||||
|
|
||||||
type UserFlag byte
|
|
||||||
|
|
||||||
const (
|
|
||||||
FLAG_NORMAL UserFlag = 1
|
|
||||||
FLAG_AWAY_1 UserFlag = 2
|
|
||||||
FLAG_AWAY_2 UserFlag = 3
|
|
||||||
FLAG_SERVER_1 UserFlag = 4
|
|
||||||
FLAG_SERVER_2 UserFlag = 5
|
|
||||||
FLAG_SERVER_AWAY_1 UserFlag = 6
|
|
||||||
FLAG_SERVER_AWAY_2 UserFlag = 7
|
|
||||||
FLAG_FIREBALL_1 UserFlag = 8
|
|
||||||
FLAG_FIREBALL_2 UserFlag = 9
|
|
||||||
FLAG_FIREBALL_AWAY_1 UserFlag = 10
|
|
||||||
FLAG_FIREBALL_AWAY_2 UserFlag = 11
|
|
||||||
)
|
|
||||||
|
|
||||||
type ConnectionMode rune
|
|
||||||
|
|
||||||
const (
|
|
||||||
CONNECTIONMODE_ACTIVE ConnectionMode = 'A'
|
|
||||||
CONNECTIONMODE_PASSIVE ConnectionMode = 'P'
|
|
||||||
CONNECTIONMODE_SOCKS5 ConnectionMode = '5'
|
|
||||||
)
|
|
||||||
|
|
||||||
// This structure represents a user connected to a hub.
|
|
||||||
type UserInfo struct {
|
|
||||||
Nick string
|
|
||||||
Description string
|
|
||||||
ClientTag string
|
|
||||||
Email string
|
|
||||||
ShareSize uint64
|
|
||||||
ConnectionMode ConnectionMode
|
|
||||||
Flag UserFlag
|
|
||||||
Slots uint64
|
|
||||||
Speed string
|
|
||||||
HubsUnregistered uint64
|
|
||||||
HubsRegistered uint64
|
|
||||||
HubsOperator uint64
|
|
||||||
}
|
|
||||||
|
|
||||||
var rx_myinfo *regexp.Regexp
|
|
||||||
var rx_myinfo_notag *regexp.Regexp
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
// $ALL <nick> <description>$ $<connection><flag>$<e-mail>$<sharesize>$
|
|
||||||
rx_myinfo = regexp.MustCompile("(?ms)^\\$ALL ([^ ]+) ([^<]*)<([^,]*),M:(.),H:([0-9]+)/([0-9]+)/([0-9]+),S:([0-9]+)>\\$.\\$(.+?)(.)\\$([^$]*)\\$([0-9]*)\\$$")
|
|
||||||
rx_myinfo_notag = regexp.MustCompile("(?ms)^\\$ALL ([^ ]+) ([^$]*)\\$.\\$(.*)(.)\\$([^$]*)\\$([0-9]*)\\$$") // Fallback for no tag
|
|
||||||
|
|
||||||
/*
|
|
||||||
sample := "$ALL Betty description<ApexDC++ V:1.4.3,M:P,H:9/0/2,S:1>$ $0.01\x01$xyz@xyz.com$53054999578$"
|
|
||||||
sample = "$ALL ivysaur80 $P$10A$$0$"
|
|
||||||
|
|
||||||
u := UserInfo{}
|
|
||||||
err := u.fromMyINFO(sample)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println(err.Error())
|
|
||||||
} else {
|
|
||||||
fmt.Printf("%+v\n", u)
|
|
||||||
}
|
|
||||||
|
|
||||||
os.Exit(1)
|
|
||||||
*/
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewUserInfo(username string) *UserInfo {
|
|
||||||
return &UserInfo{
|
|
||||||
Nick: username,
|
|
||||||
ConnectionMode: CONNECTIONMODE_PASSIVE,
|
|
||||||
HubsUnregistered: 1,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func maybeParse(str string, dest *uint64, default_val uint64) {
|
|
||||||
sz, err := strconv.ParseUint(str, 10, 64)
|
|
||||||
if err == nil {
|
|
||||||
*dest = sz
|
|
||||||
} else {
|
|
||||||
*dest = default_val
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (this *UserInfo) fromMyINFO(protomsg string) error {
|
|
||||||
// Normal format (with tag in exact M/H/S order)
|
|
||||||
matches := rx_myinfo.FindStringSubmatch(protomsg)
|
|
||||||
if matches != nil {
|
|
||||||
this.Nick = matches[1]
|
|
||||||
this.Description = NMDCUnescape(matches[2])
|
|
||||||
this.ClientTag = NMDCUnescape(matches[3])
|
|
||||||
this.ConnectionMode = ConnectionMode(matches[4][0])
|
|
||||||
maybeParse(matches[5], &this.HubsUnregistered, 0)
|
|
||||||
maybeParse(matches[6], &this.HubsRegistered, 0)
|
|
||||||
maybeParse(matches[7], &this.HubsOperator, 0)
|
|
||||||
maybeParse(matches[8], &this.Slots, 0)
|
|
||||||
this.Speed = matches[9]
|
|
||||||
this.Flag = UserFlag(matches[10][0])
|
|
||||||
this.Email = NMDCUnescape(matches[11])
|
|
||||||
maybeParse(matches[12], &this.ShareSize, 0)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// No-tag format, used in early connection
|
|
||||||
matches = rx_myinfo_notag.FindStringSubmatch(protomsg)
|
|
||||||
if matches != nil {
|
|
||||||
this.Nick = matches[1]
|
|
||||||
this.Description = NMDCUnescape(matches[2])
|
|
||||||
this.ClientTag = ""
|
|
||||||
this.ConnectionMode = CONNECTIONMODE_PASSIVE
|
|
||||||
this.HubsUnregistered = 0
|
|
||||||
this.HubsRegistered = 0
|
|
||||||
this.HubsOperator = 0
|
|
||||||
this.Slots = 0
|
|
||||||
this.Speed = matches[3]
|
|
||||||
this.Flag = UserFlag(matches[4][0])
|
|
||||||
this.Email = NMDCUnescape(matches[5])
|
|
||||||
maybeParse(matches[6], &this.ShareSize, 0)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Couldn't get anything out of it...
|
|
||||||
return errors.New("Malformed MyINFO")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Returns the MyINFO command, WITH leading $MyINFO, and WITHOUT trailing pipe
|
|
||||||
func (this *UserInfo) toMyINFO() string {
|
|
||||||
return fmt.Sprintf(
|
|
||||||
"$MyINFO $ALL %s %s<%s,M:%c,H:%d/%d/%d,S:%d>$ $%s%c$%s$%d$",
|
|
||||||
this.Nick,
|
|
||||||
this.Description,
|
|
||||||
this.ClientTag,
|
|
||||||
this.ConnectionMode,
|
|
||||||
this.HubsUnregistered,
|
|
||||||
this.HubsRegistered,
|
|
||||||
this.HubsOperator,
|
|
||||||
this.Slots,
|
|
||||||
this.Speed,
|
|
||||||
this.Flag,
|
|
||||||
this.Email,
|
|
||||||
this.ShareSize,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
@@ -1,29 +0,0 @@
|
|||||||
An NMDC client protocol library for Golang.
|
|
||||||
|
|
||||||
- Offers both channel-based and synchronous APIs.
|
|
||||||
- Includes a sample logging client using the channel-based API.
|
|
||||||
- This code hosting site isn't (yet) compatible with `go get`.
|
|
||||||
|
|
||||||
Written in golang
|
|
||||||
Tags: nmdc
|
|
||||||
|
|
||||||
=CHANGELOG=
|
|
||||||
|
|
||||||
2016-04-03 r4
|
|
||||||
- Feature: Add synchronous API
|
|
||||||
- Fix an issue with reading HubConnection's state parameter
|
|
||||||
- Fix an issue with buffered protocol commands
|
|
||||||
|
|
||||||
2016-04-03 r3
|
|
||||||
- Feature: Add `SkipVerifyTLS` option
|
|
||||||
- Fix an issue with calling `panic()` if connection failed
|
|
||||||
|
|
||||||
2016-04-02 r2
|
|
||||||
- Enhancement: Support NMDC-over-TLS (NMDCS)
|
|
||||||
- Fix an issue recieving private messages
|
|
||||||
- Fix an issue with calling `panic()` if connection failed
|
|
||||||
- Fix an issue parsing URIs without a specified port
|
|
||||||
- Move sample content into directory with excluded build
|
|
||||||
|
|
||||||
2016-02-12 r1
|
|
||||||
- Initial public release
|
|
||||||
@@ -1,38 +0,0 @@
|
|||||||
// +build ignore
|
|
||||||
|
|
||||||
// This sample file demonstrates use of the libnmdc.go library. It's excluded
|
|
||||||
// when building the library package, but you can run it via `go run libnmdc_sample.go`.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"libnmdc"
|
|
||||||
)
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
|
|
||||||
opts := libnmdc.HubConnectionOptions{
|
|
||||||
Address: "127.0.0.1",
|
|
||||||
Self: libnmdc.UserInfo{Nick: "slowpoke9"},
|
|
||||||
}
|
|
||||||
hub := opts.Connect()
|
|
||||||
|
|
||||||
for {
|
|
||||||
event := <-hub.OnEvent
|
|
||||||
switch event.EventType {
|
|
||||||
case libnmdc.EVENT_CONNECTION_STATE_CHANGED:
|
|
||||||
fmt.Printf("Connection -- %s (%s)\n", event.StateChange.Format(), event.Message)
|
|
||||||
|
|
||||||
case libnmdc.EVENT_PUBLIC:
|
|
||||||
fmt.Printf("Message from '%s': '%s'\n", event.Nick, event.Message)
|
|
||||||
if event.Message == "how are you" {
|
|
||||||
hub.SayPublic("good thanks!")
|
|
||||||
}
|
|
||||||
|
|
||||||
default:
|
|
||||||
fmt.Printf("%+v\n", event)
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,43 +0,0 @@
|
|||||||
A logging service for NMDC hubs.
|
|
||||||
|
|
||||||
It logs public chat messages to a file, categorised by months. Binaries are provided for Windows/Linux amd64/i386.
|
|
||||||
|
|
||||||
Written in golang
|
|
||||||
Tags: nmdc
|
|
||||||
|
|
||||||
=USAGE=
|
|
||||||
|
|
||||||
`$nmdc-log-service -Help
|
|
||||||
|
|
||||||
Usage of nmdc-log-service:
|
|
||||||
-Debug
|
|
||||||
Print additional information on stdout
|
|
||||||
-Dir string
|
|
||||||
Output directory (default ".")
|
|
||||||
-LogConnectionState
|
|
||||||
Include connection state changes in log (default true)
|
|
||||||
-Nick string
|
|
||||||
Nick (default "nmdc-log-service")
|
|
||||||
-PMResponse string
|
|
||||||
Message to respond with on PM (default "This is an automated service. For enquiries, please contact an administrator.")
|
|
||||||
-Password string
|
|
||||||
Registered nick password
|
|
||||||
-Server string
|
|
||||||
Addresses to connect to (comma-separated)
|
|
||||||
-VerifyTLS
|
|
||||||
Verify TLS certificates (default true)
|
|
||||||
`
|
|
||||||
|
|
||||||
=CHANGELOG=
|
|
||||||
|
|
||||||
2016-04-03 1.0.2
|
|
||||||
- Enhancement: Upgrade `libnmdc` from `r3` to `r4`
|
|
||||||
|
|
||||||
2016-04-03 1.0.1
|
|
||||||
- Enhancement: Add `-VerifyTLS` option
|
|
||||||
- Enhancement: Upgrade `libnmdc` from `r2` to `r3`
|
|
||||||
- Fix an issue writing log files on Linux
|
|
||||||
- Fix a cosmetic issue with error message formatting
|
|
||||||
|
|
||||||
2016-04-02 1.0.0
|
|
||||||
- Initial public release
|
|
||||||
@@ -1,49 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -eu
|
|
||||||
|
|
||||||
export GOPATH=$(
|
|
||||||
cd ../../
|
|
||||||
cygpath -w "$(pwd)"
|
|
||||||
)
|
|
||||||
|
|
||||||
main() {
|
|
||||||
|
|
||||||
local version=""
|
|
||||||
read -p "Enter version string (blank for timestamp)> " version
|
|
||||||
if [[ $version == "" ]] ; then
|
|
||||||
version=$(date +%s)
|
|
||||||
fi
|
|
||||||
echo "Using '${version}' as the version."
|
|
||||||
|
|
||||||
if [[ -f nmdc-log-service.exe ]] ; then
|
|
||||||
rm ./nmdc-log-service.exe
|
|
||||||
fi
|
|
||||||
if [[ -f nmdc-log-service ]] ; then
|
|
||||||
rm ./nmdc-log-service
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo "Building win64..."
|
|
||||||
GOARCH=amd64 GOOS=windows go build -a -ldflags -s -o nmdc-log-service.exe
|
|
||||||
7z a -mx9 nmdc-log-service-${version}-win64.7z nmdc-log-service.exe >/dev/null
|
|
||||||
rm ./nmdc-log-service.exe
|
|
||||||
|
|
||||||
echo "Building win32..."
|
|
||||||
GOARCH=386 GOOS=windows go build -a -ldflags -s -o nmdc-log-service.exe
|
|
||||||
7z a -mx9 nmdc-log-service-${version}-win32.7z nmdc-log-service.exe >/dev/null
|
|
||||||
rm ./nmdc-log-service.exe
|
|
||||||
|
|
||||||
echo "Building linux64..."
|
|
||||||
GOARCH=amd64 GOOS=linux go build -a -ldflags -s -o nmdc-log-service
|
|
||||||
XZ_OPT=-9 tar caf nmdc-log-service-${version}-linux64.tar.xz nmdc-log-service --owner=0 --group=0
|
|
||||||
rm ./nmdc-log-service
|
|
||||||
|
|
||||||
echo "Building linux32..."
|
|
||||||
GOARCH=386 GOOS=linux go build -a -ldflags -s -o nmdc-log-service
|
|
||||||
XZ_OPT=-9 tar caf nmdc-log-service-${version}-linux32.tar.xz nmdc-log-service --owner=0 --group=0
|
|
||||||
rm ./nmdc-log-service
|
|
||||||
|
|
||||||
echo "Build complete."
|
|
||||||
}
|
|
||||||
|
|
||||||
main "$@"
|
|
||||||
@@ -1,153 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"libnmdc"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"regexp"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var BaseDir string = "."
|
|
||||||
var PMResponse string = ""
|
|
||||||
var LogConnectionState bool
|
|
||||||
var DebugMode bool
|
|
||||||
var VerifyTLS bool
|
|
||||||
var CharacterMatcher *regexp.Regexp
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
CharacterMatcher = regexp.MustCompile("[^a-zA-Z0-9]")
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetDirectoryNameForHub(hub string) string {
|
|
||||||
return filepath.Join(BaseDir, CharacterMatcher.ReplaceAllString(hub, "_"))
|
|
||||||
}
|
|
||||||
|
|
||||||
func LogMessage(hub, message string) {
|
|
||||||
|
|
||||||
instant := time.Now() // using system timezone
|
|
||||||
|
|
||||||
// Log file path
|
|
||||||
monthStamp := instant.Format("2006-01")
|
|
||||||
logFilePath := filepath.Join(GetDirectoryNameForHub(hub), monthStamp+".log")
|
|
||||||
|
|
||||||
// Produce a timestamp
|
|
||||||
timeStamp := instant.Format("2006-01-02 15:04:05")
|
|
||||||
|
|
||||||
fh, err := os.OpenFile(logFilePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "Couldn't open file '%s': %s\n", logFilePath, err.Error())
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer fh.Close()
|
|
||||||
|
|
||||||
_, err = fh.WriteString("[" + timeStamp + "] " + message + "\n")
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "Error writing to file '%s': %s\n", logFilePath, err.Error())
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func HubWorker(addr, nick, password string) {
|
|
||||||
|
|
||||||
opts := libnmdc.HubConnectionOptions{
|
|
||||||
Address: libnmdc.HubAddress(addr),
|
|
||||||
SkipVerifyTLS: !VerifyTLS,
|
|
||||||
Self: libnmdc.UserInfo{Nick: nick},
|
|
||||||
NickPassword: password,
|
|
||||||
}
|
|
||||||
hub := opts.Connect()
|
|
||||||
|
|
||||||
for {
|
|
||||||
event := <-hub.OnEvent
|
|
||||||
|
|
||||||
if DebugMode {
|
|
||||||
fmt.Printf("DEBUG: %s %v\n", addr, event)
|
|
||||||
}
|
|
||||||
|
|
||||||
switch event.EventType {
|
|
||||||
case libnmdc.EVENT_CONNECTION_STATE_CHANGED:
|
|
||||||
if LogConnectionState {
|
|
||||||
if len(event.Message) > 0 {
|
|
||||||
LogMessage(addr, "* "+event.StateChange.Format()+" ("+event.Message+")")
|
|
||||||
} else {
|
|
||||||
LogMessage(addr, "* "+event.StateChange.Format())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
case libnmdc.EVENT_PUBLIC:
|
|
||||||
LogMessage(addr, "<"+event.Nick+"> "+event.Message)
|
|
||||||
|
|
||||||
case libnmdc.EVENT_PRIVATE:
|
|
||||||
fmt.Printf("Got PM %v\n", event)
|
|
||||||
hub.SayPrivate(event.Nick, PMResponse)
|
|
||||||
|
|
||||||
case libnmdc.EVENT_SYSTEM_MESSAGE_FROM_CONN, libnmdc.EVENT_SYSTEM_MESSAGE_FROM_HUB:
|
|
||||||
if strings.HasPrefix(event.Message, "* ") {
|
|
||||||
LogMessage(addr, event.Message)
|
|
||||||
} else {
|
|
||||||
LogMessage(addr, "* "+event.Message)
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
|
|
||||||
// Parse arguments
|
|
||||||
hubs := flag.String("Server", "", "Addresses to connect to (comma-separated)")
|
|
||||||
nick := flag.String("Nick", "nmdc-log-service", "Nick")
|
|
||||||
password := flag.String("Password", "", "Registered nick password")
|
|
||||||
flag.StringVar(&BaseDir, "Dir", ".", "Output directory")
|
|
||||||
flag.BoolVar(&LogConnectionState, "LogConnectionState", true, "Include connection state changes in log")
|
|
||||||
flag.StringVar(&PMResponse, "PMResponse", "This is an automated service. For enquiries, please contact an administrator.", "Message to respond with on PM")
|
|
||||||
flag.BoolVar(&DebugMode, "Debug", false, "Print additional information on stdout")
|
|
||||||
flag.BoolVar(&VerifyTLS, "VerifyTLS", true, "Verify TLS certificates")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
// Assert dir exists
|
|
||||||
dinfo, err := os.Stat(BaseDir)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "FATAL: %s\n", err.Error())
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
if !dinfo.IsDir() {
|
|
||||||
fmt.Fprintf(os.Stderr, "FATAL: Path '%s' is not a directory\n", BaseDir)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !VerifyTLS {
|
|
||||||
fmt.Fprintf(os.Stderr, "WARNING: TLS certificates will not be verified!\n")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Launch loggers
|
|
||||||
all_hubs := strings.Split(*hubs, ",")
|
|
||||||
launch_ct := 0
|
|
||||||
for _, hubaddr := range all_hubs {
|
|
||||||
if len(hubaddr) == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
// Assert target directory exists
|
|
||||||
os.MkdirAll(GetDirectoryNameForHub(hubaddr), 0755)
|
|
||||||
|
|
||||||
// Launch logger
|
|
||||||
go HubWorker(hubaddr, *nick, *password)
|
|
||||||
|
|
||||||
launch_ct++
|
|
||||||
}
|
|
||||||
|
|
||||||
if launch_ct == 0 {
|
|
||||||
fmt.Fprintln(os.Stderr, "FATAL: No hubs specified")
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Wait forever
|
|
||||||
var forever chan bool = nil
|
|
||||||
select {
|
|
||||||
case <-forever:
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user