Sfoglia il codice sorgente

Rename secret ee to hex

tags/v2.0.0-rc1
9seconds 5 anni fa
parent
commit
a34949f15c
3 ha cambiato i file con 22 aggiunte e 13 eliminazioni
  1. 9
    4
      cli_access.go
  2. 1
    1
      cli_generate_secret.go
  3. 12
    8
      mtglib/secret.go

+ 9
- 4
cli_access.go Vedi File

@@ -64,8 +64,8 @@ func runAccess(cli *CLI) {
64 64
 		IPv4: runMakeAccessResponseURLs(ipv4, conf, cli),
65 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 70
 	encoder := json.NewEncoder(os.Stdout)
71 71
 
@@ -92,6 +92,10 @@ func runAccessGetIP(ntw *network.Network, protocol string) net.IP {
92 92
 		return nil
93 93
 	}
94 94
 
95
+	if resp.StatusCode != http.StatusOK {
96
+		return nil
97
+	}
98
+
95 99
 	defer exhaustResponse(resp)
96 100
 
97 101
 	data, err := ioutil.ReadAll(resp.Body)
@@ -111,10 +115,11 @@ func runMakeAccessResponseURLs(ip net.IP, conf *config, cli *CLI) *runAccessResp
111 115
 
112 116
 	values.Set("server", ip.String())
113 117
 	values.Set("port", strconv.Itoa(int(conf.BindTo.port.Value(0))))
114
-	values.Set("secret", conf.Secret.Base64())
115 118
 
116 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 125
 	urlQuery := values.Encode()

+ 1
- 1
cli_generate_secret.go Vedi File

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

+ 12
- 8
mtglib/secret.go Vedi File

@@ -57,19 +57,23 @@ func (s *Secret) UnmarshalText(data []byte) error {
57 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 79
 func GenerateSecret(hostname string) Secret {

Loading…
Annulla
Salva