-
Notifications
You must be signed in to change notification settings - Fork 1
/
account.go
78 lines (67 loc) · 1.79 KB
/
account.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package main
import (
"crypto/sha512"
"encoding/hex"
"errors"
"github.com/gin-gonic/gin"
"github.com/jackc/pgerrcode"
"github.com/jackc/pgx/v5/pgconn"
"net/http"
)
const CreateAccount = `INSERT INTO accounts (mlid, password, mlchkid) VALUES ($1, $2, $3)`
func account(c *gin.Context) {
mlid := c.PostForm("mlid")
if mlid == "" {
cgi := GenCGIError(610, "mlid not found")
c.String(http.StatusOK, ConvertToCGI(cgi))
return
}
if !validateFriendCode(mlid[1:]) {
cgi := GenCGIError(610, "Invalid Wii Friend Code")
c.String(http.StatusOK, ConvertToCGI(cgi))
return
}
c.Header("Content-Type", "text/plain;charset=utf-8")
// Password can be any length up to 32 characters. 16 seems like a good middle ground.
password := RandStringBytesMaskImprSrc(16)
passwordByte := sha512.Sum512([]byte(password))
passwordHash := hex.EncodeToString(passwordByte[:])
// Mlchkid must be a string of 32 characters
mlchkid := RandStringBytesMaskImprSrc(32)
mlchkidByte := sha512.Sum512([]byte(mlchkid))
mlchkidHash := hex.EncodeToString(mlchkidByte[:])
_, err := pool.Exec(c.Copy(), CreateAccount, mlid[1:], passwordHash, mlchkidHash)
if err != nil {
var v *pgconn.PgError
if errors.As(err, &v) {
if pgerrcode.IsIntegrityConstraintViolation(v.Code) {
cgi := GenCGIError(211, "Duplicate registration.")
c.String(http.StatusOK, ConvertToCGI(cgi))
return
}
}
cgi := GenCGIError(410, "An error has occurred while querying the database.")
ReportError(err)
c.String(http.StatusOK, ConvertToCGI(cgi))
return
}
cgi := CGIResponse{
code: 100,
message: "Success.",
other: []KV{
{
key: "mlid",
value: mlid,
},
{
key: "passwd",
value: password,
},
{
key: "mlchkid",
value: mlchkid,
},
},
}
c.String(http.StatusOK, ConvertToCGI(cgi))
}