Enables mssql support (#383)

* Enables mssql support

Port of dlobs work in gogs.
Enables options in index.js
Enables MSSQL as a database option in go.
Sets ID to 0 on initial migration. Required for
MSSQL insert statements.

Signed-off-by: Beau Trepp <beautrepp@gmail.com>

* Vendors in denisenkom/go-mssqldb

Includes golang.org/x/crypto/md4
as this is required by go-msssqldb

Signed-off-by: Beau Trepp <beautrepp@gmail.com>

1 files changed, 561 insertions, 0 deletions

diff --git a/vendor/github.com/denisenkom/go-mssqldb/token.go b/vendor/github.com/denisenkom/go-mssqldb/token.go
new file mode 100644
index 0000000000..f20bd14cc9
--- /dev/null
+++ b/vendor/github.com/denisenkom/go-mssqldb/token.go
@@ -0,0 +1,561 @@
+package mssql
+
+import (
+	"encoding/binary"
+	"io"
+	"strconv"
+	"strings"
+)
+
+// token ids
+const (
+	tokenReturnStatus = 121 // 0x79
+	tokenColMetadata  = 129 // 0x81
+	tokenOrder        = 169 // 0xA9
+	tokenError        = 170 // 0xAA
+	tokenInfo         = 171 // 0xAB
+	tokenLoginAck     = 173 // 0xad
+	tokenRow          = 209 // 0xd1
+	tokenNbcRow       = 210 // 0xd2
+	tokenEnvChange    = 227 // 0xE3
+	tokenSSPI         = 237 // 0xED
+	tokenDone         = 253 // 0xFD
+	tokenDoneProc     = 254
+	tokenDoneInProc   = 255
+)
+
+// done flags
+const (
+	doneFinal    = 0
+	doneMore     = 1
+	doneError    = 2
+	doneInxact   = 4
+	doneCount    = 0x10
+	doneAttn     = 0x20
+	doneSrvError = 0x100
+)
+
+// ENVCHANGE types
+// http://msdn.microsoft.com/en-us/library/dd303449.aspx
+const (
+	envTypDatabase           = 1
+	envTypLanguage           = 2
+	envTypCharset            = 3
+	envTypPacketSize         = 4
+	envSortId                = 5
+	envSortFlags             = 6
+	envSqlCollation          = 7
+	envTypBeginTran          = 8
+	envTypCommitTran         = 9
+	envTypRollbackTran       = 10
+	envEnlistDTC             = 11
+	envDefectTran            = 12
+	envDatabaseMirrorPartner = 13
+	envPromoteTran           = 15
+	envTranMgrAddr           = 16
+	envTranEnded             = 17
+	envResetConnAck          = 18
+	envStartedInstanceName   = 19
+	envRouting               = 20
+)
+
+// interface for all tokens
+type tokenStruct interface{}
+
+type orderStruct struct {
+	ColIds []uint16
+}
+
+type doneStruct struct {
+	Status   uint16
+	CurCmd   uint16
+	RowCount uint64
+}
+
+type doneInProcStruct doneStruct
+
+var doneFlags2str = map[uint16]string{
+	doneFinal:    "final",
+	doneMore:     "more",
+	doneError:    "error",
+	doneInxact:   "inxact",
+	doneCount:    "count",
+	doneAttn:     "attn",
+	doneSrvError: "srverror",
+}
+
+func doneFlags2Str(flags uint16) string {
+	strs := make([]string, 0, len(doneFlags2str))
+	for flag, tag := range doneFlags2str {
+		if flags&flag != 0 {
+			strs = append(strs, tag)
+		}
+	}
+	return strings.Join(strs, "|")
+}
+
+// ENVCHANGE stream
+// http://msdn.microsoft.com/en-us/library/dd303449.aspx
+func processEnvChg(sess *tdsSession) {
+	size := sess.buf.uint16()
+	r := &io.LimitedReader{R: sess.buf, N: int64(size)}
+	for {
+		var err error
+		var envtype uint8
+		err = binary.Read(r, binary.LittleEndian, &envtype)
+		if err == io.EOF {
+			return
+		}
+		if err != nil {
+			badStreamPanic(err)
+		}
+		switch envtype {
+		case envTypDatabase:
+			sess.database, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			_, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+		case envTypLanguage:
+			//currently ignored
+			// old value
+			_, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			// new value
+			_, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+		case envTypCharset:
+			//currently ignored
+			// old value
+			_, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			// new value
+			_, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+		case envTypPacketSize:
+			packetsize, err := readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			_, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			packetsizei, err := strconv.Atoi(packetsize)
+			if err != nil {
+				badStreamPanicf("Invalid Packet size value returned from server (%s): %s", packetsize, err.Error())
+			}
+			if len(sess.buf.buf) != packetsizei {
+				newbuf := make([]byte, packetsizei)
+				copy(newbuf, sess.buf.buf)
+				sess.buf.buf = newbuf
+			}
+		case envSortId:
+			// currently ignored
+			// old value, should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// new value
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envSortFlags:
+			// currently ignored
+			// old value, should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// new value
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envSqlCollation:
+			// currently ignored
+			// old value
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// new value
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envTypBeginTran:
+			tranid, err := readBVarByte(r)
+			if len(tranid) != 8 {
+				badStreamPanicf("invalid size of transaction identifier: %d", len(tranid))
+			}
+			sess.tranid = binary.LittleEndian.Uint64(tranid)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			if sess.logFlags&logTransaction != 0 {
+				sess.log.Printf("BEGIN TRANSACTION %x\n", sess.tranid)
+			}
+			_, err = readBVarByte(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+		case envTypCommitTran, envTypRollbackTran:
+			_, err = readBVarByte(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			_, err = readBVarByte(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			if sess.logFlags&logTransaction != 0 {
+				if envtype == envTypCommitTran {
+					sess.log.Printf("COMMIT TRANSACTION %x\n", sess.tranid)
+				} else {
+					sess.log.Printf("ROLLBACK TRANSACTION %x\n", sess.tranid)
+				}
+			}
+			sess.tranid = 0
+		case envEnlistDTC:
+			// currently ignored
+			// old value
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// new value, should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envDefectTran:
+			// currently ignored
+			// old value, should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// new value
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envDatabaseMirrorPartner:
+			sess.partner, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			_, err = readBVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+		case envPromoteTran:
+			// currently ignored
+			// old value, should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// dtc token
+			// spec says it should be L_VARBYTE, so this code might be wrong
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envTranMgrAddr:
+			// currently ignored
+			// old value, should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// XACT_MANAGER_ADDRESS = B_VARBYTE
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envTranEnded:
+			// currently ignored
+			// old value, B_VARBYTE
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envResetConnAck:
+			// currently ignored
+			// old value, should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envStartedInstanceName:
+			// currently ignored
+			// old value, should be 0
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+			// instance name
+			if _, err = readBVarChar(r); err != nil {
+				badStreamPanic(err)
+			}
+		case envRouting:
+			// RoutingData message is:
+			// ValueLength                 USHORT
+			// Protocol (TCP = 0)          BYTE
+			// ProtocolProperty (new port) USHORT
+			// AlternateServer             US_VARCHAR
+			_, err := readUshort(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			protocol, err := readByte(r)
+			if err != nil || protocol != 0 {
+				badStreamPanic(err)
+			}
+			newPort, err := readUshort(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			newServer, err := readUsVarChar(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			// consume the OLDVALUE = %x00 %x00
+			_, err = readUshort(r)
+			if err != nil {
+				badStreamPanic(err)
+			}
+			sess.routedServer = newServer
+			sess.routedPort = newPort
+		default:
+			// ignore rest of records because we don't know how to skip those
+			sess.log.Printf("WARN: Unknown ENVCHANGE record detected with type id = %d\n", envtype)
+			break
+		}
+
+	}
+}
+
+type returnStatus int32
+
+// http://msdn.microsoft.com/en-us/library/dd358180.aspx
+func parseReturnStatus(r *tdsBuffer) returnStatus {
+	return returnStatus(r.int32())
+}
+
+func parseOrder(r *tdsBuffer) (res orderStruct) {
+	len := int(r.uint16())
+	res.ColIds = make([]uint16, len/2)
+	for i := 0; i < len/2; i++ {
+		res.ColIds[i] = r.uint16()
+	}
+	return res
+}
+
+func parseDone(r *tdsBuffer) (res doneStruct) {
+	res.Status = r.uint16()
+	res.CurCmd = r.uint16()
+	res.RowCount = r.uint64()
+	return res
+}
+
+func parseDoneInProc(r *tdsBuffer) (res doneInProcStruct) {
+	res.Status = r.uint16()
+	res.CurCmd = r.uint16()
+	res.RowCount = r.uint64()
+	return res
+}
+
+type sspiMsg []byte
+
+func parseSSPIMsg(r *tdsBuffer) sspiMsg {
+	size := r.uint16()
+	buf := make([]byte, size)
+	r.ReadFull(buf)
+	return sspiMsg(buf)
+}
+
+type loginAckStruct struct {
+	Interface  uint8
+	TDSVersion uint32
+	ProgName   string
+	ProgVer    uint32
+}
+
+func parseLoginAck(r *tdsBuffer) loginAckStruct {
+	size := r.uint16()
+	buf := make([]byte, size)
+	r.ReadFull(buf)
+	var res loginAckStruct
+	res.Interface = buf[0]
+	res.TDSVersion = binary.BigEndian.Uint32(buf[1:])
+	prognamelen := buf[1+4]
+	var err error
+	if res.ProgName, err = ucs22str(buf[1+4+1 : 1+4+1+prognamelen*2]); err != nil {
+		badStreamPanic(err)
+	}
+	res.ProgVer = binary.BigEndian.Uint32(buf[size-4:])
+	return res
+}
+
+// http://msdn.microsoft.com/en-us/library/dd357363.aspx
+func parseColMetadata72(r *tdsBuffer) (columns []columnStruct) {
+	count := r.uint16()
+	if count == 0xffff {
+		// no metadata is sent
+		return nil
+	}
+	columns = make([]columnStruct, count)
+	for i := range columns {
+		column := &columns[i]
+		column.UserType = r.uint32()
+		column.Flags = r.uint16()
+
+		// parsing TYPE_INFO structure
+		column.ti = readTypeInfo(r)
+		column.ColName = r.BVarChar()
+	}
+	return columns
+}
+
+// http://msdn.microsoft.com/en-us/library/dd357254.aspx
+func parseRow(r *tdsBuffer, columns []columnStruct, row []interface{}) {
+	for i, column := range columns {
+		row[i] = column.ti.Reader(&column.ti, r)
+	}
+}
+
+// http://msdn.microsoft.com/en-us/library/dd304783.aspx
+func parseNbcRow(r *tdsBuffer, columns []columnStruct, row []interface{}) {
+	bitlen := (len(columns) + 7) / 8
+	pres := make([]byte, bitlen)
+	r.ReadFull(pres)
+	for i, col := range columns {
+		if pres[i/8]&(1<<(uint(i)%8)) != 0 {
+			row[i] = nil
+			continue
+		}
+		row[i] = col.ti.Reader(&col.ti, r)
+	}
+}
+
+// http://msdn.microsoft.com/en-us/library/dd304156.aspx
+func parseError72(r *tdsBuffer) (res Error) {
+	length := r.uint16()
+	_ = length // ignore length
+	res.Number = r.int32()
+	res.State = r.byte()
+	res.Class = r.byte()
+	res.Message = r.UsVarChar()
+	res.ServerName = r.BVarChar()
+	res.ProcName = r.BVarChar()
+	res.LineNo = r.int32()
+	return
+}
+
+// http://msdn.microsoft.com/en-us/library/dd304156.aspx
+func parseInfo(r *tdsBuffer) (res Error) {
+	length := r.uint16()
+	_ = length // ignore length
+	res.Number = r.int32()
+	res.State = r.byte()
+	res.Class = r.byte()
+	res.Message = r.UsVarChar()
+	res.ServerName = r.BVarChar()
+	res.ProcName = r.BVarChar()
+	res.LineNo = r.int32()
+	return
+}
+
+func processResponse(sess *tdsSession, ch chan tokenStruct) {
+	defer func() {
+		if err := recover(); err != nil {
+			ch <- err
+		}
+		close(ch)
+	}()
+	packet_type, err := sess.buf.BeginRead()
+	if err != nil {
+		ch <- err
+		return
+	}
+	if packet_type != packReply {
+		badStreamPanicf("invalid response packet type, expected REPLY, actual: %d", packet_type)
+	}
+	var columns []columnStruct
+	var lastError Error
+	var failed bool
+	for {
+		token := sess.buf.byte()
+		switch token {
+		case tokenSSPI:
+			ch <- parseSSPIMsg(sess.buf)
+			return
+		case tokenReturnStatus:
+			returnStatus := parseReturnStatus(sess.buf)
+			ch <- returnStatus
+		case tokenLoginAck:
+			loginAck := parseLoginAck(sess.buf)
+			ch <- loginAck
+		case tokenOrder:
+			order := parseOrder(sess.buf)
+			ch <- order
+		case tokenDoneInProc:
+			done := parseDoneInProc(sess.buf)
+			if sess.logFlags&logRows != 0 && done.Status&doneCount != 0 {
+				sess.log.Printf("(%d row(s) affected)\n", done.RowCount)
+			}
+			ch <- done
+		case tokenDone, tokenDoneProc:
+			done := parseDone(sess.buf)
+			if sess.logFlags&logRows != 0 && done.Status&doneCount != 0 {
+				sess.log.Printf("(%d row(s) affected)\n", done.RowCount)
+			}
+			if done.Status&doneError != 0 || failed {
+				ch <- lastError
+				return
+			}
+			if done.Status&doneSrvError != 0 {
+				lastError.Message = "Server Error"
+				ch <- lastError
+				return
+			}
+			ch <- done
+			if done.Status&doneMore == 0 {
+				return
+			}
+		case tokenColMetadata:
+			columns = parseColMetadata72(sess.buf)
+			ch <- columns
+		case tokenRow:
+			row := make([]interface{}, len(columns))
+			parseRow(sess.buf, columns, row)
+			ch <- row
+		case tokenNbcRow:
+			row := make([]interface{}, len(columns))
+			parseNbcRow(sess.buf, columns, row)
+			ch <- row
+		case tokenEnvChange:
+			processEnvChg(sess)
+		case tokenError:
+			lastError = parseError72(sess.buf)
+			failed = true
+			if sess.logFlags&logErrors != 0 {
+				sess.log.Println(lastError.Message)
+			}
+		case tokenInfo:
+			info := parseInfo(sess.buf)
+			if sess.logFlags&logMessages != 0 {
+				sess.log.Println(info.Message)
+			}
+		default:
+			badStreamPanicf("Unknown token type: %d", token)
+		}
+	}
+}