mirror of
https://github.com/ergochat/ergo.git
synced 2024-12-23 03:02:48 +01:00
commit
2340464265
@ -1,47 +1,12 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"code.google.com/p/go.crypto/bcrypt"
|
|
||||||
"database/sql"
|
|
||||||
"encoding/base64"
|
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/jlatt/ergonomadic/irc"
|
"github.com/jlatt/ergonomadic/irc"
|
||||||
_ "github.com/mattn/go-sqlite3"
|
|
||||||
"log"
|
"log"
|
||||||
"os"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func genPasswd(passwd string) {
|
|
||||||
crypted, err := bcrypt.GenerateFromPassword([]byte(passwd), bcrypt.MinCost)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
encoded := base64.StdEncoding.EncodeToString(crypted)
|
|
||||||
fmt.Println(encoded)
|
|
||||||
}
|
|
||||||
|
|
||||||
func initDB(config *irc.Config) {
|
|
||||||
os.Remove(config.Database())
|
|
||||||
|
|
||||||
db, err := sql.Open("sqlite3", config.Database())
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
defer db.Close()
|
|
||||||
|
|
||||||
_, err = db.Exec(`
|
|
||||||
CREATE TABLE channel (
|
|
||||||
name TEXT NOT NULL UNIQUE,
|
|
||||||
flags TEXT NOT NULL,
|
|
||||||
key TEXT NOT NULL,
|
|
||||||
topic TEXT NOT NULL,
|
|
||||||
user_limit INTEGER DEFAULT 0)`)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
conf := flag.String("conf", "ergonomadic.json", "ergonomadic config file")
|
conf := flag.String("conf", "ergonomadic.json", "ergonomadic config file")
|
||||||
initdb := flag.Bool("initdb", false, "initialize database")
|
initdb := flag.Bool("initdb", false, "initialize database")
|
||||||
@ -49,7 +14,11 @@ func main() {
|
|||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
if *passwd != "" {
|
if *passwd != "" {
|
||||||
genPasswd(*passwd)
|
encoded, err := irc.GenerateEncodedPassword(*passwd)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
fmt.Println(encoded)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -59,7 +28,8 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if *initdb {
|
if *initdb {
|
||||||
initDB(config)
|
irc.InitDB(config.Database())
|
||||||
|
log.Println("database initialized: " + config.Database())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,7 +1,6 @@
|
|||||||
package irc
|
package irc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"code.google.com/p/go.crypto/bcrypt"
|
|
||||||
"code.google.com/p/go.text/unicode/norm"
|
"code.google.com/p/go.text/unicode/norm"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
@ -214,7 +213,7 @@ func (cmd *PassCommand) CheckPassword() {
|
|||||||
if cmd.hash == nil {
|
if cmd.hash == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
cmd.err = bcrypt.CompareHashAndPassword(cmd.hash, cmd.password)
|
cmd.err = ComparePassword(cmd.hash, cmd.password)
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewPassCommand(args []string) (editableCommand, error) {
|
func NewPassCommand(args []string) (editableCommand, error) {
|
||||||
|
@ -1,7 +1,6 @@
|
|||||||
package irc
|
package irc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/base64"
|
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
@ -9,10 +8,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func decodePassword(password string) []byte {
|
func decodePassword(password string) []byte {
|
||||||
if password == "" {
|
bytes, err := DecodePassword(password)
|
||||||
return nil
|
|
||||||
}
|
|
||||||
bytes, err := base64.StdEncoding.DecodeString(password)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
|
32
irc/database.go
Normal file
32
irc/database.go
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
package irc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
_ "github.com/mattn/go-sqlite3"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
func InitDB(path string) {
|
||||||
|
os.Remove(path)
|
||||||
|
db := OpenDB(path)
|
||||||
|
defer db.Close()
|
||||||
|
_, err := db.Exec(`
|
||||||
|
CREATE TABLE channel (
|
||||||
|
name TEXT NOT NULL UNIQUE,
|
||||||
|
flags TEXT NOT NULL,
|
||||||
|
key TEXT NOT NULL,
|
||||||
|
topic TEXT NOT NULL,
|
||||||
|
user_limit INTEGER DEFAULT 0)`)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func OpenDB(path string) *sql.DB {
|
||||||
|
db, err := sql.Open("sqlite3", path)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
return db
|
||||||
|
}
|
37
irc/password.go
Normal file
37
irc/password.go
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
package irc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"code.google.com/p/go.crypto/bcrypt"
|
||||||
|
"encoding/base64"
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
EmptyPasswordError = errors.New("empty password")
|
||||||
|
)
|
||||||
|
|
||||||
|
func GenerateEncodedPassword(passwd string) (encoded string, err error) {
|
||||||
|
if passwd == "" {
|
||||||
|
err = EmptyPasswordError
|
||||||
|
return
|
||||||
|
}
|
||||||
|
bcrypted, err := bcrypt.GenerateFromPassword([]byte(passwd), bcrypt.MinCost)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
encoded = base64.StdEncoding.EncodeToString(bcrypted)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func DecodePassword(encoded string) (decoded []byte, err error) {
|
||||||
|
if encoded == "" {
|
||||||
|
err = EmptyPasswordError
|
||||||
|
return
|
||||||
|
}
|
||||||
|
decoded, err = base64.StdEncoding.DecodeString(encoded)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func ComparePassword(hash, password []byte) error {
|
||||||
|
return bcrypt.CompareHashAndPassword(hash, password)
|
||||||
|
}
|
@ -7,7 +7,6 @@ import (
|
|||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
"fmt"
|
"fmt"
|
||||||
_ "github.com/mattn/go-sqlite3"
|
|
||||||
"log"
|
"log"
|
||||||
"net"
|
"net"
|
||||||
"os"
|
"os"
|
||||||
@ -37,17 +36,12 @@ type Server struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func NewServer(config *Config) *Server {
|
func NewServer(config *Config) *Server {
|
||||||
db, err := sql.Open("sqlite3", config.Database())
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
server := &Server{
|
server := &Server{
|
||||||
channels: make(ChannelNameMap),
|
channels: make(ChannelNameMap),
|
||||||
clients: make(ClientNameMap),
|
clients: make(ClientNameMap),
|
||||||
commands: make(chan Command, 16),
|
commands: make(chan Command, 16),
|
||||||
ctime: time.Now(),
|
ctime: time.Now(),
|
||||||
db: db,
|
db: OpenDB(config.Database()),
|
||||||
idle: make(chan *Client, 16),
|
idle: make(chan *Client, 16),
|
||||||
motdFile: config.MOTD,
|
motdFile: config.MOTD,
|
||||||
name: config.Name,
|
name: config.Name,
|
||||||
|
Loading…
Reference in New Issue
Block a user