Add utils

Author: zachhuff386

utils/crypto.go

@@ -0,0 +1,68 @@
+package utils
+
+import (
+	"crypto/rand"
+	"encoding/base64"
+	"math"
+	"math/big"
+	mathrand "math/rand"
+	"regexp"
+
+	"github.com/dropbox/godropbox/errors"
+	"github.com/pritunl/pritunl-vault/errortypes"
+)
+
+var (
+	randRe = regexp.MustCompile("[^a-zA-Z0-9]+")
+)
+
+func RandStr(n int) (str string, err error) {
+	for i := 0; i < 10; i++ {
+		input, e := RandBytes(int(math.Ceil(float64(n) * 1.25)))
+		if e != nil {
+			err = e
+			return
+		}
+
+		output := base64.RawStdEncoding.EncodeToString(input)
+		output = randRe.ReplaceAllString(output, "")
+
+		if len(output) < n {
+			continue
+		}
+
+		str = output[:n]
+		break
+	}
+
+	if str == "" {
+		err = &errortypes.UnknownError{
+			errors.Wrap(err, "utils: Random generate error"),
+		}
+		return
+	}
+
+	return
+}
+
+func RandBytes(size int) (bytes []byte, err error) {
+	bytes = make([]byte, size)
+	_, err = rand.Read(bytes)
+	if err != nil {
+		err = &errortypes.UnknownError{
+			errors.Wrap(err, "utils: Random read error"),
+		}
+		return
+	}
+
+	return
+}
+
+func init() {
+	n, err := rand.Int(rand.Reader, big.NewInt(9223372036854775806))
+	if err != nil {
+		panic(err)
+	}
+
+	mathrand.Seed(n.Int64())
+}

utils/request.go

@@ -0,0 +1,130 @@
+package utils
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin/render"
+)
+
+type NopCloser struct {
+	io.Reader
+}
+
+func (NopCloser) Close() error {
+	return nil
+}
+
+var httpErrCodes = map[int]string{
+	400: "Bad Request",
+	401: "Unauthorized",
+	402: "Payment Required",
+	403: "Forbidden",
+	404: "Not Found",
+	405: "Method Not Allowed",
+	406: "Not Acceptable",
+	407: "Proxy Authentication Required",
+	408: "Request Timeout",
+	409: "Conflict",
+	410: "Gone",
+	411: "Length Required",
+	412: "Precondition Failed",
+	413: "Payload Too Large",
+	414: "URI Too Long",
+	415: "Unsupported Media Type",
+	416: "Range Not Satisfiable",
+	417: "Expectation Failed",
+	421: "Misdirected Request",
+	422: "Unprocessable Entity",
+	423: "Locked",
+	424: "Failed Dependency",
+	426: "Upgrade Required",
+	428: "Precondition Required",
+	429: "Too Many Requests",
+	431: "Request Header Fields Too Large",
+	451: "Unavailable For Legal Reasons",
+	500: "Internal Server Error",
+	501: "Not Implemented",
+	502: "Bad Gateway",
+	503: "Service Unavailable",
+	504: "Gateway Timeout",
+	505: "HTTP Version Not Supported",
+	506: "Variant Also Negotiates",
+	507: "Insufficient Storage",
+	508: "Loop Detected",
+	510: "Not Extended",
+	511: "Network Authentication Required",
+}
+
+func StripPort(hostport string) string {
+	colon := strings.IndexByte(hostport, ':')
+	if colon == -1 {
+		return hostport
+	}
+	if i := strings.IndexByte(hostport, ']'); i != -1 {
+		return strings.TrimPrefix(hostport[:i], "[")
+	}
+	return hostport[:colon]
+}
+
+func FormatHostPort(hostname string, port int) string {
+	if strings.Contains(hostname, ":") {
+		hostname = "[" + hostname + "]"
+	}
+	return fmt.Sprintf("%s:%d", hostname, port)
+}
+
+func GetStatusMessage(code int) string {
+	return fmt.Sprintf("%d %s", code, http.StatusText(code))
+}
+
+func AbortWithStatus(c *gin.Context, code int) {
+	r := render.String{
+		Format: GetStatusMessage(code),
+	}
+
+	c.Status(code)
+	r.WriteContentType(c.Writer)
+	c.Writer.WriteHeaderNow()
+	r.Render(c.Writer)
+	c.Abort()
+}
+
+func AbortWithError(c *gin.Context, code int, err error) {
+	AbortWithStatus(c, code)
+	c.Error(err)
+}
+
+func WriteStatus(w http.ResponseWriter, code int) {
+	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	w.Header().Set("X-Content-Type-Options", "nosniff")
+	w.WriteHeader(code)
+	fmt.Fprintln(w, GetStatusMessage(code))
+}
+
+func WriteText(w http.ResponseWriter, code int, text string) {
+	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	w.Header().Set("X-Content-Type-Options", "nosniff")
+	w.WriteHeader(code)
+	fmt.Fprintln(w, text)
+}
+
+func WriteUnauthorized(w http.ResponseWriter, msg string) {
+	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	w.Header().Set("X-Content-Type-Options", "nosniff")
+	w.WriteHeader(401)
+	fmt.Fprintln(w, "401 "+msg)
+}
+
+func CloneHeader(src http.Header) (dst http.Header) {
+	dst = make(http.Header, len(src))
+	for k, vv := range src {
+		vv2 := make([]string, len(vv))
+		copy(vv2, vv)
+		dst[k] = vv2
+	}
+	return dst
+}