YumeMichi/honoka-chan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify error handling codes

Browse Source 
Signed-off-by: Yuan Si <do4suki@gmail.com>
This commit is contained in:
YumeMichi
2023-04-11 13:08:06 +08:00
parent 8f1aa59e67
commit 497b3adba2
28 changed files with 195 additions and 548 deletions
Download Patch File Download Diff File Expand all files Collapse all files
handler/album.go
+11 -33
View File
@@ -19,9 +19,7 @@ import (
func AlbumSeriesAllHandler(ctx *gin.Context) {
db, err := sql.Open("sqlite3", "assets/main.db")
if err != nil {
panic(err)
}
CheckErr(err)
defer db.Close()
reqTime := time.Now().Unix()
@@ -58,34 +56,24 @@ func AlbumSeriesAllHandler(ctx *gin.Context) {
//
sql := `SELECT album_series_id FROM album_series_m`
seriesRows, err := db.Query(sql)
if err != nil {
panic(err)
}
CheckErr(err)
albumSeriesAllResp := []model.AlbumSeriesResp{}
for seriesRows.Next() {
var series int
err = seriesRows.Scan(&series)
if err != nil {
panic(err)
}
CheckErr(err)
albumSeriesAll := []model.AlbumResult{}
stmt, err := db.Prepare("SELECT unit_id,rarity FROM unit_m WHERE album_series_id = ?")
if err != nil {
panic(err)
}
CheckErr(err)
unitRows, err := stmt.Query(series)
if err != nil {
panic(err)
}
CheckErr(err)
for unitRows.Next() {
var unitId, rarity int
err = unitRows.Scan(&unitId, &rarity)
if err != nil {
panic(err)
}
CheckErr(err)
albumSeries := model.AlbumResult{
UnitID: unitId,
@@ -118,20 +106,14 @@ func AlbumSeriesAllHandler(ctx *gin.Context) {
// IsSigned
stmt, err = db.Prepare("SELECT COUNT(*) AS ct FROM unit_sign_asset_m WHERE unit_id = ?")
if err != nil {
panic(err)
}
CheckErr(err)
signRows, err := stmt.Query(unitId)
if err != nil {
panic(err)
}
CheckErr(err)
ct := 0
for signRows.Next() {
err = signRows.Scan(&ct)
if err != nil {
panic(err)
}
CheckErr(err)
}
if ct > 0 {
albumSeries.SignFlag = true
@@ -148,18 +130,14 @@ func AlbumSeriesAllHandler(ctx *gin.Context) {
})
}
rb, err := json.Marshal(albumSeriesAllResp)
if err != nil {
panic(err)
}
CheckErr(err)
resp := model.Response{
ResponseData: rb,
ReleaseInfo: []interface{}{},
StatusCode: 200,
}
respb, err := json.Marshal(resp)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(respb))
xms := encrypt.RSA_Sign_SHA1(respb, "privatekey.pem")
handler/announce.go
+1 -3
View File
@@ -73,9 +73,7 @@ func AnnounceCheckStateHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(announceResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/api.go
+5 -15
View File
@@ -148,36 +148,26 @@ func ApiHandler(ctx *gin.Context) {
err = errors.New("invalid option")
}
if err != nil {
panic(err)
}
CheckErr(err)
res, err = database.LevelDb.Get([]byte(key))
if err != nil {
panic(err)
}
CheckErr(err)
var result interface{}
err = json.Unmarshal([]byte(res), &result)
if err != nil {
panic(err)
}
CheckErr(err)
results = append(results, result)
}
// fmt.Println(results)
b, err := json.Marshal(results)
if err != nil {
panic(err)
}
CheckErr(err)
rp := model.Response{
ResponseData: b,
ReleaseInfo: []interface{}{},
StatusCode: 200,
}
b, err = json.Marshal(rp)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(b))
authorizeStr := ctx.Request.Header["Authorize"]
handler/download.go
+16 -48
View File
@@ -19,9 +19,7 @@ import (
func DownloadAdditionalHandler(ctx *gin.Context) {
db, err := sql.Open("sqlite3", "assets/main.db")
if err != nil {
panic(err)
}
CheckErr(err)
defer db.Close()
reqTime := time.Now().Unix()
@@ -63,19 +61,13 @@ func DownloadAdditionalHandler(ctx *gin.Context) {
if CdnUrl != "" {
pkgType, pkgId := downloadReq.PackageType, downloadReq.PackageID
stmt, err := db.Prepare("SELECT pkg_order,pkg_size FROM download_db WHERE pkg_type = ? AND pkg_id = ? ORDER BY pkg_order ASC")
if err != nil {
panic(err)
}
CheckErr(err)
rows, err := stmt.Query(pkgType, pkgId)
if err != nil {
panic(err)
}
CheckErr(err)
for rows.Next() {
var pkgOrder, pkgSize int
err = rows.Scan(&pkgOrder, &pkgSize)
if err != nil {
panic(err)
}
CheckErr(err)
pkgList = append(pkgList, model.AdditionalResult{
Size: pkgSize,
URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder),
@@ -89,9 +81,7 @@ func DownloadAdditionalHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(addResp)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(resp))
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -104,9 +94,7 @@ func DownloadAdditionalHandler(ctx *gin.Context) {
func DownloadBatchHandler(ctx *gin.Context) {
db, err := sql.Open("sqlite3", "assets/main.db")
if err != nil {
panic(err)
}
CheckErr(err)
defer db.Close()
reqTime := time.Now().Unix()
@@ -156,19 +144,13 @@ func DownloadBatchHandler(ctx *gin.Context) {
}
// fmt.Println(exPkgId)
stmt, err := db.Prepare("SELECT pkg_id,pkg_order,pkg_size FROM download_db WHERE pkg_type = ? " + exPkgId + " ORDER BY pkg_id ASC, pkg_order ASC")
if err != nil {
panic(err)
}
CheckErr(err)
rows, err := stmt.Query(pkgType)
if err != nil {
panic(err)
}
CheckErr(err)
for rows.Next() {
var pkgId, pkgOrder, pkgSize int
err = rows.Scan(&pkgId, &pkgOrder, &pkgSize)
if err != nil {
panic(err)
}
CheckErr(err)
pkgList = append(pkgList, model.BatchResult{
Size: pkgSize,
URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder),
@@ -182,9 +164,7 @@ func DownloadBatchHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(batchResp)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(resp))
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -197,9 +177,7 @@ func DownloadBatchHandler(ctx *gin.Context) {
func DownloadUpdateHandler(ctx *gin.Context) {
db, err := sql.Open("sqlite3", "assets/main.db")
if err != nil {
panic(err)
}
CheckErr(err)
defer db.Close()
reqTime := time.Now().Unix()
@@ -241,15 +219,11 @@ func DownloadUpdateHandler(ctx *gin.Context) {
if downloadReq.ExternalVersion != PackageVersion && CdnUrl != "" {
pkgType := 99
rows, err := db.Query("SELECT pkg_id,pkg_order,pkg_size FROM download_db WHERE pkg_type = 99 ORDER BY pkg_id ASC, pkg_order ASC")
if err != nil {
panic(err)
}
CheckErr(err)
for rows.Next() {
var pkgId, pkgOrder, pkgSize int
err = rows.Scan(&pkgId, &pkgOrder, &pkgSize)
if err != nil {
panic(err)
}
CheckErr(err)
pkgList = append(pkgList, model.UpdateResult{
Size: pkgSize,
URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder),
@@ -264,9 +238,7 @@ func DownloadUpdateHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(updateResp)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(resp))
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -327,9 +299,7 @@ func DownloadUrlHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(urlResp)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(resp))
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -378,9 +348,7 @@ func DownloadEventHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(eventResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/event.go
+1 -3
View File
@@ -78,9 +78,7 @@ func EventListHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(eventsResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/gdpr.go
+1 -3
View File
@@ -68,9 +68,7 @@ func GdprHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(gdprResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/global.go
+6
View File
@@ -14,3 +14,9 @@ func init() {
CdnUrl = config.Conf.Cdn.CdnUrl
}
}
func CheckErr(err error) {
if err != nil {
panic(err)
}
}
handler/lbonus.go
+1 -3
View File
@@ -185,9 +185,7 @@ func LBonusExecuteHandler(ctx *gin.Context) {
}
resp, err := json.Marshal(res)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/live.go
+25 -75
View File
@@ -72,9 +72,7 @@ func PlayLiveHandler(ctx *gin.Context) {
playReq := model.PlayReq{}
err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &playReq)
if err != nil {
panic(err)
}
CheckErr(err)
authorizeStr := ctx.Request.Header["Authorize"]
authToken, err := utils.GetAuthorizeToken(authorizeStr)
@@ -106,34 +104,26 @@ func PlayLiveHandler(ctx *gin.Context) {
// fmt.Println(newAuthorizeStr)
db, err := sql.Open("sqlite3", "assets/main.db")
if err != nil {
panic(err)
}
CheckErr(err)
defer db.Close()
db.SetMaxOpenConns(1)
tDifficultyId := playReq.LiveDifficultyID
difficultyId, err := strconv.Atoi(tDifficultyId)
if err != nil {
panic(err)
}
CheckErr(err)
deckId := playReq.UnitDeckID
// Song type: normal / special
// sqlite3 doesn't support FULL OUTER JOIN so use UNION ALL here.
sql := `SELECT notes_list,c_rank_score,b_rank_score,a_rank_score,s_rank_score,ac_flag,swing_flag FROM live_setting_m WHERE live_setting_id IN (SELECT live_setting_id FROM normal_live_m WHERE live_difficulty_id = ? UNION ALL SELECT live_setting_id FROM special_live_m WHERE live_difficulty_id = ?)`
rows, err := db.Query(sql, difficultyId, difficultyId)
if err != nil {
panic(err)
}
CheckErr(err)
var notes_list string
var c_rank_score, b_rank_score, a_rank_score, s_rank_score, ac_flag, swing_flag int
for rows.Next() {
err = rows.Scan(&notes_list, &c_rank_score, &b_rank_score, &a_rank_score, &s_rank_score, &ac_flag, &swing_flag)
if err != nil {
panic(err)
}
CheckErr(err)
}
// fmt.Println(len(notes_list))
@@ -141,9 +131,7 @@ func PlayLiveHandler(ctx *gin.Context) {
notes := []model.NotesList{}
err = json.Unmarshal([]byte(notes_list), &notes)
if err != nil {
panic(err)
}
CheckErr(err)
ranks := []model.RankInfo{}
ranks = append(ranks, model.RankInfo{
@@ -211,9 +199,7 @@ func PlayLiveHandler(ctx *gin.Context) {
}
m, err := json.Marshal(resp)
if err != nil {
panic(err)
}
CheckErr(err)
res := model.Response{
ResponseData: m,
@@ -222,9 +208,7 @@ func PlayLiveHandler(ctx *gin.Context) {
}
mm, err := json.Marshal(res)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(mm, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -274,9 +258,7 @@ func GameOverHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(overResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -292,9 +274,7 @@ func PlayScoreHandler(ctx *gin.Context) {
playScoreReq := model.PlayScoreReq{}
err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &playScoreReq)
if err != nil {
panic(err)
}
CheckErr(err)
authorizeStr := ctx.Request.Header["Authorize"]
authToken, err := utils.GetAuthorizeToken(authorizeStr)
@@ -326,33 +306,25 @@ func PlayScoreHandler(ctx *gin.Context) {
// fmt.Println(newAuthorizeStr)
db, err := sql.Open("sqlite3", "assets/main.db")
if err != nil {
panic(err)
}
CheckErr(err)
defer db.Close()
db.SetMaxOpenConns(1)
tDifficultyId := playScoreReq.LiveDifficultyID
difficultyId, err := strconv.Atoi(tDifficultyId)
if err != nil {
panic(err)
}
CheckErr(err)
// Song type: normal / special
// sqlite3 doesn't support FULL OUTER JOIN so use UNION ALL here.
sql := `SELECT notes_list,c_rank_score,b_rank_score,a_rank_score,s_rank_score,ac_flag,swing_flag FROM live_setting_m WHERE live_setting_id IN (SELECT live_setting_id FROM normal_live_m WHERE live_difficulty_id = ? UNION ALL SELECT live_setting_id FROM special_live_m WHERE live_difficulty_id = ?)`
rows, err := db.Query(sql, difficultyId, difficultyId)
if err != nil {
panic(err)
}
CheckErr(err)
var notes_list string
var c_rank_score, b_rank_score, a_rank_score, s_rank_score, ac_flag, swing_flag int
for rows.Next() {
err = rows.Scan(&notes_list, &c_rank_score, &b_rank_score, &a_rank_score, &s_rank_score, &ac_flag, &swing_flag)
if err != nil {
panic(err)
}
CheckErr(err)
}
// fmt.Println(len(notes_list))
@@ -360,9 +332,7 @@ func PlayScoreHandler(ctx *gin.Context) {
notes := []model.NotesList{}
err = json.Unmarshal([]byte(notes_list), &notes)
if err != nil {
panic(err)
}
CheckErr(err)
ranks := []model.RankInfo{}
ranks = append(ranks, model.RankInfo{
@@ -414,9 +384,7 @@ func PlayScoreHandler(ctx *gin.Context) {
}
m, err := json.Marshal(resp)
if err != nil {
panic(err)
}
CheckErr(err)
res := model.Response{
ResponseData: m,
@@ -425,9 +393,7 @@ func PlayScoreHandler(ctx *gin.Context) {
}
mm, err := json.Marshal(res)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(mm))
xms := encrypt.RSA_Sign_SHA1(mm, "privatekey.pem")
@@ -445,9 +411,7 @@ func PlayRewardHandler(ctx *gin.Context) {
playRewardReq := model.PlayRewardReq{}
err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &playRewardReq)
if err != nil {
panic(err)
}
CheckErr(err)
authorizeStr := ctx.Request.Header["Authorize"]
authToken, err := utils.GetAuthorizeToken(authorizeStr)
@@ -479,9 +443,7 @@ func PlayRewardHandler(ctx *gin.Context) {
// fmt.Println(newAuthorizeStr)
db, err := sql.Open("sqlite3", "assets/main.db")
if err != nil {
panic(err)
}
CheckErr(err)
defer db.Close()
db.SetMaxOpenConns(1)
@@ -491,27 +453,19 @@ func PlayRewardHandler(ctx *gin.Context) {
// sqlite3 doesn't support FULL OUTER JOIN so use UNION ALL here.
sql := `SELECT c_rank_score,b_rank_score,a_rank_score,s_rank_score,c_rank_combo,b_rank_combo,a_rank_combo,s_rank_combo,ac_flag,swing_flag FROM live_setting_m WHERE live_setting_id IN (SELECT live_setting_id FROM normal_live_m WHERE live_difficulty_id = ? UNION ALL SELECT live_setting_id FROM special_live_m WHERE live_difficulty_id = ?)`
rows, err := db.Query(sql, difficultyId, difficultyId)
if err != nil {
panic(err)
}
CheckErr(err)
var c_rank_score, b_rank_score, a_rank_score, s_rank_score, c_rank_combo, b_rank_combo, a_rank_combo, s_rank_combo, ac_flag, swing_flag int
for rows.Next() {
err = rows.Scan(&c_rank_score, &b_rank_score, &a_rank_score, &s_rank_score, &c_rank_combo, &b_rank_combo, &a_rank_combo, &s_rank_combo, &ac_flag, &swing_flag)
if err != nil {
panic(err)
}
CheckErr(err)
}
unitsList := []model.PlayRewardUnitList{}
deck, err := database.LevelDb.Get([]byte("deck_info"))
if err != nil {
panic(err)
}
CheckErr(err)
err = json.Unmarshal(deck, &unitsList)
if err != nil {
panic(err)
}
CheckErr(err)
totalScore := playRewardReq.ScoreSmile + playRewardReq.ScoreCool + playRewardReq.ScoreCute
resp := model.RewardResponseData{
@@ -662,9 +616,7 @@ func PlayRewardHandler(ctx *gin.Context) {
}
m, err := json.Marshal(resp)
if err != nil {
panic(err)
}
CheckErr(err)
res := model.Response{
ResponseData: m,
@@ -673,9 +625,7 @@ func PlayRewardHandler(ctx *gin.Context) {
}
mm, err := json.Marshal(res)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(mm))
xms := encrypt.RSA_Sign_SHA1(mm, "privatekey.pem")
handler/login.go
+14 -42
View File
@@ -20,21 +20,15 @@ func AuthKeyHandler(ctx *gin.Context) {
authReq := model.AuthKeyReq{}
err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &authReq)
if err != nil {
panic(err)
}
CheckErr(err)
dummyToken64, err := base64.StdEncoding.DecodeString(authReq.DummyToken)
if err != nil {
panic(err)
}
CheckErr(err)
dummyTokenDecrypted := encrypt.RSA_Decrypt(dummyToken64, "privatekey.pem")
// aesKey := dummyTokenDecrypted[0:16]
// data64, err := base64.StdEncoding.DecodeString(authReq.AuthData)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// dataDecrypted := utils.Sub16(encrypt.AES_CBC_Decrypt(data64, aesKey))
// fmt.Println(string(dataDecrypted))
@@ -51,22 +45,16 @@ func AuthKeyHandler(ctx *gin.Context) {
"server_token": serverToken,
}
authJson, err := json.Marshal(authData)
if err != nil {
panic(err)
}
CheckErr(err)
database.LevelDb.Put([]byte(authorizeToken), authJson)
if err != nil {
panic(err)
}
CheckErr(err)
authResp := model.AuthKeyResp{}
authResp.ResponseData.DummyToken = serverToken
authResp.ResponseData.AuthorizeToken = authorizeToken
authResp.StatusCode = 200
resp, err := json.Marshal(authResp)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(resp))
respTime := time.Now().Unix()
@@ -94,39 +82,27 @@ func LoginHandler(ctx *gin.Context) {
// authData, err := database.RedisCli.HGetAll(database.RedisCtx, authToken).Result()
// authData, err := database.LevelDb.Get([]byte(authToken))
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(authData)
// clientToken, serverToken := authData["client_token"], authData["server_token"]
// clientToken := gjson.Get(string(authData), "client_token").String()
// serverToken := gjson.Get(string(authData), "server_token").String()
// clientToken64, err := base64.RawStdEncoding.DecodeString(clientToken)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// serverToken64, err := base64.RawStdEncoding.DecodeString(serverToken)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// xmcKey := utils.SliceXor([]byte(clientToken64), []byte(serverToken64))
// aesKey := xmcKey[0:16]
loginReq := model.LoginReq{}
err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &loginReq)
if err != nil {
panic(err)
}
CheckErr(err)
// key64, err := base64.StdEncoding.DecodeString(loginReq.LoginKey)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// pass64, err := base64.StdEncoding.DecodeString(loginReq.LoginPasswd)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// keyDescrypted := utils.Sub16(encrypt.AES_CBC_Decrypt(key64, aesKey))
// fmt.Println(string(keyDescrypted))
// passDescrypted := utils.Sub16(encrypt.AES_CBC_Decrypt(pass64, aesKey))
@@ -151,9 +127,7 @@ func LoginHandler(ctx *gin.Context) {
// _, err = database.RedisCli.HSet(database.RedisCtx, "token_uid", authorizeToken, userId).Result()
err = database.LevelDb.Put([]byte(authorizeToken), []byte(sUserId))
if err != nil {
panic(err)
}
CheckErr(err)
loginResp := model.LoginResp{}
loginResp.ResponseData.AuthorizeToken = authorizeToken
@@ -162,9 +136,7 @@ func LoginHandler(ctx *gin.Context) {
loginResp.ResponseData.AdultFlag = 2
loginResp.StatusCode = 200
resp, err := json.Marshal(loginResp)
if err != nil {
panic(err)
}
CheckErr(err)
// fmt.Println(string(resp))
respTime := time.Now().Unix()
handler/multiunit.go
+2 -6
View File
@@ -40,9 +40,7 @@ func MultiUnitStartUpHandler(ctx *gin.Context) {
startReq := MultiUnitStartUpReq{}
err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &startReq)
if err != nil {
panic(err)
}
CheckErr(err)
authorizeStr := ctx.Request.Header["Authorize"]
authToken, err := utils.GetAuthorizeToken(authorizeStr)
@@ -83,9 +81,7 @@ func MultiUnitStartUpHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(startResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/notice.go
+3 -9
View File
@@ -93,9 +93,7 @@ func NoticeFriendVarietyHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(noticeResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -148,9 +146,7 @@ func NoticeFriendGreetingHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(noticeResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -204,9 +200,7 @@ func NoticeUserGreetingHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(noticeResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/payment.go
+1 -3
View File
@@ -91,9 +91,7 @@ func ProductListHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(prodReesp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/personalnotice.go
+1 -3
View File
@@ -74,9 +74,7 @@ func PersonalNoticeHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(noticeResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/private.go
+23 -69
View File
@@ -37,33 +37,25 @@ func PublicKeyHandler(ctx *gin.Context) {
func HandshakeHandler(ctx *gin.Context) {
body, err := io.ReadAll(ctx.Request.Body)
if err != nil {
panic(err)
}
CheckErr(err)
defer ctx.Request.Body.Close()
// fmt.Println(string(body))
body64, err := base64.StdEncoding.DecodeString(string(body))
if err != nil {
panic(err)
}
CheckErr(err)
decryptedBody := encrypt.RSA_Decrypt(body64, "privatekey.pem")
// fmt.Println(decryptedBody)
// fmt.Println(string(decryptedBody))
params, err := url.ParseQuery(string(decryptedBody))
if err != nil {
panic(err)
}
CheckErr(err)
randKey = params.Get("randkey")
// fmt.Println(randKey)
token := strings.ToUpper(utils.RandomStr(33))
token = fmt.Sprintf(`{"message":"ok","result":0,"token":"%s"}`, token)
encryptedToken, err := openssl.Des3ECBEncrypt([]byte(token), []byte(randKey)[0:24], openssl.PKCS7_PADDING)
if err != nil {
panic(err)
}
CheckErr(err)
encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken)
// fmt.Println(encryptedToken64)
@@ -75,22 +67,16 @@ func HandshakeHandler(ctx *gin.Context) {
func InitializeHandler(ctx *gin.Context) {
// body, err := io.ReadAll(ctx.Request.Body)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// defer ctx.Request.Body.Close()
// fmt.Println(string(body))
// body64, err := base64.StdEncoding.DecodeString(string(body))
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(string(body64))
// decryptedBody, err := openssl.Des3ECBDecrypt(body64, []byte(randKey)[0:24], openssl.PKCS7_PADDING)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(string(decryptedBody))
data := `{
@@ -129,9 +115,7 @@ func InitializeHandler(ctx *gin.Context) {
"weixin_key":""
}`
encryptedToken, err := openssl.Des3ECBEncrypt([]byte(data), []byte(randKey)[0:24], openssl.PKCS7_PADDING)
if err != nil {
panic(err)
}
CheckErr(err)
encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken)
// fmt.Println(encryptedToken64)
@@ -149,30 +133,22 @@ func GetCodeHandler(ctx *gin.Context) {
func LoginAutoHandler(ctx *gin.Context) {
// body, err := io.ReadAll(ctx.Request.Body)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// defer ctx.Request.Body.Close()
// fmt.Println(string(body))
// body64, err := base64.StdEncoding.DecodeString(string(body))
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(string(body64))
// decryptedBody, err := openssl.Des3ECBDecrypt(body64, []byte(randKey)[0:24], openssl.PKCS7_PADDING)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(string(decryptedBody))
// Unable to decrypt server data
token := `{"message":"Hello, world!"}`
encryptedToken, err := openssl.Des3ECBEncrypt([]byte(token), []byte(randKey)[0:24], openssl.PKCS7_PADDING)
if err != nil {
panic(err)
}
CheckErr(err)
encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken)
// fmt.Println(encryptedToken64)
@@ -194,22 +170,16 @@ func LoginAreaHandler(ctx *gin.Context) {
func AccountLoginHandler(ctx *gin.Context) {
// body, err := io.ReadAll(ctx.Request.Body)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// defer ctx.Request.Body.Close()
// fmt.Println(string(body))
// body64, err := base64.StdEncoding.DecodeString(string(body))
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(string(body64))
// decryptedBody, err := openssl.Des3ECBDecrypt(body64, []byte(randKey)[0:24], openssl.PKCS7_PADDING)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(string(decryptedBody))
// userid: 实际登录用的账号
@@ -241,9 +211,7 @@ func AccountLoginHandler(ctx *gin.Context) {
"userid":"2592007136"
}`
encryptedToken, err := openssl.Des3ECBEncrypt([]byte(data), []byte(randKey)[0:24], openssl.PKCS7_PADDING)
if err != nil {
panic(err)
}
CheckErr(err)
encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken)
// fmt.Println(encryptedToken64)
@@ -255,36 +223,26 @@ func AccountLoginHandler(ctx *gin.Context) {
func ReportRoleHandler(ctx *gin.Context) {
// body, err := io.ReadAll(ctx.Request.Body)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// defer ctx.Request.Body.Close()
// fmt.Println(string(body))
// body64, err := base64.StdEncoding.DecodeString(string(body))
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(string(body64))
// decryptedBody, err := openssl.Des3ECBDecrypt(body64, []byte(randKey)[0:24], openssl.PKCS7_PADDING)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(string(decryptedBody))
// decrypted, err := url.QueryUnescape(string(decryptedBody))
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// fmt.Println(decrypted)
// Unable to decrypt server data
token := `{"message":"Hello, world!"}`
encryptedToken, err := openssl.Des3ECBEncrypt([]byte(token), []byte(randKey)[0:24], openssl.PKCS7_PADDING)
if err != nil {
panic(err)
}
CheckErr(err)
encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken)
// fmt.Println(encryptedToken64)
@@ -302,9 +260,7 @@ func GetProductListHandler(ctx *gin.Context) {
func ReportLog(ctx *gin.Context) {
// body, err := io.ReadAll(ctx.Request.Body)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// defer ctx.Request.Body.Close()
// fmt.Println(string(body))
ctx.Header("Content-Type", "text/html;charset=utf-8")
@@ -313,9 +269,7 @@ func ReportLog(ctx *gin.Context) {
func ReportApp(ctx *gin.Context) {
// body, err := io.ReadAll(ctx.Request.Body)
// if err != nil {
// panic(err)
// }
// CheckErr(err)
// defer ctx.Request.Body.Close()
// fmt.Println(string(body))
resp := `{ "code": 0, "msg": "", "data": { "needReport": 0 } }`
handler/scenario.go
+2 -6
View File
@@ -40,9 +40,7 @@ func ScenarioStartupHandler(ctx *gin.Context) {
startReq := ScenarioReq{}
err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &startReq)
if err != nil {
panic(err)
}
CheckErr(err)
authorizeStr := ctx.Request.Header["Authorize"]
authToken, err := utils.GetAuthorizeToken(authorizeStr)
@@ -83,9 +81,7 @@ func ScenarioStartupHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(startResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/subscenario.go
+2 -6
View File
@@ -40,9 +40,7 @@ func SubScenarioStartupHandler(ctx *gin.Context) {
startReq := SubScenarioReq{}
err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &startReq)
if err != nil {
panic(err)
}
CheckErr(err)
authorizeStr := ctx.Request.Header["Authorize"]
authToken, err := utils.GetAuthorizeToken(authorizeStr)
@@ -83,9 +81,7 @@ func SubScenarioStartupHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(startResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/tos.go
+1 -3
View File
@@ -69,9 +69,7 @@ func TosCheckHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(tosResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/unit.go
+1 -3
View File
@@ -58,9 +58,7 @@ func SetDisplayRankHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(dispResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/user.go
+2 -6
View File
@@ -58,9 +58,7 @@ func SetNotificationTokenHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(notifResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
@@ -109,9 +107,7 @@ func ChangeNaviHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(notifResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)
handler/userinfo.go
+1 -3
View File
@@ -141,9 +141,7 @@ func UserInfoHandler(ctx *gin.Context) {
StatusCode: 200,
}
resp, err := json.Marshal(userResp)
if err != nil {
panic(err)
}
CheckErr(err)
xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
xms64 := base64.RawStdEncoding.EncodeToString(xms)