qowevisa/tricrypt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Some changes to users

Browse Source
This commit is contained in:
qowevisa 2024-06-07 12:19:37 +03:00
parent 5996975b4f
commit 666bfbea4b
2 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/server/main.go
View File

@ -96,8 +96,8 @@ func handleClient(conn net.Conn) {
}
conn.Write(answ)
} else {
idBytes := make([]byte, 4)
binary.BigEndian.PutUint32(idBytes, uint32(id))
idBytes := make([]byte, 2)
binary.BigEndian.PutUint16(idBytes, id)
answ, err := com.ServerSendClientHisID(idBytes)
if err != nil {
log.Printf("ERROR: %v\n", err)

24
cmd/server/users.go
View File

@ -11,14 +11,14 @@ var (
)
type UserCenter struct {
UsersSTOI map[string]int32
UsersITOS map[int32]string
UsersSTOI map[string]uint16
UsersITOS map[uint16]string
Mu sync.Mutex
}
func (u *UserCenter) Init() {
u.UsersSTOI = make(map[string]int32)
u.UsersITOS = make(map[int32]string)
u.UsersSTOI = make(map[string]uint16)
u.UsersITOS = make(map[uint16]string)
}
var (
@ -26,19 +26,19 @@ var (
ERROR_DONT_HAVE = errors.New("Username was not found")
)
func (u *UserCenter) AddUser(name string) (int32, error) {
func (u *UserCenter) AddUser(name string) (uint16, error) {
u.Mu.Lock()
defer u.Mu.Unlock()
_, alreadyHave := u.UsersSTOI[name]
if alreadyHave {
return -1, ERROR_ALREADY_HAVE
return 0, ERROR_ALREADY_HAVE
}
var ret int32
var ret uint16
for {
rInt := rand.Int31()
_, taken := u.UsersITOS[rInt]
_, taken := u.UsersITOS[uint16(rInt)]
if !taken {
ret = rInt
ret = uint16(rInt)
break
}
}
@ -47,15 +47,15 @@ func (u *UserCenter) AddUser(name string) (int32, error) {
return ret, nil
}
func (u *UserCenter) GetID(name string) (int32, error) {
func (u *UserCenter) GetID(name string) (uint16, error) {
id, have := u.UsersSTOI[name]
if !have {
return -1, ERROR_DONT_HAVE
return 0, ERROR_DONT_HAVE
}
return id, nil
}
func (u *UserCenter) GetName(id int32) (string, error) {
func (u *UserCenter) GetName(id uint16) (string, error) {
name, have := u.UsersITOS[id]
if !have {
return "", ERROR_DONT_HAVE