]>
Commit | Line | Data |
---|---|---|
6988a057 JH |
1 | package hotline |
2 | ||
3 | import ( | |
7cd900d6 | 4 | "bufio" |
5cc444c8 | 5 | "bytes" |
6988a057 | 6 | "context" |
f8e4cd54 | 7 | "crypto/rand" |
6988a057 JH |
8 | "encoding/binary" |
9 | "errors" | |
10 | "fmt" | |
041c2df6 | 11 | "github.com/go-playground/validator/v10" |
2e1aec0f | 12 | "golang.org/x/text/encoding/charmap" |
7cd900d6 | 13 | "gopkg.in/yaml.v3" |
6988a057 | 14 | "io" |
a6216dd8 JH |
15 | "log" |
16 | "log/slog" | |
6988a057 JH |
17 | "net" |
18 | "os" | |
180d6544 | 19 | "path" |
6988a057 | 20 | "path/filepath" |
6988a057 JH |
21 | "strings" |
22 | "sync" | |
a2ef262a | 23 | "sync/atomic" |
6988a057 | 24 | "time" |
6988a057 JH |
25 | ) |
26 | ||
7cd900d6 JH |
27 | type contextKey string |
28 | ||
29 | var contextKeyReq = contextKey("req") | |
30 | ||
31 | type requestCtx struct { | |
32 | remoteAddr string | |
7cd900d6 JH |
33 | } |
34 | ||
2e1aec0f JH |
35 | // Converts bytes from Mac Roman encoding to UTF-8 |
36 | var txtDecoder = charmap.Macintosh.NewDecoder() | |
37 | ||
38 | // Converts bytes from UTF-8 to Mac Roman encoding | |
39 | var txtEncoder = charmap.Macintosh.NewEncoder() | |
40 | ||
6988a057 | 41 | type Server struct { |
a2ef262a JH |
42 | NetInterface string |
43 | Port int | |
44 | Accounts map[string]*Account | |
45 | Agreement []byte | |
46 | ||
47 | Clients map[[2]byte]*ClientConn | |
df1ade54 JH |
48 | fileTransfers map[[4]byte]*FileTransfer |
49 | ||
c1c44744 JH |
50 | Config *Config |
51 | ConfigDir string | |
a6216dd8 | 52 | Logger *slog.Logger |
5cc444c8 | 53 | banner []byte |
c1c44744 JH |
54 | |
55 | PrivateChatsMu sync.Mutex | |
a2ef262a | 56 | PrivateChats map[[4]byte]*PrivateChat |
c1c44744 | 57 | |
a2ef262a | 58 | nextClientID atomic.Uint32 |
40414f92 | 59 | TrackerPassID [4]byte |
00913df3 | 60 | |
a2ef262a | 61 | statsMu sync.Mutex |
00913df3 | 62 | Stats *Stats |
6988a057 | 63 | |
7cd900d6 | 64 | FS FileStore // Storage backend to use for File storage |
6988a057 JH |
65 | |
66 | outbox chan Transaction | |
7cd900d6 | 67 | mux sync.Mutex |
6988a057 | 68 | |
8eb43f95 JH |
69 | threadedNewsMux sync.Mutex |
70 | ThreadedNews *ThreadedNews | |
71 | ||
6988a057 | 72 | flatNewsMux sync.Mutex |
7cd900d6 | 73 | FlatNews []byte |
46862572 JH |
74 | |
75 | banListMU sync.Mutex | |
76 | banList map[string]*time.Time | |
6988a057 JH |
77 | } |
78 | ||
00913df3 | 79 | func (s *Server) CurrentStats() Stats { |
a2ef262a JH |
80 | s.statsMu.Lock() |
81 | defer s.statsMu.Unlock() | |
00913df3 JH |
82 | |
83 | stats := s.Stats | |
84 | stats.CurrentlyConnected = len(s.Clients) | |
85 | ||
86 | return *stats | |
87 | } | |
88 | ||
6988a057 JH |
89 | type PrivateChat struct { |
90 | Subject string | |
a2ef262a | 91 | ClientConn map[[2]byte]*ClientConn |
6988a057 JH |
92 | } |
93 | ||
a2ef262a | 94 | func (s *Server) ListenAndServe(ctx context.Context) error { |
6988a057 JH |
95 | var wg sync.WaitGroup |
96 | ||
97 | wg.Add(1) | |
8168522a | 98 | go func() { |
2d0f2abe | 99 | ln, err := net.Listen("tcp", fmt.Sprintf("%s:%v", s.NetInterface, s.Port)) |
8168522a | 100 | if err != nil { |
a6216dd8 | 101 | log.Fatal(err) |
8168522a JH |
102 | } |
103 | ||
a6216dd8 | 104 | log.Fatal(s.Serve(ctx, ln)) |
8168522a | 105 | }() |
6988a057 JH |
106 | |
107 | wg.Add(1) | |
8168522a | 108 | go func() { |
2d0f2abe | 109 | ln, err := net.Listen("tcp", fmt.Sprintf("%s:%v", s.NetInterface, s.Port+1)) |
8168522a | 110 | if err != nil { |
a6216dd8 | 111 | log.Fatal(err) |
8168522a JH |
112 | } |
113 | ||
a6216dd8 | 114 | log.Fatal(s.ServeFileTransfers(ctx, ln)) |
8168522a | 115 | }() |
6988a057 JH |
116 | |
117 | wg.Wait() | |
118 | ||
119 | return nil | |
120 | } | |
121 | ||
7cd900d6 | 122 | func (s *Server) ServeFileTransfers(ctx context.Context, ln net.Listener) error { |
6988a057 JH |
123 | for { |
124 | conn, err := ln.Accept() | |
125 | if err != nil { | |
126 | return err | |
127 | } | |
128 | ||
129 | go func() { | |
7cd900d6 JH |
130 | defer func() { _ = conn.Close() }() |
131 | ||
132 | err = s.handleFileTransfer( | |
a2ef262a | 133 | context.WithValue(ctx, contextKeyReq, requestCtx{remoteAddr: conn.RemoteAddr().String()}), |
7cd900d6 JH |
134 | conn, |
135 | ) | |
136 | ||
137 | if err != nil { | |
a6216dd8 | 138 | s.Logger.Error("file transfer error", "reason", err) |
6988a057 JH |
139 | } |
140 | }() | |
141 | } | |
142 | } | |
143 | ||
144 | func (s *Server) sendTransaction(t Transaction) error { | |
6988a057 | 145 | s.mux.Lock() |
a2ef262a | 146 | client, ok := s.Clients[t.clientID] |
75e4191b | 147 | s.mux.Unlock() |
a2ef262a | 148 | |
95159e55 | 149 | if !ok || client == nil { |
a2ef262a | 150 | return nil |
6988a057 | 151 | } |
6988a057 | 152 | |
a2ef262a | 153 | _, err := io.Copy(client.Connection, &t) |
75e4191b | 154 | if err != nil { |
a2ef262a | 155 | return fmt.Errorf("failed to send transaction to client %v: %v", t.clientID, err) |
6988a057 | 156 | } |
3178ae58 | 157 | |
6988a057 JH |
158 | return nil |
159 | } | |
160 | ||
67db911d JH |
161 | func (s *Server) processOutbox() { |
162 | for { | |
163 | t := <-s.outbox | |
164 | go func() { | |
165 | if err := s.sendTransaction(t); err != nil { | |
a6216dd8 | 166 | s.Logger.Error("error sending transaction", "err", err) |
67db911d JH |
167 | } |
168 | }() | |
169 | } | |
170 | } | |
171 | ||
7cd900d6 | 172 | func (s *Server) Serve(ctx context.Context, ln net.Listener) error { |
67db911d JH |
173 | go s.processOutbox() |
174 | ||
6988a057 JH |
175 | for { |
176 | conn, err := ln.Accept() | |
177 | if err != nil { | |
a6216dd8 | 178 | s.Logger.Error("error accepting connection", "err", err) |
6988a057 | 179 | } |
67db911d JH |
180 | connCtx := context.WithValue(ctx, contextKeyReq, requestCtx{ |
181 | remoteAddr: conn.RemoteAddr().String(), | |
182 | }) | |
6988a057 JH |
183 | |
184 | go func() { | |
a6216dd8 | 185 | s.Logger.Info("Connection established", "RemoteAddr", conn.RemoteAddr()) |
0da28a1f | 186 | |
46862572 | 187 | defer conn.Close() |
67db911d | 188 | if err := s.handleNewConnection(connCtx, conn, conn.RemoteAddr().String()); err != nil { |
6988a057 | 189 | if err == io.EOF { |
a6216dd8 | 190 | s.Logger.Info("Client disconnected", "RemoteAddr", conn.RemoteAddr()) |
6988a057 | 191 | } else { |
a6216dd8 | 192 | s.Logger.Error("error serving request", "RemoteAddr", conn.RemoteAddr(), "err", err) |
6988a057 JH |
193 | } |
194 | } | |
195 | }() | |
196 | } | |
197 | } | |
198 | ||
199 | const ( | |
c7e932c0 | 200 | agreementFile = "Agreement.txt" |
6988a057 JH |
201 | ) |
202 | ||
203 | // NewServer constructs a new Server from a config dir | |
a2ef262a | 204 | // TODO: move config file reads out of this function |
a6216dd8 | 205 | func NewServer(configDir, netInterface string, netPort int, logger *slog.Logger, fs FileStore) (*Server, error) { |
6988a057 | 206 | server := Server{ |
2d0f2abe | 207 | NetInterface: netInterface, |
6988a057 JH |
208 | Port: netPort, |
209 | Accounts: make(map[string]*Account), | |
210 | Config: new(Config), | |
a2ef262a | 211 | Clients: make(map[[2]byte]*ClientConn), |
df1ade54 | 212 | fileTransfers: make(map[[4]byte]*FileTransfer), |
a2ef262a | 213 | PrivateChats: make(map[[4]byte]*PrivateChat), |
6988a057 JH |
214 | ConfigDir: configDir, |
215 | Logger: logger, | |
6988a057 | 216 | outbox: make(chan Transaction), |
00913df3 | 217 | Stats: &Stats{Since: time.Now()}, |
6988a057 | 218 | ThreadedNews: &ThreadedNews{}, |
d6141467 | 219 | FS: fs, |
46862572 | 220 | banList: make(map[string]*time.Time), |
6988a057 JH |
221 | } |
222 | ||
8168522a | 223 | var err error |
6988a057 JH |
224 | |
225 | // generate a new random passID for tracker registration | |
40414f92 | 226 | if _, err := rand.Read(server.TrackerPassID[:]); err != nil { |
6988a057 JH |
227 | return nil, err |
228 | } | |
229 | ||
f22acf38 | 230 | server.Agreement, err = os.ReadFile(filepath.Join(configDir, agreementFile)) |
b196a50a | 231 | if err != nil { |
6988a057 JH |
232 | return nil, err |
233 | } | |
234 | ||
f22acf38 | 235 | if server.FlatNews, err = os.ReadFile(filepath.Join(configDir, "MessageBoard.txt")); err != nil { |
6988a057 JH |
236 | return nil, err |
237 | } | |
238 | ||
46862572 | 239 | // try to load the ban list, but ignore errors as this file may not be present or may be empty |
a2ef262a | 240 | //_ = server.loadBanList(filepath.Join(configDir, "Banlist.yaml")) |
46862572 | 241 | |
a2ef262a JH |
242 | _ = loadFromYAMLFile(filepath.Join(configDir, "Banlist.yaml"), &server.banList) |
243 | ||
244 | err = loadFromYAMLFile(filepath.Join(configDir, "ThreadedNews.yaml"), &server.ThreadedNews) | |
245 | if err != nil { | |
0ed51327 | 246 | return nil, fmt.Errorf("error loading threaded news: %w", err) |
6988a057 JH |
247 | } |
248 | ||
a2ef262a JH |
249 | err = server.loadConfig(filepath.Join(configDir, "config.yaml")) |
250 | if err != nil { | |
251 | return nil, fmt.Errorf("error loading config: %w", err) | |
6988a057 JH |
252 | } |
253 | ||
f22acf38 | 254 | if err := server.loadAccounts(filepath.Join(configDir, "Users/")); err != nil { |
6988a057 JH |
255 | return nil, err |
256 | } | |
257 | ||
70c2c110 JH |
258 | // If the FileRoot is an absolute path, use it, otherwise treat as a relative path to the config dir. |
259 | if !filepath.IsAbs(server.Config.FileRoot) { | |
260 | server.Config.FileRoot = filepath.Join(configDir, server.Config.FileRoot) | |
261 | } | |
6988a057 | 262 | |
5cc444c8 JH |
263 | server.banner, err = os.ReadFile(filepath.Join(server.ConfigDir, server.Config.BannerFile)) |
264 | if err != nil { | |
265 | return nil, fmt.Errorf("error opening banner: %w", err) | |
266 | } | |
267 | ||
6988a057 | 268 | if server.Config.EnableTrackerRegistration { |
a6216dd8 | 269 | server.Logger.Info( |
e42888eb JH |
270 | "Tracker registration enabled", |
271 | "frequency", fmt.Sprintf("%vs", trackerUpdateFrequency), | |
272 | "trackers", server.Config.Trackers, | |
273 | ) | |
274 | ||
6988a057 JH |
275 | go func() { |
276 | for { | |
e42888eb | 277 | tr := &TrackerRegistration{ |
6988a057 | 278 | UserCount: server.userCount(), |
9c44621e | 279 | PassID: server.TrackerPassID, |
6988a057 JH |
280 | Name: server.Config.Name, |
281 | Description: server.Config.Description, | |
282 | } | |
40414f92 | 283 | binary.BigEndian.PutUint16(tr.Port[:], uint16(server.Port)) |
6988a057 | 284 | for _, t := range server.Config.Trackers { |
a2ef262a | 285 | if err := register(&RealDialer{}, t, tr); err != nil { |
a6216dd8 | 286 | server.Logger.Error("unable to register with tracker %v", "error", err) |
6988a057 | 287 | } |
a6216dd8 | 288 | server.Logger.Debug("Sent Tracker registration", "addr", t) |
6988a057 JH |
289 | } |
290 | ||
291 | time.Sleep(trackerUpdateFrequency * time.Second) | |
292 | } | |
293 | }() | |
294 | } | |
295 | ||
296 | // Start Client Keepalive go routine | |
297 | go server.keepaliveHandler() | |
298 | ||
299 | return &server, nil | |
300 | } | |
301 | ||
302 | func (s *Server) userCount() int { | |
303 | s.mux.Lock() | |
304 | defer s.mux.Unlock() | |
305 | ||
306 | return len(s.Clients) | |
307 | } | |
308 | ||
309 | func (s *Server) keepaliveHandler() { | |
310 | for { | |
311 | time.Sleep(idleCheckInterval * time.Second) | |
312 | s.mux.Lock() | |
313 | ||
314 | for _, c := range s.Clients { | |
61c272e1 JH |
315 | c.IdleTime += idleCheckInterval |
316 | if c.IdleTime > userIdleSeconds && !c.Idle { | |
6988a057 JH |
317 | c.Idle = true |
318 | ||
a2ef262a JH |
319 | c.flagsMU.Lock() |
320 | c.Flags.Set(UserFlagAway, 1) | |
321 | c.flagsMU.Unlock() | |
6988a057 | 322 | c.sendAll( |
d005ef04 | 323 | TranNotifyChangeUser, |
a2ef262a JH |
324 | NewField(FieldUserID, c.ID[:]), |
325 | NewField(FieldUserFlags, c.Flags[:]), | |
d005ef04 JH |
326 | NewField(FieldUserName, c.UserName), |
327 | NewField(FieldUserIconID, c.Icon), | |
6988a057 JH |
328 | ) |
329 | } | |
330 | } | |
331 | s.mux.Unlock() | |
332 | } | |
333 | } | |
334 | ||
46862572 JH |
335 | func (s *Server) writeBanList() error { |
336 | s.banListMU.Lock() | |
337 | defer s.banListMU.Unlock() | |
338 | ||
339 | out, err := yaml.Marshal(s.banList) | |
340 | if err != nil { | |
341 | return err | |
342 | } | |
7152b7e5 | 343 | err = os.WriteFile( |
46862572 JH |
344 | filepath.Join(s.ConfigDir, "Banlist.yaml"), |
345 | out, | |
346 | 0666, | |
347 | ) | |
348 | return err | |
349 | } | |
350 | ||
6988a057 | 351 | func (s *Server) writeThreadedNews() error { |
8eb43f95 JH |
352 | s.threadedNewsMux.Lock() |
353 | defer s.threadedNewsMux.Unlock() | |
6988a057 JH |
354 | |
355 | out, err := yaml.Marshal(s.ThreadedNews) | |
356 | if err != nil { | |
357 | return err | |
358 | } | |
8eb43f95 | 359 | err = s.FS.WriteFile( |
f22acf38 | 360 | filepath.Join(s.ConfigDir, "ThreadedNews.yaml"), |
6988a057 JH |
361 | out, |
362 | 0666, | |
363 | ) | |
364 | return err | |
365 | } | |
366 | ||
67db911d | 367 | func (s *Server) NewClientConn(conn io.ReadWriteCloser, remoteAddr string) *ClientConn { |
6988a057 JH |
368 | s.mux.Lock() |
369 | defer s.mux.Unlock() | |
370 | ||
371 | clientConn := &ClientConn{ | |
a2ef262a | 372 | Icon: []byte{0, 0}, // TODO: make array type |
6988a057 JH |
373 | Connection: conn, |
374 | Server: s, | |
d4c152a4 | 375 | RemoteAddr: remoteAddr, |
180d6544 JH |
376 | transfers: map[int]map[[4]byte]*FileTransfer{ |
377 | FileDownload: {}, | |
378 | FileUpload: {}, | |
379 | FolderDownload: {}, | |
380 | FolderUpload: {}, | |
381 | bannerDownload: {}, | |
382 | }, | |
df1ade54 JH |
383 | } |
384 | ||
a2ef262a | 385 | s.nextClientID.Add(1) |
6988a057 | 386 | |
a2ef262a JH |
387 | binary.BigEndian.PutUint16(clientConn.ID[:], uint16(s.nextClientID.Load())) |
388 | s.Clients[clientConn.ID] = clientConn | |
6988a057 JH |
389 | |
390 | return clientConn | |
391 | } | |
392 | ||
393 | // NewUser creates a new user account entry in the server map and config file | |
187d6dc5 | 394 | func (s *Server) NewUser(login, name, password string, access accessBitmap) error { |
6988a057 JH |
395 | s.mux.Lock() |
396 | defer s.mux.Unlock() | |
397 | ||
a2ef262a | 398 | account := NewAccount(login, name, password, access) |
180d6544 JH |
399 | |
400 | // Create account file, returning an error if one already exists. | |
401 | file, err := os.OpenFile( | |
402 | filepath.Join(s.ConfigDir, "Users", path.Join("/", login)+".yaml"), | |
a2ef262a | 403 | os.O_CREATE|os.O_EXCL|os.O_WRONLY, 0644, |
180d6544 JH |
404 | ) |
405 | if err != nil { | |
a2ef262a | 406 | return fmt.Errorf("error creating account file: %w", err) |
180d6544 JH |
407 | } |
408 | defer file.Close() | |
409 | ||
a2ef262a JH |
410 | b, err := yaml.Marshal(account) |
411 | if err != nil { | |
412 | return err | |
413 | } | |
414 | ||
415 | _, err = file.Write(b) | |
180d6544 JH |
416 | if err != nil { |
417 | return fmt.Errorf("error writing account file: %w", err) | |
418 | } | |
419 | ||
a2ef262a | 420 | s.Accounts[login] = account |
6988a057 | 421 | |
180d6544 | 422 | return nil |
6988a057 JH |
423 | } |
424 | ||
187d6dc5 | 425 | func (s *Server) UpdateUser(login, newLogin, name, password string, access accessBitmap) error { |
d2810ae9 JH |
426 | s.mux.Lock() |
427 | defer s.mux.Unlock() | |
428 | ||
a2ef262a | 429 | // If the login has changed, rename the account file. |
d2810ae9 | 430 | if login != newLogin { |
a2ef262a JH |
431 | err := os.Rename( |
432 | filepath.Join(s.ConfigDir, "Users", path.Join("/", login)+".yaml"), | |
433 | filepath.Join(s.ConfigDir, "Users", path.Join("/", newLogin)+".yaml"), | |
434 | ) | |
d2810ae9 | 435 | if err != nil { |
a2ef262a | 436 | return fmt.Errorf("error renaming account file: %w", err) |
d2810ae9 JH |
437 | } |
438 | s.Accounts[newLogin] = s.Accounts[login] | |
b8b0a6c9 | 439 | s.Accounts[newLogin].Login = newLogin |
d2810ae9 JH |
440 | delete(s.Accounts, login) |
441 | } | |
442 | ||
443 | account := s.Accounts[newLogin] | |
187d6dc5 | 444 | account.Access = access |
d2810ae9 JH |
445 | account.Name = name |
446 | account.Password = password | |
447 | ||
448 | out, err := yaml.Marshal(&account) | |
449 | if err != nil { | |
450 | return err | |
451 | } | |
f22acf38 JH |
452 | |
453 | if err := os.WriteFile(filepath.Join(s.ConfigDir, "Users", newLogin+".yaml"), out, 0666); err != nil { | |
a2ef262a | 454 | return fmt.Errorf("error writing account file: %w", err) |
d2810ae9 JH |
455 | } |
456 | ||
457 | return nil | |
458 | } | |
459 | ||
6988a057 JH |
460 | // DeleteUser deletes the user account |
461 | func (s *Server) DeleteUser(login string) error { | |
462 | s.mux.Lock() | |
463 | defer s.mux.Unlock() | |
464 | ||
180d6544 JH |
465 | err := s.FS.Remove(filepath.Join(s.ConfigDir, "Users", path.Join("/", login)+".yaml")) |
466 | if err != nil { | |
467 | return err | |
468 | } | |
469 | ||
6988a057 JH |
470 | delete(s.Accounts, login) |
471 | ||
180d6544 | 472 | return nil |
6988a057 JH |
473 | } |
474 | ||
475 | func (s *Server) connectedUsers() []Field { | |
a2ef262a JH |
476 | //s.mux.Lock() |
477 | //defer s.mux.Unlock() | |
6988a057 JH |
478 | |
479 | var connectedUsers []Field | |
c7e932c0 | 480 | for _, c := range sortedClients(s.Clients) { |
9cf66aea | 481 | b, err := io.ReadAll(&User{ |
a2ef262a | 482 | ID: c.ID, |
a7216f67 | 483 | Icon: c.Icon, |
a2ef262a | 484 | Flags: c.Flags[:], |
72dd37f1 | 485 | Name: string(c.UserName), |
9cf66aea JH |
486 | }) |
487 | if err != nil { | |
488 | return nil | |
6988a057 | 489 | } |
9cf66aea | 490 | connectedUsers = append(connectedUsers, NewField(FieldUsernameWithInfo, b)) |
6988a057 JH |
491 | } |
492 | return connectedUsers | |
493 | } | |
494 | ||
a2ef262a JH |
495 | // loadFromYAMLFile loads data from a YAML file into the provided data structure. |
496 | func loadFromYAMLFile(path string, data interface{}) error { | |
46862572 JH |
497 | fh, err := os.Open(path) |
498 | if err != nil { | |
499 | return err | |
500 | } | |
a2ef262a | 501 | defer fh.Close() |
46862572 | 502 | |
6988a057 | 503 | decoder := yaml.NewDecoder(fh) |
a2ef262a | 504 | return decoder.Decode(data) |
6988a057 JH |
505 | } |
506 | ||
507 | // loadAccounts loads account data from disk | |
508 | func (s *Server) loadAccounts(userDir string) error { | |
f22acf38 | 509 | matches, err := filepath.Glob(filepath.Join(userDir, "*.yaml")) |
6988a057 JH |
510 | if err != nil { |
511 | return err | |
512 | } | |
513 | ||
514 | if len(matches) == 0 { | |
a2ef262a | 515 | return fmt.Errorf("no accounts found in directory: %s", userDir) |
6988a057 JH |
516 | } |
517 | ||
518 | for _, file := range matches { | |
a2ef262a JH |
519 | var account Account |
520 | if err = loadFromYAMLFile(file, &account); err != nil { | |
180d6544 | 521 | return fmt.Errorf("error loading account %s: %w", file, err) |
6988a057 JH |
522 | } |
523 | ||
524 | s.Accounts[account.Login] = &account | |
525 | } | |
526 | return nil | |
527 | } | |
528 | ||
529 | func (s *Server) loadConfig(path string) error { | |
b196a50a | 530 | fh, err := s.FS.Open(path) |
6988a057 JH |
531 | if err != nil { |
532 | return err | |
533 | } | |
534 | ||
535 | decoder := yaml.NewDecoder(fh) | |
6988a057 JH |
536 | err = decoder.Decode(s.Config) |
537 | if err != nil { | |
538 | return err | |
539 | } | |
041c2df6 JH |
540 | |
541 | validate := validator.New() | |
542 | err = validate.Struct(s.Config) | |
543 | if err != nil { | |
544 | return err | |
545 | } | |
6988a057 JH |
546 | return nil |
547 | } | |
548 | ||
a2ef262a JH |
549 | func sendBanMessage(rwc io.Writer, message string) { |
550 | t := NewTransaction( | |
551 | TranServerMsg, | |
552 | [2]byte{0, 0}, | |
553 | NewField(FieldData, []byte(message)), | |
554 | NewField(FieldChatOptions, []byte{0, 0}), | |
555 | ) | |
556 | _, _ = io.Copy(rwc, &t) | |
557 | time.Sleep(1 * time.Second) | |
558 | } | |
559 | ||
6988a057 | 560 | // handleNewConnection takes a new net.Conn and performs the initial login sequence |
3178ae58 | 561 | func (s *Server) handleNewConnection(ctx context.Context, rwc io.ReadWriteCloser, remoteAddr string) error { |
c4208f86 JH |
562 | defer dontPanic(s.Logger) |
563 | ||
a2ef262a JH |
564 | // Check if remoteAddr is present in the ban list |
565 | ipAddr := strings.Split(remoteAddr, ":")[0] | |
566 | if banUntil, ok := s.banList[ipAddr]; ok { | |
567 | // permaban | |
568 | if banUntil == nil { | |
569 | sendBanMessage(rwc, "You are permanently banned on this server") | |
570 | s.Logger.Debug("Disconnecting permanently banned IP", "remoteAddr", ipAddr) | |
571 | return nil | |
572 | } | |
573 | ||
574 | // temporary ban | |
575 | if time.Now().Before(*banUntil) { | |
576 | sendBanMessage(rwc, "You are temporarily banned on this server") | |
577 | s.Logger.Debug("Disconnecting temporarily banned IP", "remoteAddr", ipAddr) | |
578 | return nil | |
579 | } | |
580 | } | |
581 | ||
582 | if err := performHandshake(rwc); err != nil { | |
583 | return fmt.Errorf("error performing handshake: %w", err) | |
6988a057 JH |
584 | } |
585 | ||
3178ae58 JH |
586 | // Create a new scanner for parsing incoming bytes into transaction tokens |
587 | scanner := bufio.NewScanner(rwc) | |
588 | scanner.Split(transactionScanner) | |
589 | ||
590 | scanner.Scan() | |
6988a057 | 591 | |
f4cdaddc JH |
592 | // Make a new []byte slice and copy the scanner bytes to it. This is critical to avoid a data race as the |
593 | // scanner re-uses the buffer for subsequent scans. | |
594 | buf := make([]byte, len(scanner.Bytes())) | |
595 | copy(buf, scanner.Bytes()) | |
596 | ||
854a92fc | 597 | var clientLogin Transaction |
f4cdaddc | 598 | if _, err := clientLogin.Write(buf); err != nil { |
a2ef262a | 599 | return fmt.Errorf("error writing login transaction: %w", err) |
46862572 | 600 | } |
e9c043c0 JH |
601 | |
602 | c := s.NewClientConn(rwc, remoteAddr) | |
6988a057 | 603 | defer c.Disconnect() |
6988a057 | 604 | |
d005ef04 JH |
605 | encodedPassword := clientLogin.GetField(FieldUserPassword).Data |
606 | c.Version = clientLogin.GetField(FieldVersion).Data | |
6988a057 | 607 | |
a2ef262a | 608 | login := string(encodeString(clientLogin.GetField(FieldUserLogin).Data)) |
6988a057 JH |
609 | if login == "" { |
610 | login = GuestAccount | |
611 | } | |
612 | ||
0da28a1f JH |
613 | c.logger = s.Logger.With("remoteAddr", remoteAddr, "login", login) |
614 | ||
6988a057 JH |
615 | // If authentication fails, send error reply and close connection |
616 | if !c.Authenticate(login, encodedPassword) { | |
a2ef262a | 617 | t := c.NewErrReply(&clientLogin, "Incorrect login.")[0] |
95159e55 JH |
618 | |
619 | _, err := io.Copy(rwc, &t) | |
72dd37f1 JH |
620 | if err != nil { |
621 | return err | |
622 | } | |
0da28a1f | 623 | |
a6216dd8 | 624 | c.logger.Info("Login failed", "clientVersion", fmt.Sprintf("%x", c.Version)) |
0da28a1f JH |
625 | |
626 | return nil | |
6988a057 JH |
627 | } |
628 | ||
d005ef04 JH |
629 | if clientLogin.GetField(FieldUserIconID).Data != nil { |
630 | c.Icon = clientLogin.GetField(FieldUserIconID).Data | |
59097464 JH |
631 | } |
632 | ||
a2ef262a | 633 | c.Lock() |
59097464 | 634 | c.Account = c.Server.Accounts[login] |
a2ef262a | 635 | c.Unlock() |
59097464 | 636 | |
d005ef04 | 637 | if clientLogin.GetField(FieldUserName).Data != nil { |
ea5d8c51 | 638 | if c.Authorize(accessAnyName) { |
d005ef04 | 639 | c.UserName = clientLogin.GetField(FieldUserName).Data |
ea5d8c51 JH |
640 | } else { |
641 | c.UserName = []byte(c.Account.Name) | |
642 | } | |
6988a057 JH |
643 | } |
644 | ||
6988a057 | 645 | if c.Authorize(accessDisconUser) { |
a2ef262a | 646 | c.Flags.Set(UserFlagAdmin, 1) |
6988a057 JH |
647 | } |
648 | ||
854a92fc | 649 | s.outbox <- c.NewReply(&clientLogin, |
d005ef04 JH |
650 | NewField(FieldVersion, []byte{0x00, 0xbe}), |
651 | NewField(FieldCommunityBannerID, []byte{0, 0}), | |
652 | NewField(FieldServerName, []byte(s.Config.Name)), | |
6988a057 JH |
653 | ) |
654 | ||
655 | // Send user access privs so client UI knows how to behave | |
a2ef262a | 656 | c.Server.outbox <- NewTransaction(TranUserAccess, c.ID, NewField(FieldUserAccess, c.Account.Access[:])) |
6988a057 | 657 | |
a322be02 | 658 | // Accounts with accessNoAgreement do not receive the server agreement on login. The behavior is different between |
d005ef04 JH |
659 | // client versions. For 1.2.3 client, we do not send TranShowAgreement. For other client versions, we send |
660 | // TranShowAgreement but with the NoServerAgreement field set to 1. | |
688c86d2 | 661 | if c.Authorize(accessNoAgreement) { |
a322be02 JH |
662 | // If client version is nil, then the client uses the 1.2.3 login behavior |
663 | if c.Version != nil { | |
a2ef262a | 664 | c.Server.outbox <- NewTransaction(TranShowAgreement, c.ID, NewField(FieldNoServerAgreement, []byte{1})) |
a322be02 | 665 | } |
688c86d2 | 666 | } else { |
a2ef262a | 667 | c.Server.outbox <- NewTransaction(TranShowAgreement, c.ID, NewField(FieldData, s.Agreement)) |
688c86d2 | 668 | } |
6988a057 | 669 | |
2f8472fa JH |
670 | // If the client has provided a username as part of the login, we can infer that it is using the 1.2.3 login |
671 | // flow and not the 1.5+ flow. | |
672 | if len(c.UserName) != 0 { | |
673 | // Add the client username to the logger. For 1.5+ clients, we don't have this information yet as it comes as | |
d005ef04 | 674 | // part of TranAgreed |
95159e55 | 675 | c.logger = c.logger.With("Name", string(c.UserName)) |
a6216dd8 | 676 | c.logger.Info("Login successful", "clientVersion", "Not sent (probably 1.2.3)") |
2f8472fa JH |
677 | |
678 | // Notify other clients on the server that the new user has logged in. For 1.5+ clients we don't have this | |
d005ef04 | 679 | // information yet, so we do it in TranAgreed instead |
2f8472fa | 680 | for _, t := range c.notifyOthers( |
a2ef262a JH |
681 | NewTransaction( |
682 | TranNotifyChangeUser, [2]byte{0, 0}, | |
d005ef04 | 683 | NewField(FieldUserName, c.UserName), |
a2ef262a | 684 | NewField(FieldUserID, c.ID[:]), |
d005ef04 | 685 | NewField(FieldUserIconID, c.Icon), |
a2ef262a | 686 | NewField(FieldUserFlags, c.Flags[:]), |
2f8472fa JH |
687 | ), |
688 | ) { | |
689 | c.Server.outbox <- t | |
690 | } | |
6988a057 | 691 | } |
bd1ce113 | 692 | |
a2ef262a | 693 | c.Server.mux.Lock() |
00913df3 JH |
694 | c.Server.Stats.ConnectionCounter += 1 |
695 | if len(s.Clients) > c.Server.Stats.ConnectionPeak { | |
696 | c.Server.Stats.ConnectionPeak = len(s.Clients) | |
697 | } | |
a2ef262a | 698 | c.Server.mux.Unlock() |
6988a057 | 699 | |
3178ae58 JH |
700 | // Scan for new transactions and handle them as they come in. |
701 | for scanner.Scan() { | |
a2ef262a | 702 | // Copy the scanner bytes to a new slice to it to avoid a data race when the scanner re-uses the buffer. |
3178ae58 JH |
703 | buf := make([]byte, len(scanner.Bytes())) |
704 | copy(buf, scanner.Bytes()) | |
6988a057 | 705 | |
854a92fc JH |
706 | var t Transaction |
707 | if _, err := t.Write(buf); err != nil { | |
708 | return err | |
6988a057 | 709 | } |
854a92fc | 710 | |
a2ef262a | 711 | c.handleTransaction(t) |
6988a057 | 712 | } |
3178ae58 | 713 | return nil |
6988a057 JH |
714 | } |
715 | ||
a2ef262a | 716 | func (s *Server) NewPrivateChat(cc *ClientConn) [4]byte { |
c1c44744 JH |
717 | s.PrivateChatsMu.Lock() |
718 | defer s.PrivateChatsMu.Unlock() | |
6988a057 | 719 | |
a2ef262a JH |
720 | var randID [4]byte |
721 | _, _ = rand.Read(randID[:]) | |
f8e4cd54 | 722 | |
a2ef262a JH |
723 | s.PrivateChats[randID] = &PrivateChat{ |
724 | ClientConn: make(map[[2]byte]*ClientConn), | |
6988a057 | 725 | } |
a2ef262a | 726 | s.PrivateChats[randID].ClientConn[cc.ID] = cc |
6988a057 JH |
727 | |
728 | return randID | |
729 | } | |
730 | ||
731 | const dlFldrActionSendFile = 1 | |
732 | const dlFldrActionResumeFile = 2 | |
733 | const dlFldrActionNextFile = 3 | |
734 | ||
85767504 | 735 | // handleFileTransfer receives a client net.Conn from the file transfer server, performs the requested transfer type, then closes the connection |
7cd900d6 | 736 | func (s *Server) handleFileTransfer(ctx context.Context, rwc io.ReadWriter) error { |
37a954c8 | 737 | defer dontPanic(s.Logger) |
0a92e50b | 738 | |
a2ef262a | 739 | // The first 16 bytes contain the file transfer. |
df2735b2 | 740 | var t transfer |
a2ef262a JH |
741 | if _, err := io.CopyN(&t, rwc, 16); err != nil { |
742 | return fmt.Errorf("error reading file transfer: %w", err) | |
6988a057 JH |
743 | } |
744 | ||
0a92e50b JH |
745 | defer func() { |
746 | s.mux.Lock() | |
df1ade54 | 747 | delete(s.fileTransfers, t.ReferenceNumber) |
0a92e50b | 748 | s.mux.Unlock() |
df1ade54 | 749 | |
94742e2f JH |
750 | // Wait a few seconds before closing the connection: this is a workaround for problems |
751 | // observed with Windows clients where the client must initiate close of the TCP connection before | |
752 | // the server does. This is gross and seems unnecessary. TODO: Revisit? | |
753 | time.Sleep(3 * time.Second) | |
0a92e50b | 754 | }() |
6988a057 | 755 | |
0a92e50b | 756 | s.mux.Lock() |
df1ade54 | 757 | fileTransfer, ok := s.fileTransfers[t.ReferenceNumber] |
0a92e50b JH |
758 | s.mux.Unlock() |
759 | if !ok { | |
760 | return errors.New("invalid transaction ID") | |
761 | } | |
16a4ad70 | 762 | |
df1ade54 JH |
763 | defer func() { |
764 | fileTransfer.ClientConn.transfersMU.Lock() | |
765 | delete(fileTransfer.ClientConn.transfers[fileTransfer.Type], t.ReferenceNumber) | |
766 | fileTransfer.ClientConn.transfersMU.Unlock() | |
767 | }() | |
768 | ||
7cd900d6 JH |
769 | rLogger := s.Logger.With( |
770 | "remoteAddr", ctx.Value(contextKeyReq).(requestCtx).remoteAddr, | |
df1ade54 | 771 | "login", fileTransfer.ClientConn.Account.Login, |
95159e55 | 772 | "Name", string(fileTransfer.ClientConn.UserName), |
7cd900d6 JH |
773 | ) |
774 | ||
df1ade54 JH |
775 | fullPath, err := readPath(s.Config.FileRoot, fileTransfer.FilePath, fileTransfer.FileName) |
776 | if err != nil { | |
777 | return err | |
778 | } | |
779 | ||
6988a057 | 780 | switch fileTransfer.Type { |
9067f234 | 781 | case bannerDownload: |
5cc444c8 JH |
782 | if _, err := io.Copy(rwc, bytes.NewBuffer(s.banner)); err != nil { |
783 | return fmt.Errorf("error sending banner: %w", err) | |
9067f234 | 784 | } |
6988a057 | 785 | case FileDownload: |
23411fc2 | 786 | s.Stats.DownloadCounter += 1 |
00913df3 | 787 | s.Stats.DownloadsInProgress += 1 |
94742e2f JH |
788 | defer func() { |
789 | s.Stats.DownloadsInProgress -= 1 | |
790 | }() | |
23411fc2 | 791 | |
a2ef262a | 792 | err = DownloadHandler(rwc, fullPath, fileTransfer, s.FS, rLogger, true) |
6988a057 | 793 | if err != nil { |
a2ef262a | 794 | return fmt.Errorf("file download error: %w", err) |
7cd900d6 JH |
795 | } |
796 | ||
6988a057 | 797 | case FileUpload: |
23411fc2 | 798 | s.Stats.UploadCounter += 1 |
00913df3 JH |
799 | s.Stats.UploadsInProgress += 1 |
800 | defer func() { s.Stats.UploadsInProgress -= 1 }() | |
23411fc2 | 801 | |
a2ef262a | 802 | err = UploadHandler(rwc, fullPath, fileTransfer, s.FS, rLogger, s.Config.PreserveResourceForks) |
7cd900d6 | 803 | if err != nil { |
a2ef262a | 804 | return fmt.Errorf("file upload error: %w", err) |
6988a057 | 805 | } |
85767504 | 806 | |
6988a057 | 807 | case FolderDownload: |
00913df3 JH |
808 | s.Stats.DownloadCounter += 1 |
809 | s.Stats.DownloadsInProgress += 1 | |
810 | defer func() { s.Stats.DownloadsInProgress -= 1 }() | |
811 | ||
a2ef262a | 812 | err = DownloadFolderHandler(rwc, fullPath, fileTransfer, s.FS, rLogger, s.Config.PreserveResourceForks) |
67db911d | 813 | if err != nil { |
a2ef262a | 814 | return fmt.Errorf("file upload error: %w", err) |
67db911d JH |
815 | } |
816 | ||
6988a057 | 817 | case FolderUpload: |
00913df3 JH |
818 | s.Stats.UploadCounter += 1 |
819 | s.Stats.UploadsInProgress += 1 | |
820 | defer func() { s.Stats.UploadsInProgress -= 1 }() | |
a6216dd8 | 821 | rLogger.Info( |
6988a057 | 822 | "Folder upload started", |
df1ade54 JH |
823 | "dstPath", fullPath, |
824 | "TransferSize", binary.BigEndian.Uint32(fileTransfer.TransferSize), | |
6988a057 JH |
825 | "FolderItemCount", fileTransfer.FolderItemCount, |
826 | ) | |
827 | ||
a2ef262a JH |
828 | err = UploadFolderHandler(rwc, fullPath, fileTransfer, s.FS, rLogger, s.Config.PreserveResourceForks) |
829 | if err != nil { | |
830 | return fmt.Errorf("file upload error: %w", err) | |
6988a057 | 831 | } |
6988a057 | 832 | } |
6988a057 JH |
833 | return nil |
834 | } |