mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-10-30 06:38:37 -04:00 
			
		
		
		
	* update github.com/blevesearch/bleve v2.0.2 -> v2.0.3 * github.com/denisenkom/go-mssqldb v0.9.0 -> v0.10.0 * github.com/editorconfig/editorconfig-core-go v2.4.1 -> v2.4.2 * github.com/go-chi/cors v1.1.1 -> v1.2.0 * github.com/go-git/go-billy v5.0.0 -> v5.1.0 * github.com/go-git/go-git v5.2.0 -> v5.3.0 * github.com/go-ldap/ldap v3.2.4 -> v3.3.0 * github.com/go-redis/redis v8.6.0 -> v8.8.2 * github.com/go-sql-driver/mysql v1.5.0 -> v1.6.0 * github.com/go-swagger/go-swagger v0.26.1 -> v0.27.0 * github.com/lib/pq v1.9.0 -> v1.10.1 * github.com/mattn/go-sqlite3 v1.14.6 -> v1.14.7 * github.com/go-testfixtures/testfixtures v3.5.0 -> v3.6.0 * github.com/issue9/identicon v1.0.1 -> v1.2.0 * github.com/klauspost/compress v1.11.8 -> v1.12.1 * github.com/mgechev/revive v1.0.3 -> v1.0.6 * github.com/microcosm-cc/bluemonday v1.0.7 -> v1.0.8 * github.com/niklasfasching/go-org v1.4.0 -> v1.5.0 * github.com/olivere/elastic v7.0.22 -> v7.0.24 * github.com/pelletier/go-toml v1.8.1 -> v1.9.0 * github.com/prometheus/client_golang v1.9.0 -> v1.10.0 * github.com/xanzy/go-gitlab v0.44.0 -> v0.48.0 * github.com/yuin/goldmark v1.3.3 -> v1.3.5 * github.com/6543/go-version v1.2.4 -> v1.3.1 * do github.com/lib/pq v1.10.0 -> v1.10.1 again ...
		
			
				
	
	
		
			869 lines
		
	
	
		
			21 KiB
		
	
	
	
		
			Go
		
	
	
	
		
			Vendored
		
	
	
	
			
		
		
	
	
			869 lines
		
	
	
		
			21 KiB
		
	
	
	
		
			Go
		
	
	
	
		
			Vendored
		
	
	
	
| // Go MySQL Driver - A MySQL-Driver for Go's database/sql package
 | |
| //
 | |
| // Copyright 2012 The Go-MySQL-Driver Authors. All rights reserved.
 | |
| //
 | |
| // This Source Code Form is subject to the terms of the Mozilla Public
 | |
| // License, v. 2.0. If a copy of the MPL was not distributed with this file,
 | |
| // You can obtain one at http://mozilla.org/MPL/2.0/.
 | |
| 
 | |
| package mysql
 | |
| 
 | |
| import (
 | |
| 	"crypto/tls"
 | |
| 	"database/sql"
 | |
| 	"database/sql/driver"
 | |
| 	"encoding/binary"
 | |
| 	"errors"
 | |
| 	"fmt"
 | |
| 	"io"
 | |
| 	"strconv"
 | |
| 	"strings"
 | |
| 	"sync"
 | |
| 	"sync/atomic"
 | |
| 	"time"
 | |
| )
 | |
| 
 | |
| // Registry for custom tls.Configs
 | |
| var (
 | |
| 	tlsConfigLock     sync.RWMutex
 | |
| 	tlsConfigRegistry map[string]*tls.Config
 | |
| )
 | |
| 
 | |
| // RegisterTLSConfig registers a custom tls.Config to be used with sql.Open.
 | |
| // Use the key as a value in the DSN where tls=value.
 | |
| //
 | |
| // Note: The provided tls.Config is exclusively owned by the driver after
 | |
| // registering it.
 | |
| //
 | |
| //  rootCertPool := x509.NewCertPool()
 | |
| //  pem, err := ioutil.ReadFile("/path/ca-cert.pem")
 | |
| //  if err != nil {
 | |
| //      log.Fatal(err)
 | |
| //  }
 | |
| //  if ok := rootCertPool.AppendCertsFromPEM(pem); !ok {
 | |
| //      log.Fatal("Failed to append PEM.")
 | |
| //  }
 | |
| //  clientCert := make([]tls.Certificate, 0, 1)
 | |
| //  certs, err := tls.LoadX509KeyPair("/path/client-cert.pem", "/path/client-key.pem")
 | |
| //  if err != nil {
 | |
| //      log.Fatal(err)
 | |
| //  }
 | |
| //  clientCert = append(clientCert, certs)
 | |
| //  mysql.RegisterTLSConfig("custom", &tls.Config{
 | |
| //      RootCAs: rootCertPool,
 | |
| //      Certificates: clientCert,
 | |
| //  })
 | |
| //  db, err := sql.Open("mysql", "user@tcp(localhost:3306)/test?tls=custom")
 | |
| //
 | |
| func RegisterTLSConfig(key string, config *tls.Config) error {
 | |
| 	if _, isBool := readBool(key); isBool || strings.ToLower(key) == "skip-verify" || strings.ToLower(key) == "preferred" {
 | |
| 		return fmt.Errorf("key '%s' is reserved", key)
 | |
| 	}
 | |
| 
 | |
| 	tlsConfigLock.Lock()
 | |
| 	if tlsConfigRegistry == nil {
 | |
| 		tlsConfigRegistry = make(map[string]*tls.Config)
 | |
| 	}
 | |
| 
 | |
| 	tlsConfigRegistry[key] = config
 | |
| 	tlsConfigLock.Unlock()
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // DeregisterTLSConfig removes the tls.Config associated with key.
 | |
| func DeregisterTLSConfig(key string) {
 | |
| 	tlsConfigLock.Lock()
 | |
| 	if tlsConfigRegistry != nil {
 | |
| 		delete(tlsConfigRegistry, key)
 | |
| 	}
 | |
| 	tlsConfigLock.Unlock()
 | |
| }
 | |
| 
 | |
| func getTLSConfigClone(key string) (config *tls.Config) {
 | |
| 	tlsConfigLock.RLock()
 | |
| 	if v, ok := tlsConfigRegistry[key]; ok {
 | |
| 		config = v.Clone()
 | |
| 	}
 | |
| 	tlsConfigLock.RUnlock()
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // Returns the bool value of the input.
 | |
| // The 2nd return value indicates if the input was a valid bool value
 | |
| func readBool(input string) (value bool, valid bool) {
 | |
| 	switch input {
 | |
| 	case "1", "true", "TRUE", "True":
 | |
| 		return true, true
 | |
| 	case "0", "false", "FALSE", "False":
 | |
| 		return false, true
 | |
| 	}
 | |
| 
 | |
| 	// Not a valid bool value
 | |
| 	return
 | |
| }
 | |
| 
 | |
| /******************************************************************************
 | |
| *                           Time related utils                                *
 | |
| ******************************************************************************/
 | |
| 
 | |
| func parseDateTime(b []byte, loc *time.Location) (time.Time, error) {
 | |
| 	const base = "0000-00-00 00:00:00.000000"
 | |
| 	switch len(b) {
 | |
| 	case 10, 19, 21, 22, 23, 24, 25, 26: // up to "YYYY-MM-DD HH:MM:SS.MMMMMM"
 | |
| 		if string(b) == base[:len(b)] {
 | |
| 			return time.Time{}, nil
 | |
| 		}
 | |
| 
 | |
| 		year, err := parseByteYear(b)
 | |
| 		if err != nil {
 | |
| 			return time.Time{}, err
 | |
| 		}
 | |
| 		if year <= 0 {
 | |
| 			year = 1
 | |
| 		}
 | |
| 
 | |
| 		if b[4] != '-' {
 | |
| 			return time.Time{}, fmt.Errorf("bad value for field: `%c`", b[4])
 | |
| 		}
 | |
| 
 | |
| 		m, err := parseByte2Digits(b[5], b[6])
 | |
| 		if err != nil {
 | |
| 			return time.Time{}, err
 | |
| 		}
 | |
| 		if m <= 0 {
 | |
| 			m = 1
 | |
| 		}
 | |
| 		month := time.Month(m)
 | |
| 
 | |
| 		if b[7] != '-' {
 | |
| 			return time.Time{}, fmt.Errorf("bad value for field: `%c`", b[7])
 | |
| 		}
 | |
| 
 | |
| 		day, err := parseByte2Digits(b[8], b[9])
 | |
| 		if err != nil {
 | |
| 			return time.Time{}, err
 | |
| 		}
 | |
| 		if day <= 0 {
 | |
| 			day = 1
 | |
| 		}
 | |
| 		if len(b) == 10 {
 | |
| 			return time.Date(year, month, day, 0, 0, 0, 0, loc), nil
 | |
| 		}
 | |
| 
 | |
| 		if b[10] != ' ' {
 | |
| 			return time.Time{}, fmt.Errorf("bad value for field: `%c`", b[10])
 | |
| 		}
 | |
| 
 | |
| 		hour, err := parseByte2Digits(b[11], b[12])
 | |
| 		if err != nil {
 | |
| 			return time.Time{}, err
 | |
| 		}
 | |
| 		if b[13] != ':' {
 | |
| 			return time.Time{}, fmt.Errorf("bad value for field: `%c`", b[13])
 | |
| 		}
 | |
| 
 | |
| 		min, err := parseByte2Digits(b[14], b[15])
 | |
| 		if err != nil {
 | |
| 			return time.Time{}, err
 | |
| 		}
 | |
| 		if b[16] != ':' {
 | |
| 			return time.Time{}, fmt.Errorf("bad value for field: `%c`", b[16])
 | |
| 		}
 | |
| 
 | |
| 		sec, err := parseByte2Digits(b[17], b[18])
 | |
| 		if err != nil {
 | |
| 			return time.Time{}, err
 | |
| 		}
 | |
| 		if len(b) == 19 {
 | |
| 			return time.Date(year, month, day, hour, min, sec, 0, loc), nil
 | |
| 		}
 | |
| 
 | |
| 		if b[19] != '.' {
 | |
| 			return time.Time{}, fmt.Errorf("bad value for field: `%c`", b[19])
 | |
| 		}
 | |
| 		nsec, err := parseByteNanoSec(b[20:])
 | |
| 		if err != nil {
 | |
| 			return time.Time{}, err
 | |
| 		}
 | |
| 		return time.Date(year, month, day, hour, min, sec, nsec, loc), nil
 | |
| 	default:
 | |
| 		return time.Time{}, fmt.Errorf("invalid time bytes: %s", b)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func parseByteYear(b []byte) (int, error) {
 | |
| 	year, n := 0, 1000
 | |
| 	for i := 0; i < 4; i++ {
 | |
| 		v, err := bToi(b[i])
 | |
| 		if err != nil {
 | |
| 			return 0, err
 | |
| 		}
 | |
| 		year += v * n
 | |
| 		n = n / 10
 | |
| 	}
 | |
| 	return year, nil
 | |
| }
 | |
| 
 | |
| func parseByte2Digits(b1, b2 byte) (int, error) {
 | |
| 	d1, err := bToi(b1)
 | |
| 	if err != nil {
 | |
| 		return 0, err
 | |
| 	}
 | |
| 	d2, err := bToi(b2)
 | |
| 	if err != nil {
 | |
| 		return 0, err
 | |
| 	}
 | |
| 	return d1*10 + d2, nil
 | |
| }
 | |
| 
 | |
| func parseByteNanoSec(b []byte) (int, error) {
 | |
| 	ns, digit := 0, 100000 // max is 6-digits
 | |
| 	for i := 0; i < len(b); i++ {
 | |
| 		v, err := bToi(b[i])
 | |
| 		if err != nil {
 | |
| 			return 0, err
 | |
| 		}
 | |
| 		ns += v * digit
 | |
| 		digit /= 10
 | |
| 	}
 | |
| 	// nanoseconds has 10-digits. (needs to scale digits)
 | |
| 	// 10 - 6 = 4, so we have to multiple 1000.
 | |
| 	return ns * 1000, nil
 | |
| }
 | |
| 
 | |
| func bToi(b byte) (int, error) {
 | |
| 	if b < '0' || b > '9' {
 | |
| 		return 0, errors.New("not [0-9]")
 | |
| 	}
 | |
| 	return int(b - '0'), nil
 | |
| }
 | |
| 
 | |
| func parseBinaryDateTime(num uint64, data []byte, loc *time.Location) (driver.Value, error) {
 | |
| 	switch num {
 | |
| 	case 0:
 | |
| 		return time.Time{}, nil
 | |
| 	case 4:
 | |
| 		return time.Date(
 | |
| 			int(binary.LittleEndian.Uint16(data[:2])), // year
 | |
| 			time.Month(data[2]),                       // month
 | |
| 			int(data[3]),                              // day
 | |
| 			0, 0, 0, 0,
 | |
| 			loc,
 | |
| 		), nil
 | |
| 	case 7:
 | |
| 		return time.Date(
 | |
| 			int(binary.LittleEndian.Uint16(data[:2])), // year
 | |
| 			time.Month(data[2]),                       // month
 | |
| 			int(data[3]),                              // day
 | |
| 			int(data[4]),                              // hour
 | |
| 			int(data[5]),                              // minutes
 | |
| 			int(data[6]),                              // seconds
 | |
| 			0,
 | |
| 			loc,
 | |
| 		), nil
 | |
| 	case 11:
 | |
| 		return time.Date(
 | |
| 			int(binary.LittleEndian.Uint16(data[:2])), // year
 | |
| 			time.Month(data[2]),                       // month
 | |
| 			int(data[3]),                              // day
 | |
| 			int(data[4]),                              // hour
 | |
| 			int(data[5]),                              // minutes
 | |
| 			int(data[6]),                              // seconds
 | |
| 			int(binary.LittleEndian.Uint32(data[7:11]))*1000, // nanoseconds
 | |
| 			loc,
 | |
| 		), nil
 | |
| 	}
 | |
| 	return nil, fmt.Errorf("invalid DATETIME packet length %d", num)
 | |
| }
 | |
| 
 | |
| func appendDateTime(buf []byte, t time.Time) ([]byte, error) {
 | |
| 	year, month, day := t.Date()
 | |
| 	hour, min, sec := t.Clock()
 | |
| 	nsec := t.Nanosecond()
 | |
| 
 | |
| 	if year < 1 || year > 9999 {
 | |
| 		return buf, errors.New("year is not in the range [1, 9999]: " + strconv.Itoa(year)) // use errors.New instead of fmt.Errorf to avoid year escape to heap
 | |
| 	}
 | |
| 	year100 := year / 100
 | |
| 	year1 := year % 100
 | |
| 
 | |
| 	var localBuf [len("2006-01-02T15:04:05.999999999")]byte // does not escape
 | |
| 	localBuf[0], localBuf[1], localBuf[2], localBuf[3] = digits10[year100], digits01[year100], digits10[year1], digits01[year1]
 | |
| 	localBuf[4] = '-'
 | |
| 	localBuf[5], localBuf[6] = digits10[month], digits01[month]
 | |
| 	localBuf[7] = '-'
 | |
| 	localBuf[8], localBuf[9] = digits10[day], digits01[day]
 | |
| 
 | |
| 	if hour == 0 && min == 0 && sec == 0 && nsec == 0 {
 | |
| 		return append(buf, localBuf[:10]...), nil
 | |
| 	}
 | |
| 
 | |
| 	localBuf[10] = ' '
 | |
| 	localBuf[11], localBuf[12] = digits10[hour], digits01[hour]
 | |
| 	localBuf[13] = ':'
 | |
| 	localBuf[14], localBuf[15] = digits10[min], digits01[min]
 | |
| 	localBuf[16] = ':'
 | |
| 	localBuf[17], localBuf[18] = digits10[sec], digits01[sec]
 | |
| 
 | |
| 	if nsec == 0 {
 | |
| 		return append(buf, localBuf[:19]...), nil
 | |
| 	}
 | |
| 	nsec100000000 := nsec / 100000000
 | |
| 	nsec1000000 := (nsec / 1000000) % 100
 | |
| 	nsec10000 := (nsec / 10000) % 100
 | |
| 	nsec100 := (nsec / 100) % 100
 | |
| 	nsec1 := nsec % 100
 | |
| 	localBuf[19] = '.'
 | |
| 
 | |
| 	// milli second
 | |
| 	localBuf[20], localBuf[21], localBuf[22] =
 | |
| 		digits01[nsec100000000], digits10[nsec1000000], digits01[nsec1000000]
 | |
| 	// micro second
 | |
| 	localBuf[23], localBuf[24], localBuf[25] =
 | |
| 		digits10[nsec10000], digits01[nsec10000], digits10[nsec100]
 | |
| 	// nano second
 | |
| 	localBuf[26], localBuf[27], localBuf[28] =
 | |
| 		digits01[nsec100], digits10[nsec1], digits01[nsec1]
 | |
| 
 | |
| 	// trim trailing zeros
 | |
| 	n := len(localBuf)
 | |
| 	for n > 0 && localBuf[n-1] == '0' {
 | |
| 		n--
 | |
| 	}
 | |
| 
 | |
| 	return append(buf, localBuf[:n]...), nil
 | |
| }
 | |
| 
 | |
| // zeroDateTime is used in formatBinaryDateTime to avoid an allocation
 | |
| // if the DATE or DATETIME has the zero value.
 | |
| // It must never be changed.
 | |
| // The current behavior depends on database/sql copying the result.
 | |
| var zeroDateTime = []byte("0000-00-00 00:00:00.000000")
 | |
| 
 | |
| const digits01 = "0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789"
 | |
| const digits10 = "0000000000111111111122222222223333333333444444444455555555556666666666777777777788888888889999999999"
 | |
| 
 | |
| func appendMicrosecs(dst, src []byte, decimals int) []byte {
 | |
| 	if decimals <= 0 {
 | |
| 		return dst
 | |
| 	}
 | |
| 	if len(src) == 0 {
 | |
| 		return append(dst, ".000000"[:decimals+1]...)
 | |
| 	}
 | |
| 
 | |
| 	microsecs := binary.LittleEndian.Uint32(src[:4])
 | |
| 	p1 := byte(microsecs / 10000)
 | |
| 	microsecs -= 10000 * uint32(p1)
 | |
| 	p2 := byte(microsecs / 100)
 | |
| 	microsecs -= 100 * uint32(p2)
 | |
| 	p3 := byte(microsecs)
 | |
| 
 | |
| 	switch decimals {
 | |
| 	default:
 | |
| 		return append(dst, '.',
 | |
| 			digits10[p1], digits01[p1],
 | |
| 			digits10[p2], digits01[p2],
 | |
| 			digits10[p3], digits01[p3],
 | |
| 		)
 | |
| 	case 1:
 | |
| 		return append(dst, '.',
 | |
| 			digits10[p1],
 | |
| 		)
 | |
| 	case 2:
 | |
| 		return append(dst, '.',
 | |
| 			digits10[p1], digits01[p1],
 | |
| 		)
 | |
| 	case 3:
 | |
| 		return append(dst, '.',
 | |
| 			digits10[p1], digits01[p1],
 | |
| 			digits10[p2],
 | |
| 		)
 | |
| 	case 4:
 | |
| 		return append(dst, '.',
 | |
| 			digits10[p1], digits01[p1],
 | |
| 			digits10[p2], digits01[p2],
 | |
| 		)
 | |
| 	case 5:
 | |
| 		return append(dst, '.',
 | |
| 			digits10[p1], digits01[p1],
 | |
| 			digits10[p2], digits01[p2],
 | |
| 			digits10[p3],
 | |
| 		)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func formatBinaryDateTime(src []byte, length uint8) (driver.Value, error) {
 | |
| 	// length expects the deterministic length of the zero value,
 | |
| 	// negative time and 100+ hours are automatically added if needed
 | |
| 	if len(src) == 0 {
 | |
| 		return zeroDateTime[:length], nil
 | |
| 	}
 | |
| 	var dst []byte      // return value
 | |
| 	var p1, p2, p3 byte // current digit pair
 | |
| 
 | |
| 	switch length {
 | |
| 	case 10, 19, 21, 22, 23, 24, 25, 26:
 | |
| 	default:
 | |
| 		t := "DATE"
 | |
| 		if length > 10 {
 | |
| 			t += "TIME"
 | |
| 		}
 | |
| 		return nil, fmt.Errorf("illegal %s length %d", t, length)
 | |
| 	}
 | |
| 	switch len(src) {
 | |
| 	case 4, 7, 11:
 | |
| 	default:
 | |
| 		t := "DATE"
 | |
| 		if length > 10 {
 | |
| 			t += "TIME"
 | |
| 		}
 | |
| 		return nil, fmt.Errorf("illegal %s packet length %d", t, len(src))
 | |
| 	}
 | |
| 	dst = make([]byte, 0, length)
 | |
| 	// start with the date
 | |
| 	year := binary.LittleEndian.Uint16(src[:2])
 | |
| 	pt := year / 100
 | |
| 	p1 = byte(year - 100*uint16(pt))
 | |
| 	p2, p3 = src[2], src[3]
 | |
| 	dst = append(dst,
 | |
| 		digits10[pt], digits01[pt],
 | |
| 		digits10[p1], digits01[p1], '-',
 | |
| 		digits10[p2], digits01[p2], '-',
 | |
| 		digits10[p3], digits01[p3],
 | |
| 	)
 | |
| 	if length == 10 {
 | |
| 		return dst, nil
 | |
| 	}
 | |
| 	if len(src) == 4 {
 | |
| 		return append(dst, zeroDateTime[10:length]...), nil
 | |
| 	}
 | |
| 	dst = append(dst, ' ')
 | |
| 	p1 = src[4] // hour
 | |
| 	src = src[5:]
 | |
| 
 | |
| 	// p1 is 2-digit hour, src is after hour
 | |
| 	p2, p3 = src[0], src[1]
 | |
| 	dst = append(dst,
 | |
| 		digits10[p1], digits01[p1], ':',
 | |
| 		digits10[p2], digits01[p2], ':',
 | |
| 		digits10[p3], digits01[p3],
 | |
| 	)
 | |
| 	return appendMicrosecs(dst, src[2:], int(length)-20), nil
 | |
| }
 | |
| 
 | |
| func formatBinaryTime(src []byte, length uint8) (driver.Value, error) {
 | |
| 	// length expects the deterministic length of the zero value,
 | |
| 	// negative time and 100+ hours are automatically added if needed
 | |
| 	if len(src) == 0 {
 | |
| 		return zeroDateTime[11 : 11+length], nil
 | |
| 	}
 | |
| 	var dst []byte // return value
 | |
| 
 | |
| 	switch length {
 | |
| 	case
 | |
| 		8,                      // time (can be up to 10 when negative and 100+ hours)
 | |
| 		10, 11, 12, 13, 14, 15: // time with fractional seconds
 | |
| 	default:
 | |
| 		return nil, fmt.Errorf("illegal TIME length %d", length)
 | |
| 	}
 | |
| 	switch len(src) {
 | |
| 	case 8, 12:
 | |
| 	default:
 | |
| 		return nil, fmt.Errorf("invalid TIME packet length %d", len(src))
 | |
| 	}
 | |
| 	// +2 to enable negative time and 100+ hours
 | |
| 	dst = make([]byte, 0, length+2)
 | |
| 	if src[0] == 1 {
 | |
| 		dst = append(dst, '-')
 | |
| 	}
 | |
| 	days := binary.LittleEndian.Uint32(src[1:5])
 | |
| 	hours := int64(days)*24 + int64(src[5])
 | |
| 
 | |
| 	if hours >= 100 {
 | |
| 		dst = strconv.AppendInt(dst, hours, 10)
 | |
| 	} else {
 | |
| 		dst = append(dst, digits10[hours], digits01[hours])
 | |
| 	}
 | |
| 
 | |
| 	min, sec := src[6], src[7]
 | |
| 	dst = append(dst, ':',
 | |
| 		digits10[min], digits01[min], ':',
 | |
| 		digits10[sec], digits01[sec],
 | |
| 	)
 | |
| 	return appendMicrosecs(dst, src[8:], int(length)-9), nil
 | |
| }
 | |
| 
 | |
| /******************************************************************************
 | |
| *                       Convert from and to bytes                             *
 | |
| ******************************************************************************/
 | |
| 
 | |
| func uint64ToBytes(n uint64) []byte {
 | |
| 	return []byte{
 | |
| 		byte(n),
 | |
| 		byte(n >> 8),
 | |
| 		byte(n >> 16),
 | |
| 		byte(n >> 24),
 | |
| 		byte(n >> 32),
 | |
| 		byte(n >> 40),
 | |
| 		byte(n >> 48),
 | |
| 		byte(n >> 56),
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func uint64ToString(n uint64) []byte {
 | |
| 	var a [20]byte
 | |
| 	i := 20
 | |
| 
 | |
| 	// U+0030 = 0
 | |
| 	// ...
 | |
| 	// U+0039 = 9
 | |
| 
 | |
| 	var q uint64
 | |
| 	for n >= 10 {
 | |
| 		i--
 | |
| 		q = n / 10
 | |
| 		a[i] = uint8(n-q*10) + 0x30
 | |
| 		n = q
 | |
| 	}
 | |
| 
 | |
| 	i--
 | |
| 	a[i] = uint8(n) + 0x30
 | |
| 
 | |
| 	return a[i:]
 | |
| }
 | |
| 
 | |
| // treats string value as unsigned integer representation
 | |
| func stringToInt(b []byte) int {
 | |
| 	val := 0
 | |
| 	for i := range b {
 | |
| 		val *= 10
 | |
| 		val += int(b[i] - 0x30)
 | |
| 	}
 | |
| 	return val
 | |
| }
 | |
| 
 | |
| // returns the string read as a bytes slice, wheter the value is NULL,
 | |
| // the number of bytes read and an error, in case the string is longer than
 | |
| // the input slice
 | |
| func readLengthEncodedString(b []byte) ([]byte, bool, int, error) {
 | |
| 	// Get length
 | |
| 	num, isNull, n := readLengthEncodedInteger(b)
 | |
| 	if num < 1 {
 | |
| 		return b[n:n], isNull, n, nil
 | |
| 	}
 | |
| 
 | |
| 	n += int(num)
 | |
| 
 | |
| 	// Check data length
 | |
| 	if len(b) >= n {
 | |
| 		return b[n-int(num) : n : n], false, n, nil
 | |
| 	}
 | |
| 	return nil, false, n, io.EOF
 | |
| }
 | |
| 
 | |
| // returns the number of bytes skipped and an error, in case the string is
 | |
| // longer than the input slice
 | |
| func skipLengthEncodedString(b []byte) (int, error) {
 | |
| 	// Get length
 | |
| 	num, _, n := readLengthEncodedInteger(b)
 | |
| 	if num < 1 {
 | |
| 		return n, nil
 | |
| 	}
 | |
| 
 | |
| 	n += int(num)
 | |
| 
 | |
| 	// Check data length
 | |
| 	if len(b) >= n {
 | |
| 		return n, nil
 | |
| 	}
 | |
| 	return n, io.EOF
 | |
| }
 | |
| 
 | |
| // returns the number read, whether the value is NULL and the number of bytes read
 | |
| func readLengthEncodedInteger(b []byte) (uint64, bool, int) {
 | |
| 	// See issue #349
 | |
| 	if len(b) == 0 {
 | |
| 		return 0, true, 1
 | |
| 	}
 | |
| 
 | |
| 	switch b[0] {
 | |
| 	// 251: NULL
 | |
| 	case 0xfb:
 | |
| 		return 0, true, 1
 | |
| 
 | |
| 	// 252: value of following 2
 | |
| 	case 0xfc:
 | |
| 		return uint64(b[1]) | uint64(b[2])<<8, false, 3
 | |
| 
 | |
| 	// 253: value of following 3
 | |
| 	case 0xfd:
 | |
| 		return uint64(b[1]) | uint64(b[2])<<8 | uint64(b[3])<<16, false, 4
 | |
| 
 | |
| 	// 254: value of following 8
 | |
| 	case 0xfe:
 | |
| 		return uint64(b[1]) | uint64(b[2])<<8 | uint64(b[3])<<16 |
 | |
| 				uint64(b[4])<<24 | uint64(b[5])<<32 | uint64(b[6])<<40 |
 | |
| 				uint64(b[7])<<48 | uint64(b[8])<<56,
 | |
| 			false, 9
 | |
| 	}
 | |
| 
 | |
| 	// 0-250: value of first byte
 | |
| 	return uint64(b[0]), false, 1
 | |
| }
 | |
| 
 | |
| // encodes a uint64 value and appends it to the given bytes slice
 | |
| func appendLengthEncodedInteger(b []byte, n uint64) []byte {
 | |
| 	switch {
 | |
| 	case n <= 250:
 | |
| 		return append(b, byte(n))
 | |
| 
 | |
| 	case n <= 0xffff:
 | |
| 		return append(b, 0xfc, byte(n), byte(n>>8))
 | |
| 
 | |
| 	case n <= 0xffffff:
 | |
| 		return append(b, 0xfd, byte(n), byte(n>>8), byte(n>>16))
 | |
| 	}
 | |
| 	return append(b, 0xfe, byte(n), byte(n>>8), byte(n>>16), byte(n>>24),
 | |
| 		byte(n>>32), byte(n>>40), byte(n>>48), byte(n>>56))
 | |
| }
 | |
| 
 | |
| // reserveBuffer checks cap(buf) and expand buffer to len(buf) + appendSize.
 | |
| // If cap(buf) is not enough, reallocate new buffer.
 | |
| func reserveBuffer(buf []byte, appendSize int) []byte {
 | |
| 	newSize := len(buf) + appendSize
 | |
| 	if cap(buf) < newSize {
 | |
| 		// Grow buffer exponentially
 | |
| 		newBuf := make([]byte, len(buf)*2+appendSize)
 | |
| 		copy(newBuf, buf)
 | |
| 		buf = newBuf
 | |
| 	}
 | |
| 	return buf[:newSize]
 | |
| }
 | |
| 
 | |
| // escapeBytesBackslash escapes []byte with backslashes (\)
 | |
| // This escapes the contents of a string (provided as []byte) by adding backslashes before special
 | |
| // characters, and turning others into specific escape sequences, such as
 | |
| // turning newlines into \n and null bytes into \0.
 | |
| // https://github.com/mysql/mysql-server/blob/mysql-5.7.5/mysys/charset.c#L823-L932
 | |
| func escapeBytesBackslash(buf, v []byte) []byte {
 | |
| 	pos := len(buf)
 | |
| 	buf = reserveBuffer(buf, len(v)*2)
 | |
| 
 | |
| 	for _, c := range v {
 | |
| 		switch c {
 | |
| 		case '\x00':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = '0'
 | |
| 			pos += 2
 | |
| 		case '\n':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = 'n'
 | |
| 			pos += 2
 | |
| 		case '\r':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = 'r'
 | |
| 			pos += 2
 | |
| 		case '\x1a':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = 'Z'
 | |
| 			pos += 2
 | |
| 		case '\'':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = '\''
 | |
| 			pos += 2
 | |
| 		case '"':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = '"'
 | |
| 			pos += 2
 | |
| 		case '\\':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = '\\'
 | |
| 			pos += 2
 | |
| 		default:
 | |
| 			buf[pos] = c
 | |
| 			pos++
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return buf[:pos]
 | |
| }
 | |
| 
 | |
| // escapeStringBackslash is similar to escapeBytesBackslash but for string.
 | |
| func escapeStringBackslash(buf []byte, v string) []byte {
 | |
| 	pos := len(buf)
 | |
| 	buf = reserveBuffer(buf, len(v)*2)
 | |
| 
 | |
| 	for i := 0; i < len(v); i++ {
 | |
| 		c := v[i]
 | |
| 		switch c {
 | |
| 		case '\x00':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = '0'
 | |
| 			pos += 2
 | |
| 		case '\n':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = 'n'
 | |
| 			pos += 2
 | |
| 		case '\r':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = 'r'
 | |
| 			pos += 2
 | |
| 		case '\x1a':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = 'Z'
 | |
| 			pos += 2
 | |
| 		case '\'':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = '\''
 | |
| 			pos += 2
 | |
| 		case '"':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = '"'
 | |
| 			pos += 2
 | |
| 		case '\\':
 | |
| 			buf[pos] = '\\'
 | |
| 			buf[pos+1] = '\\'
 | |
| 			pos += 2
 | |
| 		default:
 | |
| 			buf[pos] = c
 | |
| 			pos++
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return buf[:pos]
 | |
| }
 | |
| 
 | |
| // escapeBytesQuotes escapes apostrophes in []byte by doubling them up.
 | |
| // This escapes the contents of a string by doubling up any apostrophes that
 | |
| // it contains. This is used when the NO_BACKSLASH_ESCAPES SQL_MODE is in
 | |
| // effect on the server.
 | |
| // https://github.com/mysql/mysql-server/blob/mysql-5.7.5/mysys/charset.c#L963-L1038
 | |
| func escapeBytesQuotes(buf, v []byte) []byte {
 | |
| 	pos := len(buf)
 | |
| 	buf = reserveBuffer(buf, len(v)*2)
 | |
| 
 | |
| 	for _, c := range v {
 | |
| 		if c == '\'' {
 | |
| 			buf[pos] = '\''
 | |
| 			buf[pos+1] = '\''
 | |
| 			pos += 2
 | |
| 		} else {
 | |
| 			buf[pos] = c
 | |
| 			pos++
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return buf[:pos]
 | |
| }
 | |
| 
 | |
| // escapeStringQuotes is similar to escapeBytesQuotes but for string.
 | |
| func escapeStringQuotes(buf []byte, v string) []byte {
 | |
| 	pos := len(buf)
 | |
| 	buf = reserveBuffer(buf, len(v)*2)
 | |
| 
 | |
| 	for i := 0; i < len(v); i++ {
 | |
| 		c := v[i]
 | |
| 		if c == '\'' {
 | |
| 			buf[pos] = '\''
 | |
| 			buf[pos+1] = '\''
 | |
| 			pos += 2
 | |
| 		} else {
 | |
| 			buf[pos] = c
 | |
| 			pos++
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return buf[:pos]
 | |
| }
 | |
| 
 | |
| /******************************************************************************
 | |
| *                               Sync utils                                    *
 | |
| ******************************************************************************/
 | |
| 
 | |
| // noCopy may be embedded into structs which must not be copied
 | |
| // after the first use.
 | |
| //
 | |
| // See https://github.com/golang/go/issues/8005#issuecomment-190753527
 | |
| // for details.
 | |
| type noCopy struct{}
 | |
| 
 | |
| // Lock is a no-op used by -copylocks checker from `go vet`.
 | |
| func (*noCopy) Lock() {}
 | |
| 
 | |
| // atomicBool is a wrapper around uint32 for usage as a boolean value with
 | |
| // atomic access.
 | |
| type atomicBool struct {
 | |
| 	_noCopy noCopy
 | |
| 	value   uint32
 | |
| }
 | |
| 
 | |
| // IsSet returns whether the current boolean value is true
 | |
| func (ab *atomicBool) IsSet() bool {
 | |
| 	return atomic.LoadUint32(&ab.value) > 0
 | |
| }
 | |
| 
 | |
| // Set sets the value of the bool regardless of the previous value
 | |
| func (ab *atomicBool) Set(value bool) {
 | |
| 	if value {
 | |
| 		atomic.StoreUint32(&ab.value, 1)
 | |
| 	} else {
 | |
| 		atomic.StoreUint32(&ab.value, 0)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // TrySet sets the value of the bool and returns whether the value changed
 | |
| func (ab *atomicBool) TrySet(value bool) bool {
 | |
| 	if value {
 | |
| 		return atomic.SwapUint32(&ab.value, 1) == 0
 | |
| 	}
 | |
| 	return atomic.SwapUint32(&ab.value, 0) > 0
 | |
| }
 | |
| 
 | |
| // atomicError is a wrapper for atomically accessed error values
 | |
| type atomicError struct {
 | |
| 	_noCopy noCopy
 | |
| 	value   atomic.Value
 | |
| }
 | |
| 
 | |
| // Set sets the error value regardless of the previous value.
 | |
| // The value must not be nil
 | |
| func (ae *atomicError) Set(value error) {
 | |
| 	ae.value.Store(value)
 | |
| }
 | |
| 
 | |
| // Value returns the current error value
 | |
| func (ae *atomicError) Value() error {
 | |
| 	if v := ae.value.Load(); v != nil {
 | |
| 		// this will panic if the value doesn't implement the error interface
 | |
| 		return v.(error)
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func namedValueToValue(named []driver.NamedValue) ([]driver.Value, error) {
 | |
| 	dargs := make([]driver.Value, len(named))
 | |
| 	for n, param := range named {
 | |
| 		if len(param.Name) > 0 {
 | |
| 			// TODO: support the use of Named Parameters #561
 | |
| 			return nil, errors.New("mysql: driver does not support the use of Named Parameters")
 | |
| 		}
 | |
| 		dargs[n] = param.Value
 | |
| 	}
 | |
| 	return dargs, nil
 | |
| }
 | |
| 
 | |
| func mapIsolationLevel(level driver.IsolationLevel) (string, error) {
 | |
| 	switch sql.IsolationLevel(level) {
 | |
| 	case sql.LevelRepeatableRead:
 | |
| 		return "REPEATABLE READ", nil
 | |
| 	case sql.LevelReadCommitted:
 | |
| 		return "READ COMMITTED", nil
 | |
| 	case sql.LevelReadUncommitted:
 | |
| 		return "READ UNCOMMITTED", nil
 | |
| 	case sql.LevelSerializable:
 | |
| 		return "SERIALIZABLE", nil
 | |
| 	default:
 | |
| 		return "", fmt.Errorf("mysql: unsupported isolation level: %v", level)
 | |
| 	}
 | |
| }
 |