frame.go 45 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103
  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) morePages() bool {
  859. return r.flags&flagHasMorePages == flagHasMorePages
  860. }
  861. func (r resultMetadata) String() string {
  862. return fmt.Sprintf("[metadata flags=0x%x paging_state=% X columns=%v]", r.flags, r.pagingState, r.columns)
  863. }
  864. func (f *framer) readCol(col *ColumnInfo, meta *resultMetadata, globalSpec bool, keyspace, table string) {
  865. if !globalSpec {
  866. col.Keyspace = f.readString()
  867. col.Table = f.readString()
  868. } else {
  869. col.Keyspace = keyspace
  870. col.Table = table
  871. }
  872. col.Name = f.readString()
  873. col.TypeInfo = f.readTypeInfo()
  874. switch v := col.TypeInfo.(type) {
  875. // maybe also UDT
  876. case TupleTypeInfo:
  877. // -1 because we already included the tuple column
  878. meta.actualColCount += len(v.Elems) - 1
  879. }
  880. }
  881. func (f *framer) parseResultMetadata() resultMetadata {
  882. var meta resultMetadata
  883. meta.flags = f.readInt()
  884. meta.colCount = f.readInt()
  885. if meta.colCount < 0 {
  886. panic(fmt.Errorf("received negative column count: %d", meta.colCount))
  887. }
  888. meta.actualColCount = meta.colCount
  889. if meta.flags&flagHasMorePages == flagHasMorePages {
  890. meta.pagingState = copyBytes(f.readBytes())
  891. }
  892. if meta.flags&flagNoMetaData == flagNoMetaData {
  893. return meta
  894. }
  895. var keyspace, table string
  896. globalSpec := meta.flags&flagGlobalTableSpec == flagGlobalTableSpec
  897. if globalSpec {
  898. keyspace = f.readString()
  899. table = f.readString()
  900. }
  901. var cols []ColumnInfo
  902. if meta.colCount < 1000 {
  903. // preallocate columninfo to avoid excess copying
  904. cols = make([]ColumnInfo, meta.colCount)
  905. for i := 0; i < meta.colCount; i++ {
  906. f.readCol(&cols[i], &meta, globalSpec, keyspace, table)
  907. }
  908. } else {
  909. // use append, huge number of columns usually indicates a corrupt frame or
  910. // just a huge row.
  911. for i := 0; i < meta.colCount; i++ {
  912. var col ColumnInfo
  913. f.readCol(&col, &meta, globalSpec, keyspace, table)
  914. cols = append(cols, col)
  915. }
  916. }
  917. meta.columns = cols
  918. return meta
  919. }
  920. type resultVoidFrame struct {
  921. frameHeader
  922. }
  923. func (f *resultVoidFrame) String() string {
  924. return "[result_void]"
  925. }
  926. func (f *framer) parseResultFrame() (frame, error) {
  927. kind := f.readInt()
  928. switch kind {
  929. case resultKindVoid:
  930. return &resultVoidFrame{frameHeader: *f.header}, nil
  931. case resultKindRows:
  932. return f.parseResultRows(), nil
  933. case resultKindKeyspace:
  934. return f.parseResultSetKeyspace(), nil
  935. case resultKindPrepared:
  936. return f.parseResultPrepared(), nil
  937. case resultKindSchemaChanged:
  938. return f.parseResultSchemaChange(), nil
  939. }
  940. return nil, NewErrProtocol("unknown result kind: %x", kind)
  941. }
  942. type resultRowsFrame struct {
  943. frameHeader
  944. meta resultMetadata
  945. // dont parse the rows here as we only need to do it once
  946. numRows int
  947. }
  948. func (f *resultRowsFrame) String() string {
  949. return fmt.Sprintf("[result_rows meta=%v]", f.meta)
  950. }
  951. func (f *framer) parseResultRows() frame {
  952. result := &resultRowsFrame{}
  953. result.meta = f.parseResultMetadata()
  954. result.numRows = f.readInt()
  955. if result.numRows < 0 {
  956. panic(fmt.Errorf("invalid row_count in result frame: %d", result.numRows))
  957. }
  958. return result
  959. }
  960. type resultKeyspaceFrame struct {
  961. frameHeader
  962. keyspace string
  963. }
  964. func (r *resultKeyspaceFrame) String() string {
  965. return fmt.Sprintf("[result_keyspace keyspace=%s]", r.keyspace)
  966. }
  967. func (f *framer) parseResultSetKeyspace() frame {
  968. return &resultKeyspaceFrame{
  969. frameHeader: *f.header,
  970. keyspace: f.readString(),
  971. }
  972. }
  973. type resultPreparedFrame struct {
  974. frameHeader
  975. preparedID []byte
  976. reqMeta preparedMetadata
  977. respMeta resultMetadata
  978. }
  979. func (f *framer) parseResultPrepared() frame {
  980. frame := &resultPreparedFrame{
  981. frameHeader: *f.header,
  982. preparedID: f.readShortBytes(),
  983. reqMeta: f.parsePreparedMetadata(),
  984. }
  985. if f.proto < protoVersion2 {
  986. return frame
  987. }
  988. frame.respMeta = f.parseResultMetadata()
  989. return frame
  990. }
  991. type schemaChangeKeyspace struct {
  992. frameHeader
  993. change string
  994. keyspace string
  995. }
  996. func (f schemaChangeKeyspace) String() string {
  997. return fmt.Sprintf("[event schema_change_keyspace change=%q keyspace=%q]", f.change, f.keyspace)
  998. }
  999. type schemaChangeTable struct {
  1000. frameHeader
  1001. change string
  1002. keyspace string
  1003. object string
  1004. }
  1005. func (f schemaChangeTable) String() string {
  1006. return fmt.Sprintf("[event schema_change change=%q keyspace=%q object=%q]", f.change, f.keyspace, f.object)
  1007. }
  1008. type schemaChangeType struct {
  1009. frameHeader
  1010. change string
  1011. keyspace string
  1012. object string
  1013. }
  1014. type schemaChangeFunction struct {
  1015. frameHeader
  1016. change string
  1017. keyspace string
  1018. name string
  1019. args []string
  1020. }
  1021. type schemaChangeAggregate struct {
  1022. frameHeader
  1023. change string
  1024. keyspace string
  1025. name string
  1026. args []string
  1027. }
  1028. func (f *framer) parseResultSchemaChange() frame {
  1029. if f.proto <= protoVersion2 {
  1030. change := f.readString()
  1031. keyspace := f.readString()
  1032. table := f.readString()
  1033. if table != "" {
  1034. return &schemaChangeTable{
  1035. frameHeader: *f.header,
  1036. change: change,
  1037. keyspace: keyspace,
  1038. object: table,
  1039. }
  1040. } else {
  1041. return &schemaChangeKeyspace{
  1042. frameHeader: *f.header,
  1043. change: change,
  1044. keyspace: keyspace,
  1045. }
  1046. }
  1047. } else {
  1048. change := f.readString()
  1049. target := f.readString()
  1050. // TODO: could just use a separate type for each target
  1051. switch target {
  1052. case "KEYSPACE":
  1053. frame := &schemaChangeKeyspace{
  1054. frameHeader: *f.header,
  1055. change: change,
  1056. }
  1057. frame.keyspace = f.readString()
  1058. return frame
  1059. case "TABLE":
  1060. frame := &schemaChangeTable{
  1061. frameHeader: *f.header,
  1062. change: change,
  1063. }
  1064. frame.keyspace = f.readString()
  1065. frame.object = f.readString()
  1066. return frame
  1067. case "TYPE":
  1068. frame := &schemaChangeType{
  1069. frameHeader: *f.header,
  1070. change: change,
  1071. }
  1072. frame.keyspace = f.readString()
  1073. frame.object = f.readString()
  1074. return frame
  1075. case "FUNCTION":
  1076. frame := &schemaChangeFunction{
  1077. frameHeader: *f.header,
  1078. change: change,
  1079. }
  1080. frame.keyspace = f.readString()
  1081. frame.name = f.readString()
  1082. frame.args = f.readStringList()
  1083. return frame
  1084. case "AGGREGATE":
  1085. frame := &schemaChangeAggregate{
  1086. frameHeader: *f.header,
  1087. change: change,
  1088. }
  1089. frame.keyspace = f.readString()
  1090. frame.name = f.readString()
  1091. frame.args = f.readStringList()
  1092. return frame
  1093. default:
  1094. panic(fmt.Errorf("gocql: unknown SCHEMA_CHANGE target: %q change: %q", target, change))
  1095. }
  1096. }
  1097. }
  1098. type authenticateFrame struct {
  1099. frameHeader
  1100. class string
  1101. }
  1102. func (a *authenticateFrame) String() string {
  1103. return fmt.Sprintf("[authenticate class=%q]", a.class)
  1104. }
  1105. func (f *framer) parseAuthenticateFrame() frame {
  1106. return &authenticateFrame{
  1107. frameHeader: *f.header,
  1108. class: f.readString(),
  1109. }
  1110. }
  1111. type authSuccessFrame struct {
  1112. frameHeader
  1113. data []byte
  1114. }
  1115. func (a *authSuccessFrame) String() string {
  1116. return fmt.Sprintf("[auth_success data=%q]", a.data)
  1117. }
  1118. func (f *framer) parseAuthSuccessFrame() frame {
  1119. return &authSuccessFrame{
  1120. frameHeader: *f.header,
  1121. data: f.readBytes(),
  1122. }
  1123. }
  1124. type authChallengeFrame struct {
  1125. frameHeader
  1126. data []byte
  1127. }
  1128. func (a *authChallengeFrame) String() string {
  1129. return fmt.Sprintf("[auth_challenge data=%q]", a.data)
  1130. }
  1131. func (f *framer) parseAuthChallengeFrame() frame {
  1132. return &authChallengeFrame{
  1133. frameHeader: *f.header,
  1134. data: f.readBytes(),
  1135. }
  1136. }
  1137. type statusChangeEventFrame struct {
  1138. frameHeader
  1139. change string
  1140. host net.IP
  1141. port int
  1142. }
  1143. func (t statusChangeEventFrame) String() string {
  1144. return fmt.Sprintf("[status_change change=%s host=%v port=%v]", t.change, t.host, t.port)
  1145. }
  1146. // essentially the same as statusChange
  1147. type topologyChangeEventFrame struct {
  1148. frameHeader
  1149. change string
  1150. host net.IP
  1151. port int
  1152. }
  1153. func (t topologyChangeEventFrame) String() string {
  1154. return fmt.Sprintf("[topology_change change=%s host=%v port=%v]", t.change, t.host, t.port)
  1155. }
  1156. func (f *framer) parseEventFrame() frame {
  1157. eventType := f.readString()
  1158. switch eventType {
  1159. case "TOPOLOGY_CHANGE":
  1160. frame := &topologyChangeEventFrame{frameHeader: *f.header}
  1161. frame.change = f.readString()
  1162. frame.host, frame.port = f.readInet()
  1163. return frame
  1164. case "STATUS_CHANGE":
  1165. frame := &statusChangeEventFrame{frameHeader: *f.header}
  1166. frame.change = f.readString()
  1167. frame.host, frame.port = f.readInet()
  1168. return frame
  1169. case "SCHEMA_CHANGE":
  1170. // this should work for all versions
  1171. return f.parseResultSchemaChange()
  1172. default:
  1173. panic(fmt.Errorf("gocql: unknown event type: %q", eventType))
  1174. }
  1175. }
  1176. type writeAuthResponseFrame struct {
  1177. data []byte
  1178. }
  1179. func (a *writeAuthResponseFrame) String() string {
  1180. return fmt.Sprintf("[auth_response data=%q]", a.data)
  1181. }
  1182. func (a *writeAuthResponseFrame) writeFrame(framer *framer, streamID int) error {
  1183. return framer.writeAuthResponseFrame(streamID, a.data)
  1184. }
  1185. func (f *framer) writeAuthResponseFrame(streamID int, data []byte) error {
  1186. f.writeHeader(f.flags, opAuthResponse, streamID)
  1187. f.writeBytes(data)
  1188. return f.finishWrite()
  1189. }
  1190. type queryValues struct {
  1191. value []byte
  1192. // optional name, will set With names for values flag
  1193. name string
  1194. isUnset bool
  1195. }
  1196. type queryParams struct {
  1197. consistency Consistency
  1198. // v2+
  1199. skipMeta bool
  1200. values []queryValues
  1201. pageSize int
  1202. pagingState []byte
  1203. serialConsistency SerialConsistency
  1204. // v3+
  1205. defaultTimestamp bool
  1206. defaultTimestampValue int64
  1207. // v5+
  1208. keyspace string
  1209. }
  1210. func (q queryParams) String() string {
  1211. 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]",
  1212. q.consistency, q.skipMeta, q.pageSize, q.pagingState, q.serialConsistency, q.defaultTimestamp, q.values, q.keyspace)
  1213. }
  1214. func (f *framer) writeQueryParams(opts *queryParams) {
  1215. f.writeConsistency(opts.consistency)
  1216. if f.proto == protoVersion1 {
  1217. return
  1218. }
  1219. var flags byte
  1220. if len(opts.values) > 0 {
  1221. flags |= flagValues
  1222. }
  1223. if opts.skipMeta {
  1224. flags |= flagSkipMetaData
  1225. }
  1226. if opts.pageSize > 0 {
  1227. flags |= flagPageSize
  1228. }
  1229. if len(opts.pagingState) > 0 {
  1230. flags |= flagWithPagingState
  1231. }
  1232. if opts.serialConsistency > 0 {
  1233. flags |= flagWithSerialConsistency
  1234. }
  1235. names := false
  1236. // protoV3 specific things
  1237. if f.proto > protoVersion2 {
  1238. if opts.defaultTimestamp {
  1239. flags |= flagDefaultTimestamp
  1240. }
  1241. if len(opts.values) > 0 && opts.values[0].name != "" {
  1242. flags |= flagWithNameValues
  1243. names = true
  1244. }
  1245. }
  1246. if opts.keyspace != "" {
  1247. if f.proto > protoVersion4 {
  1248. flags |= flagWithKeyspace
  1249. } else {
  1250. panic(fmt.Errorf("The keyspace can only be set with protocol 5 or higher"))
  1251. }
  1252. }
  1253. if f.proto > protoVersion4 {
  1254. f.writeUint(uint32(flags))
  1255. } else {
  1256. f.writeByte(flags)
  1257. }
  1258. if n := len(opts.values); n > 0 {
  1259. f.writeShort(uint16(n))
  1260. for i := 0; i < n; i++ {
  1261. if names {
  1262. f.writeString(opts.values[i].name)
  1263. }
  1264. if opts.values[i].isUnset {
  1265. f.writeUnset()
  1266. } else {
  1267. f.writeBytes(opts.values[i].value)
  1268. }
  1269. }
  1270. }
  1271. if opts.pageSize > 0 {
  1272. f.writeInt(int32(opts.pageSize))
  1273. }
  1274. if len(opts.pagingState) > 0 {
  1275. f.writeBytes(opts.pagingState)
  1276. }
  1277. if opts.serialConsistency > 0 {
  1278. f.writeConsistency(Consistency(opts.serialConsistency))
  1279. }
  1280. if f.proto > protoVersion2 && opts.defaultTimestamp {
  1281. // timestamp in microseconds
  1282. var ts int64
  1283. if opts.defaultTimestampValue != 0 {
  1284. ts = opts.defaultTimestampValue
  1285. } else {
  1286. ts = time.Now().UnixNano() / 1000
  1287. }
  1288. f.writeLong(ts)
  1289. }
  1290. if opts.keyspace != "" {
  1291. f.writeString(opts.keyspace)
  1292. }
  1293. }
  1294. type writeQueryFrame struct {
  1295. statement string
  1296. params queryParams
  1297. // v4+
  1298. customPayload map[string][]byte
  1299. }
  1300. func (w *writeQueryFrame) String() string {
  1301. return fmt.Sprintf("[query statement=%q params=%v]", w.statement, w.params)
  1302. }
  1303. func (w *writeQueryFrame) writeFrame(framer *framer, streamID int) error {
  1304. return framer.writeQueryFrame(streamID, w.statement, &w.params, w.customPayload)
  1305. }
  1306. func (f *framer) writeQueryFrame(streamID int, statement string, params *queryParams, customPayload map[string][]byte) error {
  1307. if len(customPayload) > 0 {
  1308. f.payload()
  1309. }
  1310. f.writeHeader(f.flags, opQuery, streamID)
  1311. f.writeCustomPayload(&customPayload)
  1312. f.writeLongString(statement)
  1313. f.writeQueryParams(params)
  1314. return f.finishWrite()
  1315. }
  1316. type frameWriter interface {
  1317. writeFrame(framer *framer, streamID int) error
  1318. }
  1319. type frameWriterFunc func(framer *framer, streamID int) error
  1320. func (f frameWriterFunc) writeFrame(framer *framer, streamID int) error {
  1321. return f(framer, streamID)
  1322. }
  1323. type writeExecuteFrame struct {
  1324. preparedID []byte
  1325. params queryParams
  1326. // v4+
  1327. customPayload map[string][]byte
  1328. }
  1329. func (e *writeExecuteFrame) String() string {
  1330. return fmt.Sprintf("[execute id=% X params=%v]", e.preparedID, &e.params)
  1331. }
  1332. func (e *writeExecuteFrame) writeFrame(fr *framer, streamID int) error {
  1333. return fr.writeExecuteFrame(streamID, e.preparedID, &e.params, &e.customPayload)
  1334. }
  1335. func (f *framer) writeExecuteFrame(streamID int, preparedID []byte, params *queryParams, customPayload *map[string][]byte) error {
  1336. if len(*customPayload) > 0 {
  1337. f.payload()
  1338. }
  1339. f.writeHeader(f.flags, opExecute, streamID)
  1340. f.writeCustomPayload(customPayload)
  1341. f.writeShortBytes(preparedID)
  1342. if f.proto > protoVersion1 {
  1343. f.writeQueryParams(params)
  1344. } else {
  1345. n := len(params.values)
  1346. f.writeShort(uint16(n))
  1347. for i := 0; i < n; i++ {
  1348. if params.values[i].isUnset {
  1349. f.writeUnset()
  1350. } else {
  1351. f.writeBytes(params.values[i].value)
  1352. }
  1353. }
  1354. f.writeConsistency(params.consistency)
  1355. }
  1356. return f.finishWrite()
  1357. }
  1358. // TODO: can we replace BatchStatemt with batchStatement? As they prety much
  1359. // duplicate each other
  1360. type batchStatment struct {
  1361. preparedID []byte
  1362. statement string
  1363. values []queryValues
  1364. }
  1365. type writeBatchFrame struct {
  1366. typ BatchType
  1367. statements []batchStatment
  1368. consistency Consistency
  1369. // v3+
  1370. serialConsistency SerialConsistency
  1371. defaultTimestamp bool
  1372. defaultTimestampValue int64
  1373. //v4+
  1374. customPayload map[string][]byte
  1375. }
  1376. func (w *writeBatchFrame) writeFrame(framer *framer, streamID int) error {
  1377. return framer.writeBatchFrame(streamID, w, w.customPayload)
  1378. }
  1379. func (f *framer) writeBatchFrame(streamID int, w *writeBatchFrame, customPayload map[string][]byte) error {
  1380. if len(customPayload) > 0 {
  1381. f.payload()
  1382. }
  1383. f.writeHeader(f.flags, opBatch, streamID)
  1384. f.writeCustomPayload(&customPayload)
  1385. f.writeByte(byte(w.typ))
  1386. n := len(w.statements)
  1387. f.writeShort(uint16(n))
  1388. var flags byte
  1389. for i := 0; i < n; i++ {
  1390. b := &w.statements[i]
  1391. if len(b.preparedID) == 0 {
  1392. f.writeByte(0)
  1393. f.writeLongString(b.statement)
  1394. } else {
  1395. f.writeByte(1)
  1396. f.writeShortBytes(b.preparedID)
  1397. }
  1398. f.writeShort(uint16(len(b.values)))
  1399. for j := range b.values {
  1400. col := b.values[j]
  1401. if f.proto > protoVersion2 && col.name != "" {
  1402. // TODO: move this check into the caller and set a flag on writeBatchFrame
  1403. // to indicate using named values
  1404. if f.proto <= protoVersion5 {
  1405. return fmt.Errorf("gocql: named query values are not supported in batches, please see https://issues.apache.org/jira/browse/CASSANDRA-10246")
  1406. }
  1407. flags |= flagWithNameValues
  1408. f.writeString(col.name)
  1409. }
  1410. if col.isUnset {
  1411. f.writeUnset()
  1412. } else {
  1413. f.writeBytes(col.value)
  1414. }
  1415. }
  1416. }
  1417. f.writeConsistency(w.consistency)
  1418. if f.proto > protoVersion2 {
  1419. if w.serialConsistency > 0 {
  1420. flags |= flagWithSerialConsistency
  1421. }
  1422. if w.defaultTimestamp {
  1423. flags |= flagDefaultTimestamp
  1424. }
  1425. if f.proto > protoVersion4 {
  1426. f.writeUint(uint32(flags))
  1427. } else {
  1428. f.writeByte(flags)
  1429. }
  1430. if w.serialConsistency > 0 {
  1431. f.writeConsistency(Consistency(w.serialConsistency))
  1432. }
  1433. if w.defaultTimestamp {
  1434. var ts int64
  1435. if w.defaultTimestampValue != 0 {
  1436. ts = w.defaultTimestampValue
  1437. } else {
  1438. ts = time.Now().UnixNano() / 1000
  1439. }
  1440. f.writeLong(ts)
  1441. }
  1442. }
  1443. return f.finishWrite()
  1444. }
  1445. type writeOptionsFrame struct{}
  1446. func (w *writeOptionsFrame) writeFrame(framer *framer, streamID int) error {
  1447. return framer.writeOptionsFrame(streamID, w)
  1448. }
  1449. func (f *framer) writeOptionsFrame(stream int, _ *writeOptionsFrame) error {
  1450. f.writeHeader(f.flags&^flagCompress, opOptions, stream)
  1451. return f.finishWrite()
  1452. }
  1453. type writeRegisterFrame struct {
  1454. events []string
  1455. }
  1456. func (w *writeRegisterFrame) writeFrame(framer *framer, streamID int) error {
  1457. return framer.writeRegisterFrame(streamID, w)
  1458. }
  1459. func (f *framer) writeRegisterFrame(streamID int, w *writeRegisterFrame) error {
  1460. f.writeHeader(f.flags, opRegister, streamID)
  1461. f.writeStringList(w.events)
  1462. return f.finishWrite()
  1463. }
  1464. func (f *framer) readByte() byte {
  1465. if len(f.rbuf) < 1 {
  1466. panic(fmt.Errorf("not enough bytes in buffer to read byte require 1 got: %d", len(f.rbuf)))
  1467. }
  1468. b := f.rbuf[0]
  1469. f.rbuf = f.rbuf[1:]
  1470. return b
  1471. }
  1472. func (f *framer) readInt() (n int) {
  1473. if len(f.rbuf) < 4 {
  1474. panic(fmt.Errorf("not enough bytes in buffer to read int require 4 got: %d", len(f.rbuf)))
  1475. }
  1476. n = int(int32(f.rbuf[0])<<24 | int32(f.rbuf[1])<<16 | int32(f.rbuf[2])<<8 | int32(f.rbuf[3]))
  1477. f.rbuf = f.rbuf[4:]
  1478. return
  1479. }
  1480. func (f *framer) readShort() (n uint16) {
  1481. if len(f.rbuf) < 2 {
  1482. panic(fmt.Errorf("not enough bytes in buffer to read short require 2 got: %d", len(f.rbuf)))
  1483. }
  1484. n = uint16(f.rbuf[0])<<8 | uint16(f.rbuf[1])
  1485. f.rbuf = f.rbuf[2:]
  1486. return
  1487. }
  1488. func (f *framer) readLong() (n int64) {
  1489. if len(f.rbuf) < 8 {
  1490. panic(fmt.Errorf("not enough bytes in buffer to read long require 8 got: %d", len(f.rbuf)))
  1491. }
  1492. n = int64(f.rbuf[0])<<56 | int64(f.rbuf[1])<<48 | int64(f.rbuf[2])<<40 | int64(f.rbuf[3])<<32 |
  1493. int64(f.rbuf[4])<<24 | int64(f.rbuf[5])<<16 | int64(f.rbuf[6])<<8 | int64(f.rbuf[7])
  1494. f.rbuf = f.rbuf[8:]
  1495. return
  1496. }
  1497. func (f *framer) readString() (s string) {
  1498. size := f.readShort()
  1499. if len(f.rbuf) < int(size) {
  1500. panic(fmt.Errorf("not enough bytes in buffer to read string require %d got: %d", size, len(f.rbuf)))
  1501. }
  1502. s = string(f.rbuf[:size])
  1503. f.rbuf = f.rbuf[size:]
  1504. return
  1505. }
  1506. func (f *framer) readLongString() (s string) {
  1507. size := f.readInt()
  1508. if len(f.rbuf) < size {
  1509. panic(fmt.Errorf("not enough bytes in buffer to read long string require %d got: %d", size, len(f.rbuf)))
  1510. }
  1511. s = string(f.rbuf[:size])
  1512. f.rbuf = f.rbuf[size:]
  1513. return
  1514. }
  1515. func (f *framer) readUUID() *UUID {
  1516. if len(f.rbuf) < 16 {
  1517. panic(fmt.Errorf("not enough bytes in buffer to read uuid require %d got: %d", 16, len(f.rbuf)))
  1518. }
  1519. // TODO: how to handle this error, if it is a uuid, then sureley, problems?
  1520. u, _ := UUIDFromBytes(f.rbuf[:16])
  1521. f.rbuf = f.rbuf[16:]
  1522. return &u
  1523. }
  1524. func (f *framer) readStringList() []string {
  1525. size := f.readShort()
  1526. l := make([]string, size)
  1527. for i := 0; i < int(size); i++ {
  1528. l[i] = f.readString()
  1529. }
  1530. return l
  1531. }
  1532. func (f *framer) readBytesInternal() ([]byte, error) {
  1533. size := f.readInt()
  1534. if size < 0 {
  1535. return nil, nil
  1536. }
  1537. if len(f.rbuf) < size {
  1538. return nil, fmt.Errorf("not enough bytes in buffer to read bytes require %d got: %d", size, len(f.rbuf))
  1539. }
  1540. l := f.rbuf[:size]
  1541. f.rbuf = f.rbuf[size:]
  1542. return l, nil
  1543. }
  1544. func (f *framer) readBytes() []byte {
  1545. l, err := f.readBytesInternal()
  1546. if err != nil {
  1547. panic(err)
  1548. }
  1549. return l
  1550. }
  1551. func (f *framer) readShortBytes() []byte {
  1552. size := f.readShort()
  1553. if len(f.rbuf) < int(size) {
  1554. panic(fmt.Errorf("not enough bytes in buffer to read short bytes: require %d got %d", size, len(f.rbuf)))
  1555. }
  1556. l := f.rbuf[:size]
  1557. f.rbuf = f.rbuf[size:]
  1558. return l
  1559. }
  1560. func (f *framer) readInetAdressOnly() net.IP {
  1561. if len(f.rbuf) < 1 {
  1562. panic(fmt.Errorf("not enough bytes in buffer to read inet size require %d got: %d", 1, len(f.rbuf)))
  1563. }
  1564. size := f.rbuf[0]
  1565. f.rbuf = f.rbuf[1:]
  1566. if !(size == 4 || size == 16) {
  1567. panic(fmt.Errorf("invalid IP size: %d", size))
  1568. }
  1569. if len(f.rbuf) < 1 {
  1570. panic(fmt.Errorf("not enough bytes in buffer to read inet require %d got: %d", size, len(f.rbuf)))
  1571. }
  1572. ip := make([]byte, size)
  1573. copy(ip, f.rbuf[:size])
  1574. f.rbuf = f.rbuf[size:]
  1575. return net.IP(ip)
  1576. }
  1577. func (f *framer) readInet() (net.IP, int) {
  1578. return f.readInetAdressOnly(), f.readInt()
  1579. }
  1580. func (f *framer) readConsistency() Consistency {
  1581. return Consistency(f.readShort())
  1582. }
  1583. func (f *framer) readStringMap() map[string]string {
  1584. size := f.readShort()
  1585. m := make(map[string]string, size)
  1586. for i := 0; i < int(size); i++ {
  1587. k := f.readString()
  1588. v := f.readString()
  1589. m[k] = v
  1590. }
  1591. return m
  1592. }
  1593. func (f *framer) readBytesMap() map[string][]byte {
  1594. size := f.readShort()
  1595. m := make(map[string][]byte, size)
  1596. for i := 0; i < int(size); i++ {
  1597. k := f.readString()
  1598. v := f.readBytes()
  1599. m[k] = v
  1600. }
  1601. return m
  1602. }
  1603. func (f *framer) readStringMultiMap() map[string][]string {
  1604. size := f.readShort()
  1605. m := make(map[string][]string, size)
  1606. for i := 0; i < int(size); i++ {
  1607. k := f.readString()
  1608. v := f.readStringList()
  1609. m[k] = v
  1610. }
  1611. return m
  1612. }
  1613. func (f *framer) writeByte(b byte) {
  1614. f.wbuf = append(f.wbuf, b)
  1615. }
  1616. func appendBytes(p []byte, d []byte) []byte {
  1617. if d == nil {
  1618. return appendInt(p, -1)
  1619. }
  1620. p = appendInt(p, int32(len(d)))
  1621. p = append(p, d...)
  1622. return p
  1623. }
  1624. func appendShort(p []byte, n uint16) []byte {
  1625. return append(p,
  1626. byte(n>>8),
  1627. byte(n),
  1628. )
  1629. }
  1630. func appendInt(p []byte, n int32) []byte {
  1631. return append(p, byte(n>>24),
  1632. byte(n>>16),
  1633. byte(n>>8),
  1634. byte(n))
  1635. }
  1636. func appendUint(p []byte, n uint32) []byte {
  1637. return append(p, byte(n>>24),
  1638. byte(n>>16),
  1639. byte(n>>8),
  1640. byte(n))
  1641. }
  1642. func appendLong(p []byte, n int64) []byte {
  1643. return append(p,
  1644. byte(n>>56),
  1645. byte(n>>48),
  1646. byte(n>>40),
  1647. byte(n>>32),
  1648. byte(n>>24),
  1649. byte(n>>16),
  1650. byte(n>>8),
  1651. byte(n),
  1652. )
  1653. }
  1654. func (f *framer) writeCustomPayload(customPayload *map[string][]byte) {
  1655. if len(*customPayload) > 0 {
  1656. if f.proto < protoVersion4 {
  1657. panic("Custom payload is not supported with version V3 or less")
  1658. }
  1659. f.writeBytesMap(*customPayload)
  1660. }
  1661. }
  1662. // these are protocol level binary types
  1663. func (f *framer) writeInt(n int32) {
  1664. f.wbuf = appendInt(f.wbuf, n)
  1665. }
  1666. func (f *framer) writeUint(n uint32) {
  1667. f.wbuf = appendUint(f.wbuf, n)
  1668. }
  1669. func (f *framer) writeShort(n uint16) {
  1670. f.wbuf = appendShort(f.wbuf, n)
  1671. }
  1672. func (f *framer) writeLong(n int64) {
  1673. f.wbuf = appendLong(f.wbuf, n)
  1674. }
  1675. func (f *framer) writeString(s string) {
  1676. f.writeShort(uint16(len(s)))
  1677. f.wbuf = append(f.wbuf, s...)
  1678. }
  1679. func (f *framer) writeLongString(s string) {
  1680. f.writeInt(int32(len(s)))
  1681. f.wbuf = append(f.wbuf, s...)
  1682. }
  1683. func (f *framer) writeUUID(u *UUID) {
  1684. f.wbuf = append(f.wbuf, u[:]...)
  1685. }
  1686. func (f *framer) writeStringList(l []string) {
  1687. f.writeShort(uint16(len(l)))
  1688. for _, s := range l {
  1689. f.writeString(s)
  1690. }
  1691. }
  1692. func (f *framer) writeUnset() {
  1693. // Protocol version 4 specifies that bind variables do not require having a
  1694. // value when executing a statement. Bind variables without a value are
  1695. // called 'unset'. The 'unset' bind variable is serialized as the int
  1696. // value '-2' without following bytes.
  1697. f.writeInt(-2)
  1698. }
  1699. func (f *framer) writeBytes(p []byte) {
  1700. // TODO: handle null case correctly,
  1701. // [bytes] A [int] n, followed by n bytes if n >= 0. If n < 0,
  1702. // no byte should follow and the value represented is `null`.
  1703. if p == nil {
  1704. f.writeInt(-1)
  1705. } else {
  1706. f.writeInt(int32(len(p)))
  1707. f.wbuf = append(f.wbuf, p...)
  1708. }
  1709. }
  1710. func (f *framer) writeShortBytes(p []byte) {
  1711. f.writeShort(uint16(len(p)))
  1712. f.wbuf = append(f.wbuf, p...)
  1713. }
  1714. func (f *framer) writeInet(ip net.IP, port int) {
  1715. f.wbuf = append(f.wbuf,
  1716. byte(len(ip)),
  1717. )
  1718. f.wbuf = append(f.wbuf,
  1719. []byte(ip)...,
  1720. )
  1721. f.writeInt(int32(port))
  1722. }
  1723. func (f *framer) writeConsistency(cons Consistency) {
  1724. f.writeShort(uint16(cons))
  1725. }
  1726. func (f *framer) writeStringMap(m map[string]string) {
  1727. f.writeShort(uint16(len(m)))
  1728. for k, v := range m {
  1729. f.writeString(k)
  1730. f.writeString(v)
  1731. }
  1732. }
  1733. func (f *framer) writeBytesMap(m map[string][]byte) {
  1734. f.writeShort(uint16(len(m)))
  1735. for k, v := range m {
  1736. f.writeString(k)
  1737. f.writeBytes(v)
  1738. }
  1739. }