frame.go 45 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099
  1. // Copyright (c) 2012 The gocql Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package gocql
  5. import (
  6. "context"
  7. "errors"
  8. "fmt"
  9. "io"
  10. "io/ioutil"
  11. "net"
  12. "runtime"
  13. "strings"
  14. "time"
  15. )
  16. type unsetColumn struct{}
  17. // UnsetValue represents a value used in a query binding that will be ignored by Cassandra.
  18. //
  19. // By setting a field to the unset value Cassandra will ignore the write completely.
  20. // The main advantage is the ability to keep the same prepared statement even when you don't
  21. // want to update some fields, where before you needed to make another prepared statement.
  22. //
  23. // UnsetValue is only available when using the version 4 of the protocol.
  24. var UnsetValue = unsetColumn{}
  25. type namedValue struct {
  26. name string
  27. value interface{}
  28. }
  29. // NamedValue produce a value which will bind to the named parameter in a query
  30. func NamedValue(name string, value interface{}) interface{} {
  31. return &namedValue{
  32. name: name,
  33. value: value,
  34. }
  35. }
  36. const (
  37. protoDirectionMask = 0x80
  38. protoVersionMask = 0x7F
  39. protoVersion1 = 0x01
  40. protoVersion2 = 0x02
  41. protoVersion3 = 0x03
  42. protoVersion4 = 0x04
  43. protoVersion5 = 0x05
  44. maxFrameSize = 256 * 1024 * 1024
  45. )
  46. type protoVersion byte
  47. func (p protoVersion) request() bool {
  48. return p&protoDirectionMask == 0x00
  49. }
  50. func (p protoVersion) response() bool {
  51. return p&protoDirectionMask == 0x80
  52. }
  53. func (p protoVersion) version() byte {
  54. return byte(p) & protoVersionMask
  55. }
  56. func (p protoVersion) String() string {
  57. dir := "REQ"
  58. if p.response() {
  59. dir = "RESP"
  60. }
  61. return fmt.Sprintf("[version=%d direction=%s]", p.version(), dir)
  62. }
  63. type frameOp byte
  64. const (
  65. // header ops
  66. opError frameOp = 0x00
  67. opStartup frameOp = 0x01
  68. opReady frameOp = 0x02
  69. opAuthenticate frameOp = 0x03
  70. opOptions frameOp = 0x05
  71. opSupported frameOp = 0x06
  72. opQuery frameOp = 0x07
  73. opResult frameOp = 0x08
  74. opPrepare frameOp = 0x09
  75. opExecute frameOp = 0x0A
  76. opRegister frameOp = 0x0B
  77. opEvent frameOp = 0x0C
  78. opBatch frameOp = 0x0D
  79. opAuthChallenge frameOp = 0x0E
  80. opAuthResponse frameOp = 0x0F
  81. opAuthSuccess frameOp = 0x10
  82. )
  83. func (f frameOp) String() string {
  84. switch f {
  85. case opError:
  86. return "ERROR"
  87. case opStartup:
  88. return "STARTUP"
  89. case opReady:
  90. return "READY"
  91. case opAuthenticate:
  92. return "AUTHENTICATE"
  93. case opOptions:
  94. return "OPTIONS"
  95. case opSupported:
  96. return "SUPPORTED"
  97. case opQuery:
  98. return "QUERY"
  99. case opResult:
  100. return "RESULT"
  101. case opPrepare:
  102. return "PREPARE"
  103. case opExecute:
  104. return "EXECUTE"
  105. case opRegister:
  106. return "REGISTER"
  107. case opEvent:
  108. return "EVENT"
  109. case opBatch:
  110. return "BATCH"
  111. case opAuthChallenge:
  112. return "AUTH_CHALLENGE"
  113. case opAuthResponse:
  114. return "AUTH_RESPONSE"
  115. case opAuthSuccess:
  116. return "AUTH_SUCCESS"
  117. default:
  118. return fmt.Sprintf("UNKNOWN_OP_%d", f)
  119. }
  120. }
  121. const (
  122. // result kind
  123. resultKindVoid = 1
  124. resultKindRows = 2
  125. resultKindKeyspace = 3
  126. resultKindPrepared = 4
  127. resultKindSchemaChanged = 5
  128. // rows flags
  129. flagGlobalTableSpec int = 0x01
  130. flagHasMorePages int = 0x02
  131. flagNoMetaData int = 0x04
  132. // query flags
  133. flagValues byte = 0x01
  134. flagSkipMetaData byte = 0x02
  135. flagPageSize byte = 0x04
  136. flagWithPagingState byte = 0x08
  137. flagWithSerialConsistency byte = 0x10
  138. flagDefaultTimestamp byte = 0x20
  139. flagWithNameValues byte = 0x40
  140. flagWithKeyspace byte = 0x80
  141. // prepare flags
  142. flagWithPreparedKeyspace uint32 = 0x01
  143. // header flags
  144. flagCompress byte = 0x01
  145. flagTracing byte = 0x02
  146. flagCustomPayload byte = 0x04
  147. flagWarning byte = 0x08
  148. flagBetaProtocol byte = 0x10
  149. )
  150. type Consistency uint16
  151. const (
  152. Any Consistency = 0x00
  153. One Consistency = 0x01
  154. Two Consistency = 0x02
  155. Three Consistency = 0x03
  156. Quorum Consistency = 0x04
  157. All Consistency = 0x05
  158. LocalQuorum Consistency = 0x06
  159. EachQuorum Consistency = 0x07
  160. LocalOne Consistency = 0x0A
  161. )
  162. func (c Consistency) String() string {
  163. switch c {
  164. case Any:
  165. return "ANY"
  166. case One:
  167. return "ONE"
  168. case Two:
  169. return "TWO"
  170. case Three:
  171. return "THREE"
  172. case Quorum:
  173. return "QUORUM"
  174. case All:
  175. return "ALL"
  176. case LocalQuorum:
  177. return "LOCAL_QUORUM"
  178. case EachQuorum:
  179. return "EACH_QUORUM"
  180. case LocalOne:
  181. return "LOCAL_ONE"
  182. default:
  183. return fmt.Sprintf("UNKNOWN_CONS_0x%x", uint16(c))
  184. }
  185. }
  186. func (c Consistency) MarshalText() (text []byte, err error) {
  187. return []byte(c.String()), nil
  188. }
  189. func (c *Consistency) UnmarshalText(text []byte) error {
  190. switch string(text) {
  191. case "ANY":
  192. *c = Any
  193. case "ONE":
  194. *c = One
  195. case "TWO":
  196. *c = Two
  197. case "THREE":
  198. *c = Three
  199. case "QUORUM":
  200. *c = Quorum
  201. case "ALL":
  202. *c = All
  203. case "LOCAL_QUORUM":
  204. *c = LocalQuorum
  205. case "EACH_QUORUM":
  206. *c = EachQuorum
  207. case "LOCAL_ONE":
  208. *c = LocalOne
  209. default:
  210. return fmt.Errorf("invalid consistency %q", string(text))
  211. }
  212. return nil
  213. }
  214. func ParseConsistency(s string) Consistency {
  215. var c Consistency
  216. if err := c.UnmarshalText([]byte(strings.ToUpper(s))); err != nil {
  217. panic(err)
  218. }
  219. return c
  220. }
  221. // ParseConsistencyWrapper wraps gocql.ParseConsistency to provide an err
  222. // return instead of a panic
  223. func ParseConsistencyWrapper(s string) (consistency Consistency, err error) {
  224. err = consistency.UnmarshalText([]byte(strings.ToUpper(s)))
  225. return
  226. }
  227. // MustParseConsistency is the same as ParseConsistency except it returns
  228. // an error (never). It is kept here since breaking changes are not good.
  229. // DEPRECATED: use ParseConsistency if you want a panic on parse error.
  230. func MustParseConsistency(s string) (Consistency, error) {
  231. c, err := ParseConsistencyWrapper(s)
  232. if err != nil {
  233. panic(err)
  234. }
  235. return c, nil
  236. }
  237. type SerialConsistency uint16
  238. const (
  239. Serial SerialConsistency = 0x08
  240. LocalSerial SerialConsistency = 0x09
  241. )
  242. func (s SerialConsistency) String() string {
  243. switch s {
  244. case Serial:
  245. return "SERIAL"
  246. case LocalSerial:
  247. return "LOCAL_SERIAL"
  248. default:
  249. return fmt.Sprintf("UNKNOWN_SERIAL_CONS_0x%x", uint16(s))
  250. }
  251. }
  252. func (s SerialConsistency) MarshalText() (text []byte, err error) {
  253. return []byte(s.String()), nil
  254. }
  255. func (s *SerialConsistency) UnmarshalText(text []byte) error {
  256. switch string(text) {
  257. case "SERIAL":
  258. *s = Serial
  259. case "LOCAL_SERIAL":
  260. *s = LocalSerial
  261. default:
  262. return fmt.Errorf("invalid consistency %q", string(text))
  263. }
  264. return nil
  265. }
  266. const (
  267. apacheCassandraTypePrefix = "org.apache.cassandra.db.marshal."
  268. )
  269. var (
  270. ErrFrameTooBig = errors.New("frame length is bigger than the maximum allowed")
  271. )
  272. const maxFrameHeaderSize = 9
  273. func writeInt(p []byte, n int32) {
  274. p[0] = byte(n >> 24)
  275. p[1] = byte(n >> 16)
  276. p[2] = byte(n >> 8)
  277. p[3] = byte(n)
  278. }
  279. func readInt(p []byte) int32 {
  280. return int32(p[0])<<24 | int32(p[1])<<16 | int32(p[2])<<8 | int32(p[3])
  281. }
  282. func writeShort(p []byte, n uint16) {
  283. p[0] = byte(n >> 8)
  284. p[1] = byte(n)
  285. }
  286. func readShort(p []byte) uint16 {
  287. return uint16(p[0])<<8 | uint16(p[1])
  288. }
  289. type frameHeader struct {
  290. version protoVersion
  291. flags byte
  292. stream int
  293. op frameOp
  294. length int
  295. warnings []string
  296. }
  297. func (f frameHeader) String() string {
  298. return fmt.Sprintf("[header version=%s flags=0x%x stream=%d op=%s length=%d]", f.version, f.flags, f.stream, f.op, f.length)
  299. }
  300. func (f frameHeader) Header() frameHeader {
  301. return f
  302. }
  303. const defaultBufSize = 128
  304. type ObservedFrameHeader struct {
  305. Version protoVersion
  306. Flags byte
  307. Stream int16
  308. Opcode frameOp
  309. Length int32
  310. // StartHeader is the time we started reading the frame header off the network connection.
  311. Start time.Time
  312. // EndHeader is the time we finished reading the frame header off the network connection.
  313. End time.Time
  314. }
  315. func (f ObservedFrameHeader) String() string {
  316. return fmt.Sprintf("[observed header version=%s flags=0x%x stream=%d op=%s length=%d]", f.Version, f.Flags, f.Stream, f.Opcode, f.Length)
  317. }
  318. // FrameHeaderObserver is the interface implemented by frame observers / stat collectors.
  319. //
  320. // Experimental, this interface and use may change
  321. type FrameHeaderObserver interface {
  322. // ObserveFrameHeader gets called on every received frame header.
  323. ObserveFrameHeader(context.Context, ObservedFrameHeader)
  324. }
  325. // a framer is responsible for reading, writing and parsing frames on a single stream
  326. type framer struct {
  327. r io.Reader
  328. w io.Writer
  329. proto byte
  330. // flags are for outgoing flags, enabling compression and tracing etc
  331. flags byte
  332. compres Compressor
  333. headSize int
  334. // if this frame was read then the header will be here
  335. header *frameHeader
  336. // if tracing flag is set this is not nil
  337. traceID []byte
  338. // holds a ref to the whole byte slice for rbuf so that it can be reset to
  339. // 0 after a read.
  340. readBuffer []byte
  341. rbuf []byte
  342. wbuf []byte
  343. customPayload map[string][]byte
  344. }
  345. func newFramer(r io.Reader, w io.Writer, compressor Compressor, version byte) *framer {
  346. f := &framer{
  347. wbuf: make([]byte, defaultBufSize),
  348. readBuffer: make([]byte, defaultBufSize),
  349. }
  350. var flags byte
  351. if compressor != nil {
  352. flags |= flagCompress
  353. }
  354. if version == protoVersion5 {
  355. flags |= flagBetaProtocol
  356. }
  357. version &= protoVersionMask
  358. headSize := 8
  359. if version > protoVersion2 {
  360. headSize = 9
  361. }
  362. f.compres = compressor
  363. f.proto = version
  364. f.flags = flags
  365. f.headSize = headSize
  366. f.r = r
  367. f.rbuf = f.readBuffer[:0]
  368. f.w = w
  369. f.wbuf = f.wbuf[:0]
  370. f.header = nil
  371. f.traceID = nil
  372. return f
  373. }
  374. type frame interface {
  375. Header() frameHeader
  376. }
  377. func readHeader(r io.Reader, p []byte) (head frameHeader, err error) {
  378. _, err = io.ReadFull(r, p[:1])
  379. if err != nil {
  380. return frameHeader{}, err
  381. }
  382. version := p[0] & protoVersionMask
  383. if version < protoVersion1 || version > protoVersion5 {
  384. return frameHeader{}, fmt.Errorf("gocql: unsupported protocol response version: %d", version)
  385. }
  386. headSize := 9
  387. if version < protoVersion3 {
  388. headSize = 8
  389. }
  390. _, err = io.ReadFull(r, p[1:headSize])
  391. if err != nil {
  392. return frameHeader{}, err
  393. }
  394. p = p[:headSize]
  395. head.version = protoVersion(p[0])
  396. head.flags = p[1]
  397. if version > protoVersion2 {
  398. if len(p) != 9 {
  399. return frameHeader{}, fmt.Errorf("not enough bytes to read header require 9 got: %d", len(p))
  400. }
  401. head.stream = int(int16(p[2])<<8 | int16(p[3]))
  402. head.op = frameOp(p[4])
  403. head.length = int(readInt(p[5:]))
  404. } else {
  405. if len(p) != 8 {
  406. return frameHeader{}, fmt.Errorf("not enough bytes to read header require 8 got: %d", len(p))
  407. }
  408. head.stream = int(int8(p[2]))
  409. head.op = frameOp(p[3])
  410. head.length = int(readInt(p[4:]))
  411. }
  412. return head, nil
  413. }
  414. // explicitly enables tracing for the framers outgoing requests
  415. func (f *framer) trace() {
  416. f.flags |= flagTracing
  417. }
  418. // explicitly enables the custom payload flag
  419. func (f *framer) payload() {
  420. f.flags |= flagCustomPayload
  421. }
  422. // reads a frame form the wire into the framers buffer
  423. func (f *framer) readFrame(head *frameHeader) error {
  424. if head.length < 0 {
  425. return fmt.Errorf("frame body length can not be less than 0: %d", head.length)
  426. } else if head.length > maxFrameSize {
  427. // need to free up the connection to be used again
  428. _, err := io.CopyN(ioutil.Discard, f.r, int64(head.length))
  429. if err != nil {
  430. return fmt.Errorf("error whilst trying to discard frame with invalid length: %v", err)
  431. }
  432. return ErrFrameTooBig
  433. }
  434. if cap(f.readBuffer) >= head.length {
  435. f.rbuf = f.readBuffer[:head.length]
  436. } else {
  437. f.readBuffer = make([]byte, head.length)
  438. f.rbuf = f.readBuffer
  439. }
  440. // assume the underlying reader takes care of timeouts and retries
  441. n, err := io.ReadFull(f.r, f.rbuf)
  442. if err != nil {
  443. return fmt.Errorf("unable to read frame body: read %d/%d bytes: %v", n, head.length, err)
  444. }
  445. if head.flags&flagCompress == flagCompress {
  446. if f.compres == nil {
  447. return NewErrProtocol("no compressor available with compressed frame body")
  448. }
  449. f.rbuf, err = f.compres.Decode(f.rbuf)
  450. if err != nil {
  451. return err
  452. }
  453. }
  454. f.header = head
  455. return nil
  456. }
  457. func (f *framer) parseFrame() (frame frame, err error) {
  458. defer func() {
  459. if r := recover(); r != nil {
  460. if _, ok := r.(runtime.Error); ok {
  461. panic(r)
  462. }
  463. err = r.(error)
  464. }
  465. }()
  466. if f.header.version.request() {
  467. return nil, NewErrProtocol("got a request frame from server: %v", f.header.version)
  468. }
  469. if f.header.flags&flagTracing == flagTracing {
  470. f.readTrace()
  471. }
  472. if f.header.flags&flagWarning == flagWarning {
  473. f.header.warnings = f.readStringList()
  474. }
  475. if f.header.flags&flagCustomPayload == flagCustomPayload {
  476. f.customPayload = f.readBytesMap()
  477. }
  478. // assumes that the frame body has been read into rbuf
  479. switch f.header.op {
  480. case opError:
  481. frame = f.parseErrorFrame()
  482. case opReady:
  483. frame = f.parseReadyFrame()
  484. case opResult:
  485. frame, err = f.parseResultFrame()
  486. case opSupported:
  487. frame = f.parseSupportedFrame()
  488. case opAuthenticate:
  489. frame = f.parseAuthenticateFrame()
  490. case opAuthChallenge:
  491. frame = f.parseAuthChallengeFrame()
  492. case opAuthSuccess:
  493. frame = f.parseAuthSuccessFrame()
  494. case opEvent:
  495. frame = f.parseEventFrame()
  496. default:
  497. return nil, NewErrProtocol("unknown op in frame header: %s", f.header.op)
  498. }
  499. return
  500. }
  501. func (f *framer) parseErrorFrame() frame {
  502. code := f.readInt()
  503. msg := f.readString()
  504. errD := errorFrame{
  505. frameHeader: *f.header,
  506. code: code,
  507. message: msg,
  508. }
  509. switch code {
  510. case errUnavailable:
  511. cl := f.readConsistency()
  512. required := f.readInt()
  513. alive := f.readInt()
  514. return &RequestErrUnavailable{
  515. errorFrame: errD,
  516. Consistency: cl,
  517. Required: required,
  518. Alive: alive,
  519. }
  520. case errWriteTimeout:
  521. cl := f.readConsistency()
  522. received := f.readInt()
  523. blockfor := f.readInt()
  524. writeType := f.readString()
  525. return &RequestErrWriteTimeout{
  526. errorFrame: errD,
  527. Consistency: cl,
  528. Received: received,
  529. BlockFor: blockfor,
  530. WriteType: writeType,
  531. }
  532. case errReadTimeout:
  533. cl := f.readConsistency()
  534. received := f.readInt()
  535. blockfor := f.readInt()
  536. dataPresent := f.readByte()
  537. return &RequestErrReadTimeout{
  538. errorFrame: errD,
  539. Consistency: cl,
  540. Received: received,
  541. BlockFor: blockfor,
  542. DataPresent: dataPresent,
  543. }
  544. case errAlreadyExists:
  545. ks := f.readString()
  546. table := f.readString()
  547. return &RequestErrAlreadyExists{
  548. errorFrame: errD,
  549. Keyspace: ks,
  550. Table: table,
  551. }
  552. case errUnprepared:
  553. stmtId := f.readShortBytes()
  554. return &RequestErrUnprepared{
  555. errorFrame: errD,
  556. StatementId: copyBytes(stmtId), // defensively copy
  557. }
  558. case errReadFailure:
  559. res := &RequestErrReadFailure{
  560. errorFrame: errD,
  561. }
  562. res.Consistency = f.readConsistency()
  563. res.Received = f.readInt()
  564. res.BlockFor = f.readInt()
  565. if f.proto > protoVersion4 {
  566. res.ErrorMap = f.readErrorMap()
  567. res.NumFailures = len(res.ErrorMap)
  568. } else {
  569. res.NumFailures = f.readInt()
  570. }
  571. res.DataPresent = f.readByte() != 0
  572. return res
  573. case errWriteFailure:
  574. res := &RequestErrWriteFailure{
  575. errorFrame: errD,
  576. }
  577. res.Consistency = f.readConsistency()
  578. res.Received = f.readInt()
  579. res.BlockFor = f.readInt()
  580. if f.proto > protoVersion4 {
  581. res.ErrorMap = f.readErrorMap()
  582. res.NumFailures = len(res.ErrorMap)
  583. } else {
  584. res.NumFailures = f.readInt()
  585. }
  586. res.WriteType = f.readString()
  587. return res
  588. case errFunctionFailure:
  589. res := &RequestErrFunctionFailure{
  590. errorFrame: errD,
  591. }
  592. res.Keyspace = f.readString()
  593. res.Function = f.readString()
  594. res.ArgTypes = f.readStringList()
  595. return res
  596. case errCDCWriteFailure:
  597. res := &RequestErrCDCWriteFailure{
  598. errorFrame: errD,
  599. }
  600. return res
  601. case errInvalid, errBootstrapping, errConfig, errCredentials, errOverloaded,
  602. errProtocol, errServer, errSyntax, errTruncate, errUnauthorized:
  603. // TODO(zariel): we should have some distinct types for these errors
  604. return errD
  605. default:
  606. panic(fmt.Errorf("unknown error code: 0x%x", errD.code))
  607. }
  608. }
  609. func (f *framer) readErrorMap() (errMap ErrorMap) {
  610. errMap = make(ErrorMap)
  611. numErrs := f.readInt()
  612. for i := 0; i < numErrs; i++ {
  613. ip := f.readInetAdressOnly().String()
  614. errMap[ip] = f.readShort()
  615. }
  616. return
  617. }
  618. func (f *framer) writeHeader(flags byte, op frameOp, stream int) {
  619. f.wbuf = f.wbuf[:0]
  620. f.wbuf = append(f.wbuf,
  621. f.proto,
  622. flags,
  623. )
  624. if f.proto > protoVersion2 {
  625. f.wbuf = append(f.wbuf,
  626. byte(stream>>8),
  627. byte(stream),
  628. )
  629. } else {
  630. f.wbuf = append(f.wbuf,
  631. byte(stream),
  632. )
  633. }
  634. // pad out length
  635. f.wbuf = append(f.wbuf,
  636. byte(op),
  637. 0,
  638. 0,
  639. 0,
  640. 0,
  641. )
  642. }
  643. func (f *framer) setLength(length int) {
  644. p := 4
  645. if f.proto > protoVersion2 {
  646. p = 5
  647. }
  648. f.wbuf[p+0] = byte(length >> 24)
  649. f.wbuf[p+1] = byte(length >> 16)
  650. f.wbuf[p+2] = byte(length >> 8)
  651. f.wbuf[p+3] = byte(length)
  652. }
  653. func (f *framer) finishWrite() error {
  654. if len(f.wbuf) > maxFrameSize {
  655. // huge app frame, lets remove it so it doesn't bloat the heap
  656. f.wbuf = make([]byte, defaultBufSize)
  657. return ErrFrameTooBig
  658. }
  659. if f.wbuf[1]&flagCompress == flagCompress {
  660. if f.compres == nil {
  661. panic("compress flag set with no compressor")
  662. }
  663. // TODO: only compress frames which are big enough
  664. compressed, err := f.compres.Encode(f.wbuf[f.headSize:])
  665. if err != nil {
  666. return err
  667. }
  668. f.wbuf = append(f.wbuf[:f.headSize], compressed...)
  669. }
  670. length := len(f.wbuf) - f.headSize
  671. f.setLength(length)
  672. _, err := f.w.Write(f.wbuf)
  673. if err != nil {
  674. return err
  675. }
  676. return nil
  677. }
  678. func (f *framer) readTrace() {
  679. f.traceID = f.readUUID().Bytes()
  680. }
  681. type readyFrame struct {
  682. frameHeader
  683. }
  684. func (f *framer) parseReadyFrame() frame {
  685. return &readyFrame{
  686. frameHeader: *f.header,
  687. }
  688. }
  689. type supportedFrame struct {
  690. frameHeader
  691. supported map[string][]string
  692. }
  693. // TODO: if we move the body buffer onto the frameHeader then we only need a single
  694. // framer, and can move the methods onto the header.
  695. func (f *framer) parseSupportedFrame() frame {
  696. return &supportedFrame{
  697. frameHeader: *f.header,
  698. supported: f.readStringMultiMap(),
  699. }
  700. }
  701. type writeStartupFrame struct {
  702. opts map[string]string
  703. }
  704. func (w writeStartupFrame) String() string {
  705. return fmt.Sprintf("[startup opts=%+v]", w.opts)
  706. }
  707. func (w *writeStartupFrame) writeFrame(f *framer, streamID int) error {
  708. f.writeHeader(f.flags&^flagCompress, opStartup, streamID)
  709. f.writeStringMap(w.opts)
  710. return f.finishWrite()
  711. }
  712. type writePrepareFrame struct {
  713. statement string
  714. keyspace string
  715. customPayload map[string][]byte
  716. }
  717. func (w *writePrepareFrame) writeFrame(f *framer, streamID int) error {
  718. if len(w.customPayload) > 0 {
  719. f.payload()
  720. }
  721. f.writeHeader(f.flags, opPrepare, streamID)
  722. f.writeCustomPayload(&w.customPayload)
  723. f.writeLongString(w.statement)
  724. var flags uint32 = 0
  725. if w.keyspace != "" {
  726. if f.proto > protoVersion4 {
  727. flags |= flagWithPreparedKeyspace
  728. } else {
  729. panic(fmt.Errorf("The keyspace can only be set with protocol 5 or higher"))
  730. }
  731. }
  732. if f.proto > protoVersion4 {
  733. f.writeUint(flags)
  734. }
  735. if w.keyspace != "" {
  736. f.writeString(w.keyspace)
  737. }
  738. return f.finishWrite()
  739. }
  740. func (f *framer) readTypeInfo() TypeInfo {
  741. // TODO: factor this out so the same code paths can be used to parse custom
  742. // types and other types, as much of the logic will be duplicated.
  743. id := f.readShort()
  744. simple := NativeType{
  745. proto: f.proto,
  746. typ: Type(id),
  747. }
  748. if simple.typ == TypeCustom {
  749. simple.custom = f.readString()
  750. if cassType := getApacheCassandraType(simple.custom); cassType != TypeCustom {
  751. simple.typ = cassType
  752. }
  753. }
  754. switch simple.typ {
  755. case TypeTuple:
  756. n := f.readShort()
  757. tuple := TupleTypeInfo{
  758. NativeType: simple,
  759. Elems: make([]TypeInfo, n),
  760. }
  761. for i := 0; i < int(n); i++ {
  762. tuple.Elems[i] = f.readTypeInfo()
  763. }
  764. return tuple
  765. case TypeUDT:
  766. udt := UDTTypeInfo{
  767. NativeType: simple,
  768. }
  769. udt.KeySpace = f.readString()
  770. udt.Name = f.readString()
  771. n := f.readShort()
  772. udt.Elements = make([]UDTField, n)
  773. for i := 0; i < int(n); i++ {
  774. field := &udt.Elements[i]
  775. field.Name = f.readString()
  776. field.Type = f.readTypeInfo()
  777. }
  778. return udt
  779. case TypeMap, TypeList, TypeSet:
  780. collection := CollectionType{
  781. NativeType: simple,
  782. }
  783. if simple.typ == TypeMap {
  784. collection.Key = f.readTypeInfo()
  785. }
  786. collection.Elem = f.readTypeInfo()
  787. return collection
  788. }
  789. return simple
  790. }
  791. type preparedMetadata struct {
  792. resultMetadata
  793. // proto v4+
  794. pkeyColumns []int
  795. }
  796. func (r preparedMetadata) String() string {
  797. return fmt.Sprintf("[prepared flags=0x%x pkey=%v paging_state=% X columns=%v col_count=%d actual_col_count=%d]", r.flags, r.pkeyColumns, r.pagingState, r.columns, r.colCount, r.actualColCount)
  798. }
  799. func (f *framer) parsePreparedMetadata() preparedMetadata {
  800. // TODO: deduplicate this from parseMetadata
  801. meta := preparedMetadata{}
  802. meta.flags = f.readInt()
  803. meta.colCount = f.readInt()
  804. if meta.colCount < 0 {
  805. panic(fmt.Errorf("received negative column count: %d", meta.colCount))
  806. }
  807. meta.actualColCount = meta.colCount
  808. if f.proto >= protoVersion4 {
  809. pkeyCount := f.readInt()
  810. pkeys := make([]int, pkeyCount)
  811. for i := 0; i < pkeyCount; i++ {
  812. pkeys[i] = int(f.readShort())
  813. }
  814. meta.pkeyColumns = pkeys
  815. }
  816. if meta.flags&flagHasMorePages == flagHasMorePages {
  817. meta.pagingState = copyBytes(f.readBytes())
  818. }
  819. if meta.flags&flagNoMetaData == flagNoMetaData {
  820. return meta
  821. }
  822. var keyspace, table string
  823. globalSpec := meta.flags&flagGlobalTableSpec == flagGlobalTableSpec
  824. if globalSpec {
  825. keyspace = f.readString()
  826. table = f.readString()
  827. }
  828. var cols []ColumnInfo
  829. if meta.colCount < 1000 {
  830. // preallocate columninfo to avoid excess copying
  831. cols = make([]ColumnInfo, meta.colCount)
  832. for i := 0; i < meta.colCount; i++ {
  833. f.readCol(&cols[i], &meta.resultMetadata, globalSpec, keyspace, table)
  834. }
  835. } else {
  836. // use append, huge number of columns usually indicates a corrupt frame or
  837. // just a huge row.
  838. for i := 0; i < meta.colCount; i++ {
  839. var col ColumnInfo
  840. f.readCol(&col, &meta.resultMetadata, globalSpec, keyspace, table)
  841. cols = append(cols, col)
  842. }
  843. }
  844. meta.columns = cols
  845. return meta
  846. }
  847. type resultMetadata struct {
  848. flags int
  849. // only if flagPageState
  850. pagingState []byte
  851. columns []ColumnInfo
  852. colCount int
  853. // this is a count of the total number of columns which can be scanned,
  854. // it is at minimum len(columns) but may be larger, for instance when a column
  855. // is a UDT or tuple.
  856. actualColCount int
  857. }
  858. func (r resultMetadata) String() string {
  859. return fmt.Sprintf("[metadata flags=0x%x paging_state=% X columns=%v]", r.flags, r.pagingState, r.columns)
  860. }
  861. func (f *framer) readCol(col *ColumnInfo, meta *resultMetadata, globalSpec bool, keyspace, table string) {
  862. if !globalSpec {
  863. col.Keyspace = f.readString()
  864. col.Table = f.readString()
  865. } else {
  866. col.Keyspace = keyspace
  867. col.Table = table
  868. }
  869. col.Name = f.readString()
  870. col.TypeInfo = f.readTypeInfo()
  871. switch v := col.TypeInfo.(type) {
  872. // maybe also UDT
  873. case TupleTypeInfo:
  874. // -1 because we already included the tuple column
  875. meta.actualColCount += len(v.Elems) - 1
  876. }
  877. }
  878. func (f *framer) parseResultMetadata() resultMetadata {
  879. var meta resultMetadata
  880. meta.flags = f.readInt()
  881. meta.colCount = f.readInt()
  882. if meta.colCount < 0 {
  883. panic(fmt.Errorf("received negative column count: %d", meta.colCount))
  884. }
  885. meta.actualColCount = meta.colCount
  886. if meta.flags&flagHasMorePages == flagHasMorePages {
  887. meta.pagingState = copyBytes(f.readBytes())
  888. }
  889. if meta.flags&flagNoMetaData == flagNoMetaData {
  890. return meta
  891. }
  892. var keyspace, table string
  893. globalSpec := meta.flags&flagGlobalTableSpec == flagGlobalTableSpec
  894. if globalSpec {
  895. keyspace = f.readString()
  896. table = f.readString()
  897. }
  898. var cols []ColumnInfo
  899. if meta.colCount < 1000 {
  900. // preallocate columninfo to avoid excess copying
  901. cols = make([]ColumnInfo, meta.colCount)
  902. for i := 0; i < meta.colCount; i++ {
  903. f.readCol(&cols[i], &meta, globalSpec, keyspace, table)
  904. }
  905. } else {
  906. // use append, huge number of columns usually indicates a corrupt frame or
  907. // just a huge row.
  908. for i := 0; i < meta.colCount; i++ {
  909. var col ColumnInfo
  910. f.readCol(&col, &meta, globalSpec, keyspace, table)
  911. cols = append(cols, col)
  912. }
  913. }
  914. meta.columns = cols
  915. return meta
  916. }
  917. type resultVoidFrame struct {
  918. frameHeader
  919. }
  920. func (f *resultVoidFrame) String() string {
  921. return "[result_void]"
  922. }
  923. func (f *framer) parseResultFrame() (frame, error) {
  924. kind := f.readInt()
  925. switch kind {
  926. case resultKindVoid:
  927. return &resultVoidFrame{frameHeader: *f.header}, nil
  928. case resultKindRows:
  929. return f.parseResultRows(), nil
  930. case resultKindKeyspace:
  931. return f.parseResultSetKeyspace(), nil
  932. case resultKindPrepared:
  933. return f.parseResultPrepared(), nil
  934. case resultKindSchemaChanged:
  935. return f.parseResultSchemaChange(), nil
  936. }
  937. return nil, NewErrProtocol("unknown result kind: %x", kind)
  938. }
  939. type resultRowsFrame struct {
  940. frameHeader
  941. meta resultMetadata
  942. // dont parse the rows here as we only need to do it once
  943. numRows int
  944. }
  945. func (f *resultRowsFrame) String() string {
  946. return fmt.Sprintf("[result_rows meta=%v]", f.meta)
  947. }
  948. func (f *framer) parseResultRows() frame {
  949. result := &resultRowsFrame{}
  950. result.meta = f.parseResultMetadata()
  951. result.numRows = f.readInt()
  952. if result.numRows < 0 {
  953. panic(fmt.Errorf("invalid row_count in result frame: %d", result.numRows))
  954. }
  955. return result
  956. }
  957. type resultKeyspaceFrame struct {
  958. frameHeader
  959. keyspace string
  960. }
  961. func (r *resultKeyspaceFrame) String() string {
  962. return fmt.Sprintf("[result_keyspace keyspace=%s]", r.keyspace)
  963. }
  964. func (f *framer) parseResultSetKeyspace() frame {
  965. return &resultKeyspaceFrame{
  966. frameHeader: *f.header,
  967. keyspace: f.readString(),
  968. }
  969. }
  970. type resultPreparedFrame struct {
  971. frameHeader
  972. preparedID []byte
  973. reqMeta preparedMetadata
  974. respMeta resultMetadata
  975. }
  976. func (f *framer) parseResultPrepared() frame {
  977. frame := &resultPreparedFrame{
  978. frameHeader: *f.header,
  979. preparedID: f.readShortBytes(),
  980. reqMeta: f.parsePreparedMetadata(),
  981. }
  982. if f.proto < protoVersion2 {
  983. return frame
  984. }
  985. frame.respMeta = f.parseResultMetadata()
  986. return frame
  987. }
  988. type schemaChangeKeyspace struct {
  989. frameHeader
  990. change string
  991. keyspace string
  992. }
  993. func (f schemaChangeKeyspace) String() string {
  994. return fmt.Sprintf("[event schema_change_keyspace change=%q keyspace=%q]", f.change, f.keyspace)
  995. }
  996. type schemaChangeTable struct {
  997. frameHeader
  998. change string
  999. keyspace string
  1000. object string
  1001. }
  1002. func (f schemaChangeTable) String() string {
  1003. return fmt.Sprintf("[event schema_change change=%q keyspace=%q object=%q]", f.change, f.keyspace, f.object)
  1004. }
  1005. type schemaChangeType struct {
  1006. frameHeader
  1007. change string
  1008. keyspace string
  1009. object string
  1010. }
  1011. type schemaChangeFunction struct {
  1012. frameHeader
  1013. change string
  1014. keyspace string
  1015. name string
  1016. args []string
  1017. }
  1018. type schemaChangeAggregate struct {
  1019. frameHeader
  1020. change string
  1021. keyspace string
  1022. name string
  1023. args []string
  1024. }
  1025. func (f *framer) parseResultSchemaChange() frame {
  1026. if f.proto <= protoVersion2 {
  1027. change := f.readString()
  1028. keyspace := f.readString()
  1029. table := f.readString()
  1030. if table != "" {
  1031. return &schemaChangeTable{
  1032. frameHeader: *f.header,
  1033. change: change,
  1034. keyspace: keyspace,
  1035. object: table,
  1036. }
  1037. } else {
  1038. return &schemaChangeKeyspace{
  1039. frameHeader: *f.header,
  1040. change: change,
  1041. keyspace: keyspace,
  1042. }
  1043. }
  1044. } else {
  1045. change := f.readString()
  1046. target := f.readString()
  1047. // TODO: could just use a separate type for each target
  1048. switch target {
  1049. case "KEYSPACE":
  1050. frame := &schemaChangeKeyspace{
  1051. frameHeader: *f.header,
  1052. change: change,
  1053. }
  1054. frame.keyspace = f.readString()
  1055. return frame
  1056. case "TABLE":
  1057. frame := &schemaChangeTable{
  1058. frameHeader: *f.header,
  1059. change: change,
  1060. }
  1061. frame.keyspace = f.readString()
  1062. frame.object = f.readString()
  1063. return frame
  1064. case "TYPE":
  1065. frame := &schemaChangeType{
  1066. frameHeader: *f.header,
  1067. change: change,
  1068. }
  1069. frame.keyspace = f.readString()
  1070. frame.object = f.readString()
  1071. return frame
  1072. case "FUNCTION":
  1073. frame := &schemaChangeFunction{
  1074. frameHeader: *f.header,
  1075. change: change,
  1076. }
  1077. frame.keyspace = f.readString()
  1078. frame.name = f.readString()
  1079. frame.args = f.readStringList()
  1080. return frame
  1081. case "AGGREGATE":
  1082. frame := &schemaChangeAggregate{
  1083. frameHeader: *f.header,
  1084. change: change,
  1085. }
  1086. frame.keyspace = f.readString()
  1087. frame.name = f.readString()
  1088. frame.args = f.readStringList()
  1089. return frame
  1090. default:
  1091. panic(fmt.Errorf("gocql: unknown SCHEMA_CHANGE target: %q change: %q", target, change))
  1092. }
  1093. }
  1094. }
  1095. type authenticateFrame struct {
  1096. frameHeader
  1097. class string
  1098. }
  1099. func (a *authenticateFrame) String() string {
  1100. return fmt.Sprintf("[authenticate class=%q]", a.class)
  1101. }
  1102. func (f *framer) parseAuthenticateFrame() frame {
  1103. return &authenticateFrame{
  1104. frameHeader: *f.header,
  1105. class: f.readString(),
  1106. }
  1107. }
  1108. type authSuccessFrame struct {
  1109. frameHeader
  1110. data []byte
  1111. }
  1112. func (a *authSuccessFrame) String() string {
  1113. return fmt.Sprintf("[auth_success data=%q]", a.data)
  1114. }
  1115. func (f *framer) parseAuthSuccessFrame() frame {
  1116. return &authSuccessFrame{
  1117. frameHeader: *f.header,
  1118. data: f.readBytes(),
  1119. }
  1120. }
  1121. type authChallengeFrame struct {
  1122. frameHeader
  1123. data []byte
  1124. }
  1125. func (a *authChallengeFrame) String() string {
  1126. return fmt.Sprintf("[auth_challenge data=%q]", a.data)
  1127. }
  1128. func (f *framer) parseAuthChallengeFrame() frame {
  1129. return &authChallengeFrame{
  1130. frameHeader: *f.header,
  1131. data: f.readBytes(),
  1132. }
  1133. }
  1134. type statusChangeEventFrame struct {
  1135. frameHeader
  1136. change string
  1137. host net.IP
  1138. port int
  1139. }
  1140. func (t statusChangeEventFrame) String() string {
  1141. return fmt.Sprintf("[status_change change=%s host=%v port=%v]", t.change, t.host, t.port)
  1142. }
  1143. // essentially the same as statusChange
  1144. type topologyChangeEventFrame struct {
  1145. frameHeader
  1146. change string
  1147. host net.IP
  1148. port int
  1149. }
  1150. func (t topologyChangeEventFrame) String() string {
  1151. return fmt.Sprintf("[topology_change change=%s host=%v port=%v]", t.change, t.host, t.port)
  1152. }
  1153. func (f *framer) parseEventFrame() frame {
  1154. eventType := f.readString()
  1155. switch eventType {
  1156. case "TOPOLOGY_CHANGE":
  1157. frame := &topologyChangeEventFrame{frameHeader: *f.header}
  1158. frame.change = f.readString()
  1159. frame.host, frame.port = f.readInet()
  1160. return frame
  1161. case "STATUS_CHANGE":
  1162. frame := &statusChangeEventFrame{frameHeader: *f.header}
  1163. frame.change = f.readString()
  1164. frame.host, frame.port = f.readInet()
  1165. return frame
  1166. case "SCHEMA_CHANGE":
  1167. // this should work for all versions
  1168. return f.parseResultSchemaChange()
  1169. default:
  1170. panic(fmt.Errorf("gocql: unknown event type: %q", eventType))
  1171. }
  1172. }
  1173. type writeAuthResponseFrame struct {
  1174. data []byte
  1175. }
  1176. func (a *writeAuthResponseFrame) String() string {
  1177. return fmt.Sprintf("[auth_response data=%q]", a.data)
  1178. }
  1179. func (a *writeAuthResponseFrame) writeFrame(framer *framer, streamID int) error {
  1180. return framer.writeAuthResponseFrame(streamID, a.data)
  1181. }
  1182. func (f *framer) writeAuthResponseFrame(streamID int, data []byte) error {
  1183. f.writeHeader(f.flags, opAuthResponse, streamID)
  1184. f.writeBytes(data)
  1185. return f.finishWrite()
  1186. }
  1187. type queryValues struct {
  1188. value []byte
  1189. // optional name, will set With names for values flag
  1190. name string
  1191. isUnset bool
  1192. }
  1193. type queryParams struct {
  1194. consistency Consistency
  1195. // v2+
  1196. skipMeta bool
  1197. values []queryValues
  1198. pageSize int
  1199. pagingState []byte
  1200. serialConsistency SerialConsistency
  1201. // v3+
  1202. defaultTimestamp bool
  1203. defaultTimestampValue int64
  1204. // v5+
  1205. keyspace string
  1206. }
  1207. func (q queryParams) String() string {
  1208. return fmt.Sprintf("[query_params consistency=%v skip_meta=%v page_size=%d paging_state=%q serial_consistency=%v default_timestamp=%v values=%v keyspace=%s]",
  1209. q.consistency, q.skipMeta, q.pageSize, q.pagingState, q.serialConsistency, q.defaultTimestamp, q.values, q.keyspace)
  1210. }
  1211. func (f *framer) writeQueryParams(opts *queryParams) {
  1212. f.writeConsistency(opts.consistency)
  1213. if f.proto == protoVersion1 {
  1214. return
  1215. }
  1216. var flags byte
  1217. if len(opts.values) > 0 {
  1218. flags |= flagValues
  1219. }
  1220. if opts.skipMeta {
  1221. flags |= flagSkipMetaData
  1222. }
  1223. if opts.pageSize > 0 {
  1224. flags |= flagPageSize
  1225. }
  1226. if len(opts.pagingState) > 0 {
  1227. flags |= flagWithPagingState
  1228. }
  1229. if opts.serialConsistency > 0 {
  1230. flags |= flagWithSerialConsistency
  1231. }
  1232. names := false
  1233. // protoV3 specific things
  1234. if f.proto > protoVersion2 {
  1235. if opts.defaultTimestamp {
  1236. flags |= flagDefaultTimestamp
  1237. }
  1238. if len(opts.values) > 0 && opts.values[0].name != "" {
  1239. flags |= flagWithNameValues
  1240. names = true
  1241. }
  1242. }
  1243. if opts.keyspace != "" {
  1244. if f.proto > protoVersion4 {
  1245. flags |= flagWithKeyspace
  1246. } else {
  1247. panic(fmt.Errorf("The keyspace can only be set with protocol 5 or higher"))
  1248. }
  1249. }
  1250. if f.proto > protoVersion4 {
  1251. f.writeUint(uint32(flags))
  1252. } else {
  1253. f.writeByte(flags)
  1254. }
  1255. if n := len(opts.values); n > 0 {
  1256. f.writeShort(uint16(n))
  1257. for i := 0; i < n; i++ {
  1258. if names {
  1259. f.writeString(opts.values[i].name)
  1260. }
  1261. if opts.values[i].isUnset {
  1262. f.writeUnset()
  1263. } else {
  1264. f.writeBytes(opts.values[i].value)
  1265. }
  1266. }
  1267. }
  1268. if opts.pageSize > 0 {
  1269. f.writeInt(int32(opts.pageSize))
  1270. }
  1271. if len(opts.pagingState) > 0 {
  1272. f.writeBytes(opts.pagingState)
  1273. }
  1274. if opts.serialConsistency > 0 {
  1275. f.writeConsistency(Consistency(opts.serialConsistency))
  1276. }
  1277. if f.proto > protoVersion2 && opts.defaultTimestamp {
  1278. // timestamp in microseconds
  1279. var ts int64
  1280. if opts.defaultTimestampValue != 0 {
  1281. ts = opts.defaultTimestampValue
  1282. } else {
  1283. ts = time.Now().UnixNano() / 1000
  1284. }
  1285. f.writeLong(ts)
  1286. }
  1287. if opts.keyspace != "" {
  1288. f.writeString(opts.keyspace)
  1289. }
  1290. }
  1291. type writeQueryFrame struct {
  1292. statement string
  1293. params queryParams
  1294. // v4+
  1295. customPayload map[string][]byte
  1296. }
  1297. func (w *writeQueryFrame) String() string {
  1298. return fmt.Sprintf("[query statement=%q params=%v]", w.statement, w.params)
  1299. }
  1300. func (w *writeQueryFrame) writeFrame(framer *framer, streamID int) error {
  1301. return framer.writeQueryFrame(streamID, w.statement, &w.params, w.customPayload)
  1302. }
  1303. func (f *framer) writeQueryFrame(streamID int, statement string, params *queryParams, customPayload map[string][]byte) error {
  1304. if len(customPayload) > 0 {
  1305. f.payload()
  1306. }
  1307. f.writeHeader(f.flags, opQuery, streamID)
  1308. f.writeCustomPayload(&customPayload)
  1309. f.writeLongString(statement)
  1310. f.writeQueryParams(params)
  1311. return f.finishWrite()
  1312. }
  1313. type frameWriter interface {
  1314. writeFrame(framer *framer, streamID int) error
  1315. }
  1316. type frameWriterFunc func(framer *framer, streamID int) error
  1317. func (f frameWriterFunc) writeFrame(framer *framer, streamID int) error {
  1318. return f(framer, streamID)
  1319. }
  1320. type writeExecuteFrame struct {
  1321. preparedID []byte
  1322. params queryParams
  1323. // v4+
  1324. customPayload map[string][]byte
  1325. }
  1326. func (e *writeExecuteFrame) String() string {
  1327. return fmt.Sprintf("[execute id=% X params=%v]", e.preparedID, &e.params)
  1328. }
  1329. func (e *writeExecuteFrame) writeFrame(fr *framer, streamID int) error {
  1330. return fr.writeExecuteFrame(streamID, e.preparedID, &e.params, &e.customPayload)
  1331. }
  1332. func (f *framer) writeExecuteFrame(streamID int, preparedID []byte, params *queryParams, customPayload *map[string][]byte) error {
  1333. if len(*customPayload) > 0 {
  1334. f.payload()
  1335. }
  1336. f.writeHeader(f.flags, opExecute, streamID)
  1337. f.writeCustomPayload(customPayload)
  1338. f.writeShortBytes(preparedID)
  1339. if f.proto > protoVersion1 {
  1340. f.writeQueryParams(params)
  1341. } else {
  1342. n := len(params.values)
  1343. f.writeShort(uint16(n))
  1344. for i := 0; i < n; i++ {
  1345. if params.values[i].isUnset {
  1346. f.writeUnset()
  1347. } else {
  1348. f.writeBytes(params.values[i].value)
  1349. }
  1350. }
  1351. f.writeConsistency(params.consistency)
  1352. }
  1353. return f.finishWrite()
  1354. }
  1355. // TODO: can we replace BatchStatemt with batchStatement? As they prety much
  1356. // duplicate each other
  1357. type batchStatment struct {
  1358. preparedID []byte
  1359. statement string
  1360. values []queryValues
  1361. }
  1362. type writeBatchFrame struct {
  1363. typ BatchType
  1364. statements []batchStatment
  1365. consistency Consistency
  1366. // v3+
  1367. serialConsistency SerialConsistency
  1368. defaultTimestamp bool
  1369. defaultTimestampValue int64
  1370. //v4+
  1371. customPayload map[string][]byte
  1372. }
  1373. func (w *writeBatchFrame) writeFrame(framer *framer, streamID int) error {
  1374. return framer.writeBatchFrame(streamID, w, w.customPayload)
  1375. }
  1376. func (f *framer) writeBatchFrame(streamID int, w *writeBatchFrame, customPayload map[string][]byte) error {
  1377. if len(customPayload) > 0 {
  1378. f.payload()
  1379. }
  1380. f.writeHeader(f.flags, opBatch, streamID)
  1381. f.writeCustomPayload(&customPayload)
  1382. f.writeByte(byte(w.typ))
  1383. n := len(w.statements)
  1384. f.writeShort(uint16(n))
  1385. var flags byte
  1386. for i := 0; i < n; i++ {
  1387. b := &w.statements[i]
  1388. if len(b.preparedID) == 0 {
  1389. f.writeByte(0)
  1390. f.writeLongString(b.statement)
  1391. } else {
  1392. f.writeByte(1)
  1393. f.writeShortBytes(b.preparedID)
  1394. }
  1395. f.writeShort(uint16(len(b.values)))
  1396. for j := range b.values {
  1397. col := b.values[j]
  1398. if f.proto > protoVersion2 && col.name != "" {
  1399. // TODO: move this check into the caller and set a flag on writeBatchFrame
  1400. // to indicate using named values
  1401. if f.proto <= protoVersion5 {
  1402. return fmt.Errorf("gocql: named query values are not supported in batches, please see https://issues.apache.org/jira/browse/CASSANDRA-10246")
  1403. }
  1404. flags |= flagWithNameValues
  1405. f.writeString(col.name)
  1406. }
  1407. if col.isUnset {
  1408. f.writeUnset()
  1409. } else {
  1410. f.writeBytes(col.value)
  1411. }
  1412. }
  1413. }
  1414. f.writeConsistency(w.consistency)
  1415. if f.proto > protoVersion2 {
  1416. if w.serialConsistency > 0 {
  1417. flags |= flagWithSerialConsistency
  1418. }
  1419. if w.defaultTimestamp {
  1420. flags |= flagDefaultTimestamp
  1421. }
  1422. if f.proto > protoVersion4 {
  1423. f.writeUint(uint32(flags))
  1424. } else {
  1425. f.writeByte(flags)
  1426. }
  1427. if w.serialConsistency > 0 {
  1428. f.writeConsistency(Consistency(w.serialConsistency))
  1429. }
  1430. if w.defaultTimestamp {
  1431. var ts int64
  1432. if w.defaultTimestampValue != 0 {
  1433. ts = w.defaultTimestampValue
  1434. } else {
  1435. ts = time.Now().UnixNano() / 1000
  1436. }
  1437. f.writeLong(ts)
  1438. }
  1439. }
  1440. return f.finishWrite()
  1441. }
  1442. type writeOptionsFrame struct{}
  1443. func (w *writeOptionsFrame) writeFrame(framer *framer, streamID int) error {
  1444. return framer.writeOptionsFrame(streamID, w)
  1445. }
  1446. func (f *framer) writeOptionsFrame(stream int, _ *writeOptionsFrame) error {
  1447. f.writeHeader(f.flags, opOptions, stream)
  1448. return f.finishWrite()
  1449. }
  1450. type writeRegisterFrame struct {
  1451. events []string
  1452. }
  1453. func (w *writeRegisterFrame) writeFrame(framer *framer, streamID int) error {
  1454. return framer.writeRegisterFrame(streamID, w)
  1455. }
  1456. func (f *framer) writeRegisterFrame(streamID int, w *writeRegisterFrame) error {
  1457. f.writeHeader(f.flags, opRegister, streamID)
  1458. f.writeStringList(w.events)
  1459. return f.finishWrite()
  1460. }
  1461. func (f *framer) readByte() byte {
  1462. if len(f.rbuf) < 1 {
  1463. panic(fmt.Errorf("not enough bytes in buffer to read byte require 1 got: %d", len(f.rbuf)))
  1464. }
  1465. b := f.rbuf[0]
  1466. f.rbuf = f.rbuf[1:]
  1467. return b
  1468. }
  1469. func (f *framer) readInt() (n int) {
  1470. if len(f.rbuf) < 4 {
  1471. panic(fmt.Errorf("not enough bytes in buffer to read int require 4 got: %d", len(f.rbuf)))
  1472. }
  1473. n = int(int32(f.rbuf[0])<<24 | int32(f.rbuf[1])<<16 | int32(f.rbuf[2])<<8 | int32(f.rbuf[3]))
  1474. f.rbuf = f.rbuf[4:]
  1475. return
  1476. }
  1477. func (f *framer) readShort() (n uint16) {
  1478. if len(f.rbuf) < 2 {
  1479. panic(fmt.Errorf("not enough bytes in buffer to read short require 2 got: %d", len(f.rbuf)))
  1480. }
  1481. n = uint16(f.rbuf[0])<<8 | uint16(f.rbuf[1])
  1482. f.rbuf = f.rbuf[2:]
  1483. return
  1484. }
  1485. func (f *framer) readLong() (n int64) {
  1486. if len(f.rbuf) < 8 {
  1487. panic(fmt.Errorf("not enough bytes in buffer to read long require 8 got: %d", len(f.rbuf)))
  1488. }
  1489. n = int64(f.rbuf[0])<<56 | int64(f.rbuf[1])<<48 | int64(f.rbuf[2])<<40 | int64(f.rbuf[3])<<32 |
  1490. int64(f.rbuf[4])<<24 | int64(f.rbuf[5])<<16 | int64(f.rbuf[6])<<8 | int64(f.rbuf[7])
  1491. f.rbuf = f.rbuf[8:]
  1492. return
  1493. }
  1494. func (f *framer) readString() (s string) {
  1495. size := f.readShort()
  1496. if len(f.rbuf) < int(size) {
  1497. panic(fmt.Errorf("not enough bytes in buffer to read string require %d got: %d", size, len(f.rbuf)))
  1498. }
  1499. s = string(f.rbuf[:size])
  1500. f.rbuf = f.rbuf[size:]
  1501. return
  1502. }
  1503. func (f *framer) readLongString() (s string) {
  1504. size := f.readInt()
  1505. if len(f.rbuf) < size {
  1506. panic(fmt.Errorf("not enough bytes in buffer to read long string require %d got: %d", size, len(f.rbuf)))
  1507. }
  1508. s = string(f.rbuf[:size])
  1509. f.rbuf = f.rbuf[size:]
  1510. return
  1511. }
  1512. func (f *framer) readUUID() *UUID {
  1513. if len(f.rbuf) < 16 {
  1514. panic(fmt.Errorf("not enough bytes in buffer to read uuid require %d got: %d", 16, len(f.rbuf)))
  1515. }
  1516. // TODO: how to handle this error, if it is a uuid, then sureley, problems?
  1517. u, _ := UUIDFromBytes(f.rbuf[:16])
  1518. f.rbuf = f.rbuf[16:]
  1519. return &u
  1520. }
  1521. func (f *framer) readStringList() []string {
  1522. size := f.readShort()
  1523. l := make([]string, size)
  1524. for i := 0; i < int(size); i++ {
  1525. l[i] = f.readString()
  1526. }
  1527. return l
  1528. }
  1529. func (f *framer) readBytesInternal() ([]byte, error) {
  1530. size := f.readInt()
  1531. if size < 0 {
  1532. return nil, nil
  1533. }
  1534. if len(f.rbuf) < size {
  1535. return nil, fmt.Errorf("not enough bytes in buffer to read bytes require %d got: %d", size, len(f.rbuf))
  1536. }
  1537. l := f.rbuf[:size]
  1538. f.rbuf = f.rbuf[size:]
  1539. return l, nil
  1540. }
  1541. func (f *framer) readBytes() []byte {
  1542. l, err := f.readBytesInternal()
  1543. if err != nil {
  1544. panic(err)
  1545. }
  1546. return l
  1547. }
  1548. func (f *framer) readShortBytes() []byte {
  1549. size := f.readShort()
  1550. if len(f.rbuf) < int(size) {
  1551. panic(fmt.Errorf("not enough bytes in buffer to read short bytes: require %d got %d", size, len(f.rbuf)))
  1552. }
  1553. l := f.rbuf[:size]
  1554. f.rbuf = f.rbuf[size:]
  1555. return l
  1556. }
  1557. func (f *framer) readInetAdressOnly() net.IP {
  1558. if len(f.rbuf) < 1 {
  1559. panic(fmt.Errorf("not enough bytes in buffer to read inet size require %d got: %d", 1, len(f.rbuf)))
  1560. }
  1561. size := f.rbuf[0]
  1562. f.rbuf = f.rbuf[1:]
  1563. if !(size == 4 || size == 16) {
  1564. panic(fmt.Errorf("invalid IP size: %d", size))
  1565. }
  1566. if len(f.rbuf) < 1 {
  1567. panic(fmt.Errorf("not enough bytes in buffer to read inet require %d got: %d", size, len(f.rbuf)))
  1568. }
  1569. ip := make([]byte, size)
  1570. copy(ip, f.rbuf[:size])
  1571. f.rbuf = f.rbuf[size:]
  1572. return net.IP(ip)
  1573. }
  1574. func (f *framer) readInet() (net.IP, int) {
  1575. return f.readInetAdressOnly(), f.readInt()
  1576. }
  1577. func (f *framer) readConsistency() Consistency {
  1578. return Consistency(f.readShort())
  1579. }
  1580. func (f *framer) readStringMap() map[string]string {
  1581. size := f.readShort()
  1582. m := make(map[string]string, size)
  1583. for i := 0; i < int(size); i++ {
  1584. k := f.readString()
  1585. v := f.readString()
  1586. m[k] = v
  1587. }
  1588. return m
  1589. }
  1590. func (f *framer) readBytesMap() map[string][]byte {
  1591. size := f.readShort()
  1592. m := make(map[string][]byte, size)
  1593. for i := 0; i < int(size); i++ {
  1594. k := f.readString()
  1595. v := f.readBytes()
  1596. m[k] = v
  1597. }
  1598. return m
  1599. }
  1600. func (f *framer) readStringMultiMap() map[string][]string {
  1601. size := f.readShort()
  1602. m := make(map[string][]string, size)
  1603. for i := 0; i < int(size); i++ {
  1604. k := f.readString()
  1605. v := f.readStringList()
  1606. m[k] = v
  1607. }
  1608. return m
  1609. }
  1610. func (f *framer) writeByte(b byte) {
  1611. f.wbuf = append(f.wbuf, b)
  1612. }
  1613. func appendBytes(p []byte, d []byte) []byte {
  1614. if d == nil {
  1615. return appendInt(p, -1)
  1616. }
  1617. p = appendInt(p, int32(len(d)))
  1618. p = append(p, d...)
  1619. return p
  1620. }
  1621. func appendShort(p []byte, n uint16) []byte {
  1622. return append(p,
  1623. byte(n>>8),
  1624. byte(n),
  1625. )
  1626. }
  1627. func appendInt(p []byte, n int32) []byte {
  1628. return append(p, byte(n>>24),
  1629. byte(n>>16),
  1630. byte(n>>8),
  1631. byte(n))
  1632. }
  1633. func appendUint(p []byte, n uint32) []byte {
  1634. return append(p, byte(n>>24),
  1635. byte(n>>16),
  1636. byte(n>>8),
  1637. byte(n))
  1638. }
  1639. func appendLong(p []byte, n int64) []byte {
  1640. return append(p,
  1641. byte(n>>56),
  1642. byte(n>>48),
  1643. byte(n>>40),
  1644. byte(n>>32),
  1645. byte(n>>24),
  1646. byte(n>>16),
  1647. byte(n>>8),
  1648. byte(n),
  1649. )
  1650. }
  1651. func (f *framer) writeCustomPayload(customPayload *map[string][]byte) {
  1652. if len(*customPayload) > 0 {
  1653. if f.proto < protoVersion4 {
  1654. panic("Custom payload is not supported with version V3 or less")
  1655. }
  1656. f.writeBytesMap(*customPayload)
  1657. }
  1658. }
  1659. // these are protocol level binary types
  1660. func (f *framer) writeInt(n int32) {
  1661. f.wbuf = appendInt(f.wbuf, n)
  1662. }
  1663. func (f *framer) writeUint(n uint32) {
  1664. f.wbuf = appendUint(f.wbuf, n)
  1665. }
  1666. func (f *framer) writeShort(n uint16) {
  1667. f.wbuf = appendShort(f.wbuf, n)
  1668. }
  1669. func (f *framer) writeLong(n int64) {
  1670. f.wbuf = appendLong(f.wbuf, n)
  1671. }
  1672. func (f *framer) writeString(s string) {
  1673. f.writeShort(uint16(len(s)))
  1674. f.wbuf = append(f.wbuf, s...)
  1675. }
  1676. func (f *framer) writeLongString(s string) {
  1677. f.writeInt(int32(len(s)))
  1678. f.wbuf = append(f.wbuf, s...)
  1679. }
  1680. func (f *framer) writeUUID(u *UUID) {
  1681. f.wbuf = append(f.wbuf, u[:]...)
  1682. }
  1683. func (f *framer) writeStringList(l []string) {
  1684. f.writeShort(uint16(len(l)))
  1685. for _, s := range l {
  1686. f.writeString(s)
  1687. }
  1688. }
  1689. func (f *framer) writeUnset() {
  1690. // Protocol version 4 specifies that bind variables do not require having a
  1691. // value when executing a statement. Bind variables without a value are
  1692. // called 'unset'. The 'unset' bind variable is serialized as the int
  1693. // value '-2' without following bytes.
  1694. f.writeInt(-2)
  1695. }
  1696. func (f *framer) writeBytes(p []byte) {
  1697. // TODO: handle null case correctly,
  1698. // [bytes] A [int] n, followed by n bytes if n >= 0. If n < 0,
  1699. // no byte should follow and the value represented is `null`.
  1700. if p == nil {
  1701. f.writeInt(-1)
  1702. } else {
  1703. f.writeInt(int32(len(p)))
  1704. f.wbuf = append(f.wbuf, p...)
  1705. }
  1706. }
  1707. func (f *framer) writeShortBytes(p []byte) {
  1708. f.writeShort(uint16(len(p)))
  1709. f.wbuf = append(f.wbuf, p...)
  1710. }
  1711. func (f *framer) writeInet(ip net.IP, port int) {
  1712. f.wbuf = append(f.wbuf,
  1713. byte(len(ip)),
  1714. )
  1715. f.wbuf = append(f.wbuf,
  1716. []byte(ip)...,
  1717. )
  1718. f.writeInt(int32(port))
  1719. }
  1720. func (f *framer) writeConsistency(cons Consistency) {
  1721. f.writeShort(uint16(cons))
  1722. }
  1723. func (f *framer) writeStringMap(m map[string]string) {
  1724. f.writeShort(uint16(len(m)))
  1725. for k, v := range m {
  1726. f.writeString(k)
  1727. f.writeString(v)
  1728. }
  1729. }
  1730. func (f *framer) writeBytesMap(m map[string][]byte) {
  1731. f.writeShort(uint16(len(m)))
  1732. for k, v := range m {
  1733. f.writeString(k)
  1734. f.writeBytes(v)
  1735. }
  1736. }