Fix vendor
Bel LaPointe
parent
81782bb794
commit
75a65cecd2
|
|
@ -1,98 +0,0 @@
|
||||||
package logger
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var outfile *os.File
|
|
||||||
var outfileB *os.File
|
|
||||||
|
|
||||||
var lastWasReturn = false
|
|
||||||
|
|
||||||
var ErrCantConfigLog = errors.New("cannot configure log")
|
|
||||||
|
|
||||||
func Config(path string) error {
|
|
||||||
if path == "" {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
f, err := os.OpenFile(path, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0600)
|
|
||||||
if err != nil {
|
|
||||||
return ErrCantConfigLog
|
|
||||||
}
|
|
||||||
outfile = f
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func Logf(args ...interface{}) {
|
|
||||||
if outfile == nil {
|
|
||||||
outfile = os.Stderr
|
|
||||||
}
|
|
||||||
defer outfile.Sync()
|
|
||||||
if len(args) == 0 {
|
|
||||||
fmt.Fprintln(outfile, "")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
v := fmt.Sprintf(args[0].(string), args[1:]...)
|
|
||||||
if v[len(v)-1] == '\n' {
|
|
||||||
v = v[:len(v)-1]
|
|
||||||
}
|
|
||||||
thisWasReturn := v[0] == '\r'
|
|
||||||
if lastWasReturn && !thisWasReturn {
|
|
||||||
fmt.Println()
|
|
||||||
}
|
|
||||||
if thisWasReturn {
|
|
||||||
fmt.Fprintf(outfile, "\r%s%v", prefix(), v[1:])
|
|
||||||
} else {
|
|
||||||
fmt.Fprintf(outfile, "%s%v\n", prefix(), v)
|
|
||||||
}
|
|
||||||
lastWasReturn = thisWasReturn
|
|
||||||
}
|
|
||||||
|
|
||||||
func Fatalf(args ...interface{}) {
|
|
||||||
Logf(args...)
|
|
||||||
if len(args) > 0 {
|
|
||||||
panic(args[0])
|
|
||||||
} else {
|
|
||||||
panic("Panic from logger.fatal()")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func Fatal(args ...interface{}) {
|
|
||||||
Log(args...)
|
|
||||||
if len(args) > 0 {
|
|
||||||
panic(args[0])
|
|
||||||
} else {
|
|
||||||
panic("Panic from logger.fatal()")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func Log(args ...interface{}) {
|
|
||||||
if lastWasReturn {
|
|
||||||
fmt.Println()
|
|
||||||
}
|
|
||||||
if outfile == nil {
|
|
||||||
outfile = os.Stderr
|
|
||||||
}
|
|
||||||
v := fmt.Sprintf("%v", args)
|
|
||||||
v = v[1 : len(v)-1]
|
|
||||||
fmt.Fprintf(outfile, "%s%v\n", prefix(), v)
|
|
||||||
outfile.Sync()
|
|
||||||
lastWasReturn = false
|
|
||||||
}
|
|
||||||
|
|
||||||
func LogB(args ...interface{}) {
|
|
||||||
if outfileB == nil {
|
|
||||||
outfileB = os.Stderr
|
|
||||||
}
|
|
||||||
v := fmt.Sprintf("%v", args)
|
|
||||||
v = v[1 : len(v)-1]
|
|
||||||
fmt.Fprintf(outfileB, "%s%v\n", prefix(), v)
|
|
||||||
outfileB.Sync()
|
|
||||||
}
|
|
||||||
|
|
||||||
func prefix() string {
|
|
||||||
return time.Now().Format("20060102-150405") + " : "
|
|
||||||
}
|
|
||||||
|
|
@ -1,6 +0,0 @@
|
||||||
runtime: go
|
|
||||||
api_version: go1
|
|
||||||
handlers:
|
|
||||||
- url: /.*
|
|
||||||
script: _go_app
|
|
||||||
secure: always
|
|
||||||
|
|
@ -1,134 +0,0 @@
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/binary"
|
|
||||||
"encoding/csv"
|
|
||||||
"flag"
|
|
||||||
"local/logger"
|
|
||||||
"log"
|
|
||||||
"net"
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"path"
|
|
||||||
"path/filepath"
|
|
||||||
"sort"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"google.golang.org/appengine"
|
|
||||||
)
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
exePath, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
exePath = "."
|
|
||||||
|
|
||||||
directory := flag.String("d", path.Join(exePath, "public"), "the directory of static file to host")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
getIPs()
|
|
||||||
|
|
||||||
http.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
remoteIP := strings.Split(r.RemoteAddr, ":")[0]
|
|
||||||
if notUSA(remoteIP) {
|
|
||||||
logger.Log(remoteIP, "NOT USA")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if r.URL.Scheme == "http" || strings.HasPrefix(r.Host, "http:") {
|
|
||||||
r.URL.Scheme = "https"
|
|
||||||
http.Redirect(w, r, r.URL.String(), http.StatusTemporaryRedirect)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
http.FileServer(http.Dir(*directory)).ServeHTTP(w, r)
|
|
||||||
}))
|
|
||||||
|
|
||||||
log.Printf("Serving %s\n", *directory)
|
|
||||||
|
|
||||||
appengine.Main()
|
|
||||||
}
|
|
||||||
|
|
||||||
var globalIPs []uint64
|
|
||||||
|
|
||||||
func getIPs() []uint64 {
|
|
||||||
if globalIPs != nil {
|
|
||||||
return globalIPs
|
|
||||||
}
|
|
||||||
|
|
||||||
globalIPs = make([]uint64, 0)
|
|
||||||
|
|
||||||
f, err := os.Open("private/ipv4.csv")
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
ipv4r := csv.NewReader(f)
|
|
||||||
ipv4all, err := ipv4r.ReadAll()
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
logger.Log("IPV4s")
|
|
||||||
for i := range ipv4all {
|
|
||||||
if ipv4all[i][2] == "US" || ipv4all[i][2] == "-" {
|
|
||||||
start, err := strconv.ParseUint(ipv4all[i][0], 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
stop, err := strconv.ParseUint(ipv4all[i][1], 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
globalIPs = append(globalIPs, uint64(start), uint64(stop))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
g, err := os.Open("private/ipv6.csv")
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ipv6r := csv.NewReader(g)
|
|
||||||
ipv6all, err := ipv6r.ReadAll()
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
logger.Log("IPV6s")
|
|
||||||
for i := range ipv6all {
|
|
||||||
if ipv6all[i][2] == "US" {
|
|
||||||
start, err := strconv.ParseUint(ipv6all[i][0], 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
stop, err := strconv.ParseUint(ipv6all[i][1], 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
globalIPs = append(globalIPs, uint64(start), uint64(stop))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
sort.Slice(globalIPs, func(i, j int) bool {
|
|
||||||
return globalIPs[i] > globalIPs[j]
|
|
||||||
})
|
|
||||||
return globalIPs
|
|
||||||
}
|
|
||||||
|
|
||||||
func notUSA(ip string) bool {
|
|
||||||
dec := toDec(ip)
|
|
||||||
ips := getIPs()
|
|
||||||
n := sort.Search(len(ips), func(i int) bool {
|
|
||||||
return ips[i] > dec
|
|
||||||
})
|
|
||||||
logger.Log(ip, dec, ips[0], n, len(ips))
|
|
||||||
return n%2 == 1
|
|
||||||
}
|
|
||||||
|
|
||||||
func toDec(ips string) uint64 {
|
|
||||||
ip := net.ParseIP(ips)
|
|
||||||
if ip == nil {
|
|
||||||
return uint64(0)
|
|
||||||
}
|
|
||||||
if len(ip) == 16 {
|
|
||||||
return uint64(binary.BigEndian.Uint32(ip[12:16]))
|
|
||||||
}
|
|
||||||
return uint64(binary.BigEndian.Uint32(ip))
|
|
||||||
}
|
|
||||||
|
|
@ -1,33 +0,0 @@
|
||||||
package main
|
|
||||||
|
|
||||||
import "testing"
|
|
||||||
|
|
||||||
func Test_notUSA(t *testing.T) {
|
|
||||||
cases := []struct {
|
|
||||||
ip string
|
|
||||||
ok bool
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
ip: "8.8.8.8",
|
|
||||||
ok: false,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
ip: "192.168.0.86",
|
|
||||||
ok: false,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
ip: "127.0.0.1",
|
|
||||||
ok: false,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
ip: "223.144.0.0",
|
|
||||||
ok: true,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, c := range cases {
|
|
||||||
if notUSA(c.ip) != c.ok {
|
|
||||||
t.Errorf("WRONG VALIDATION for %v, expected %v", c.ip, c.ok)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue