89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
port = flag.String("port", "26666", "Port to listen on")
|
|
logFile = flag.String("logfile", "server.log", "Log file path")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
// Open log file
|
|
file, err := os.OpenFile(*logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
|
if err != nil {
|
|
log.Fatal("Failed to open log file:", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
// Set log output to both file and console
|
|
log.SetOutput(file)
|
|
|
|
// Start listening on specified port
|
|
listener, err := net.Listen("tcp", ":"+*port)
|
|
if err != nil {
|
|
log.Fatal("Failed to start server:", err)
|
|
}
|
|
defer listener.Close()
|
|
|
|
fmt.Printf("Server listening on port %s\n", *port)
|
|
log.Printf("Server listening on port %s\n", *port)
|
|
|
|
for {
|
|
// Accept incoming connections
|
|
conn, err := listener.Accept()
|
|
if err != nil {
|
|
log.Printf("Failed to accept connection: %v", err)
|
|
continue
|
|
}
|
|
|
|
// Handle each connection in a separate goroutine
|
|
go handleConnection(conn)
|
|
}
|
|
}
|
|
|
|
func handleConnection(conn net.Conn) {
|
|
defer conn.Close()
|
|
|
|
// Get connection info
|
|
remoteAddr := conn.RemoteAddr().(*net.TCPAddr)
|
|
timestamp := time.Now().Format("2006-01-02 15:04:05")
|
|
|
|
// Print and log connection info
|
|
connInfo := fmt.Sprintf("New connection from IP: %s, Port: %d, Timestamp: %s",
|
|
remoteAddr.IP, remoteAddr.Port, timestamp)
|
|
fmt.Println(connInfo)
|
|
log.Println(connInfo)
|
|
|
|
// Create a scanner to read messages from the client
|
|
scanner := bufio.NewScanner(conn)
|
|
for scanner.Scan() {
|
|
message := scanner.Text()
|
|
msgTimestamp := time.Now().Format("2006-01-02 15:04:05")
|
|
|
|
// Print and log the received message with timestamp
|
|
logEntry := fmt.Sprintf("[%s] Received: %s", msgTimestamp, message)
|
|
fmt.Println(logEntry)
|
|
log.Println(logEntry)
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
errorMsg := fmt.Sprintf("Error reading from connection: %v", err)
|
|
fmt.Println(errorMsg)
|
|
log.Println(errorMsg)
|
|
}
|
|
|
|
// Log when client disconnects
|
|
disconnectMsg := fmt.Sprintf("Client %s:%d disconnected", remoteAddr.IP, remoteAddr.Port)
|
|
fmt.Println(disconnectMsg)
|
|
log.Println(disconnectMsg)
|
|
} |