Rooba
/
odyssey
mirror of https://github.com/yandex/odyssey.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

shapito: add support for backend sasl packets

This commit is contained in:
Dmitry Simonenko 2018-06-21 16:57:14 +03:00
parent 9eecb58980
commit b1be72d097
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
third_party/shapito/shapito/be_write.h vendored
View File

@ -115,6 +115,45 @@ shapito_be_write_authentication_md5(shapito_stream_t *stream, char salt[4])
return 0;
}
SHAPITO_API static inline int
shapito_be_write_authentication_sasl(shapito_stream_t *stream, char *method, int method_len)
{
int rc = shapito_stream_ensure(stream, sizeof(shapito_header_t) + sizeof(uint32_t) + method_len);
if (shapito_unlikely(rc == -1))
return -1;
shapito_stream_write8(stream, SHAPITO_BE_AUTHENTICATION);
shapito_stream_write32(stream, sizeof(uint32_t) + sizeof(uint32_t) + method_len);
shapito_stream_write32(stream, 10);
shapito_stream_write(stream, method, method_len);
return 0;
}
SHAPITO_API static inline int
shapito_be_write_authentication_sasl_continue(shapito_stream_t *stream, char *data, int data_len)
{
int rc = shapito_stream_ensure(stream, sizeof(shapito_header_t) + sizeof(uint32_t) + data_len);
if (shapito_unlikely(rc == -1))
return -1;
shapito_stream_write8(stream, SHAPITO_BE_AUTHENTICATION);
shapito_stream_write32(stream, sizeof(uint32_t) + sizeof(uint32_t) + data_len);
shapito_stream_write32(stream, 11);
shapito_stream_write(stream, data, data_len);
return 0;
}
SHAPITO_API static inline int
shapito_be_write_authentication_sasl_final(shapito_stream_t *stream, char *data, int data_len)
{
int rc = shapito_stream_ensure(stream, sizeof(shapito_header_t) + sizeof(uint32_t) + data_len);
if (shapito_unlikely(rc == -1))
return -1;
shapito_stream_write8(stream, SHAPITO_BE_AUTHENTICATION);
shapito_stream_write32(stream, sizeof(uint32_t) + sizeof(uint32_t) + data_len);
shapito_stream_write32(stream, 12);
shapito_stream_write(stream, data, data_len);
return 0;
}
SHAPITO_API static inline int
shapito_be_write_backend_key_data(shapito_stream_t *stream, uint32_t pid, uint32_t key)
{