8e79f2ee9c
Based on Nekogram. Key additions: - Rebrand to FoxiGram (app name, APK name, applicationId com.foxigram.app) - Embedded Xray (VLESS+Reality) proxy client via JNI libxray.so - Bundled hidden one-tap proxies (LTE + WiFi), read-only in UI - Auto-restore proxy on restart, rebind to active network (LTE/WiFi) - Server credentials externalized to git-ignored XrayServers.java (+ template) - libxray Go source included; compiled .so, keystore, google-services.json ignored
76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
// Copyright 2022 The BoringSSL Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package runner
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
shardStatusFileEnv = "TEST_SHARD_STATUS_FILE"
|
|
shardTotalEnv = "TEST_TOTAL_SHARDS"
|
|
shardIndexEnv = "TEST_SHARD_INDEX"
|
|
shardPrefix = "RUNNER_"
|
|
)
|
|
|
|
func init() {
|
|
// When run under `go test`, init() functions may be run twice if the
|
|
// test binary ends up forking and execing itself. Therefore we move
|
|
// the environment variables to names that don't interfere with Go's
|
|
// own support for sharding. If we recorded and erased them, then they
|
|
// wouldn't exist the second time the binary runs.
|
|
for _, key := range []string{shardStatusFileEnv, shardTotalEnv, shardIndexEnv} {
|
|
value := os.Getenv(key)
|
|
if len(value) > 0 {
|
|
os.Setenv(shardPrefix+key, value)
|
|
os.Setenv(key, "")
|
|
}
|
|
}
|
|
}
|
|
|
|
// getSharding returns the shard index and count, or zeros if sharding is not
|
|
// enabled.
|
|
func getSharding() (index, total int, err error) {
|
|
statusFile := os.Getenv(shardPrefix + shardStatusFileEnv)
|
|
totalNumStr := os.Getenv(shardPrefix + shardTotalEnv)
|
|
indexStr := os.Getenv(shardPrefix + shardIndexEnv)
|
|
if len(totalNumStr) == 0 || len(indexStr) == 0 {
|
|
return 0, 0, nil
|
|
}
|
|
|
|
totalNum, err := strconv.Atoi(totalNumStr)
|
|
if err != nil {
|
|
return 0, 0, fmt.Errorf("$%s is %q, but expected a number\n", shardTotalEnv, totalNumStr)
|
|
}
|
|
|
|
index, err = strconv.Atoi(indexStr)
|
|
if err != nil {
|
|
return 0, 0, fmt.Errorf("$%s is %q, but expected a number\n", shardIndexEnv, indexStr)
|
|
}
|
|
|
|
if index < 0 || index >= totalNum {
|
|
return 0, 0, fmt.Errorf("shard index/total of %d/%d is invalid\n", index, totalNum)
|
|
}
|
|
|
|
if len(statusFile) > 0 {
|
|
if err := os.WriteFile(statusFile, nil, 0664); err != nil {
|
|
return 0, 0, err
|
|
}
|
|
}
|
|
|
|
return index, totalNum, nil
|
|
}
|