Просмотр исходного кода

Rename secret ee to hex

tags/v2.0.0-rc1
9seconds 5 лет назад
Родитель
Сommit
a34949f15c
3 измененных файлов: 22 добавлений и 13 удалений
  1. 9
    4
      cli_access.go
  2. 1
    1
      cli_generate_secret.go
  3. 12
    8
      mtglib/secret.go

+ 9
- 4
cli_access.go Просмотреть файл

64
 		IPv4: runMakeAccessResponseURLs(ipv4, conf, cli),
64
 		IPv4: runMakeAccessResponseURLs(ipv4, conf, cli),
65
 		IPv6: runMakeAccessResponseURLs(ipv6, conf, cli),
65
 		IPv6: runMakeAccessResponseURLs(ipv6, conf, cli),
66
 	}
66
 	}
67
-    resp.Secret.Base64 = conf.Secret.Base64()
68
-    resp.Secret.Hex = conf.Secret.EE()
67
+	resp.Secret.Base64 = conf.Secret.Base64()
68
+	resp.Secret.Hex = conf.Secret.Hex()
69
 
69
 
70
 	encoder := json.NewEncoder(os.Stdout)
70
 	encoder := json.NewEncoder(os.Stdout)
71
 
71
 
92
 		return nil
92
 		return nil
93
 	}
93
 	}
94
 
94
 
95
+	if resp.StatusCode != http.StatusOK {
96
+		return nil
97
+	}
98
+
95
 	defer exhaustResponse(resp)
99
 	defer exhaustResponse(resp)
96
 
100
 
97
 	data, err := ioutil.ReadAll(resp.Body)
101
 	data, err := ioutil.ReadAll(resp.Body)
111
 
115
 
112
 	values.Set("server", ip.String())
116
 	values.Set("server", ip.String())
113
 	values.Set("port", strconv.Itoa(int(conf.BindTo.port.Value(0))))
117
 	values.Set("port", strconv.Itoa(int(conf.BindTo.port.Value(0))))
114
-	values.Set("secret", conf.Secret.Base64())
115
 
118
 
116
 	if cli.Access.Hex {
119
 	if cli.Access.Hex {
117
-		values.Set("secret", conf.Secret.EE())
120
+		values.Set("secret", conf.Secret.Hex())
121
+	} else {
122
+		values.Set("secret", conf.Secret.Base64())
118
 	}
123
 	}
119
 
124
 
120
 	urlQuery := values.Encode()
125
 	urlQuery := values.Encode()

+ 1
- 1
cli_generate_secret.go Просмотреть файл

10
 	secret := mtglib.GenerateSecret(cli.GenerateSecret.HostName)
10
 	secret := mtglib.GenerateSecret(cli.GenerateSecret.HostName)
11
 
11
 
12
 	if cli.GenerateSecret.Hex {
12
 	if cli.GenerateSecret.Hex {
13
-		fmt.Println(secret.EE())
13
+		fmt.Println(secret.Hex())
14
 	} else {
14
 	} else {
15
 		fmt.Println(secret.Base64())
15
 		fmt.Println(secret.Base64())
16
 	}
16
 	}

+ 12
- 8
mtglib/secret.go Просмотреть файл

57
 	return nil
57
 	return nil
58
 }
58
 }
59
 
59
 
60
-func (s Secret) Base64() string {
61
-	data := append([]byte{238}, s.Key...) // 238 = hex ee
62
-	data = append(data, s.Host...)
60
+func (s Secret) String() string {
61
+	return s.Base64()
62
+}
63
 
63
 
64
-	return base64.RawURLEncoding.EncodeToString(data)
64
+func (s Secret) Base64() string {
65
+	return base64.RawURLEncoding.EncodeToString(s.makeBytes())
65
 }
66
 }
66
 
67
 
67
-func (s Secret) String() string {
68
-	return s.Base64()
68
+func (s Secret) Hex() string {
69
+	return hex.EncodeToString(s.makeBytes())
69
 }
70
 }
70
 
71
 
71
-func (s Secret) EE() string {
72
-	return "ee" + hex.EncodeToString(append(s.Key, s.Host...))
72
+func (s *Secret) makeBytes() []byte {
73
+	data := append([]byte{238}, s.Key...) // hex 'ee' = 238
74
+	data = append(data, s.Host...)
75
+
76
+	return data
73
 }
77
 }
74
 
78
 
75
 func GenerateSecret(hostname string) Secret {
79
 func GenerateSecret(hostname string) Secret {

Загрузка…
Отмена
Сохранить