Packages

efirebirdsql

0.4.0
0.9.14 0.9.13 0.9.12 0.9.11 0.9.10 0.9.9 0.9.8 0.9.7 0.9.6 0.9.5 0.9.4 0.9.3 0.9.2 0.9.1 0.9.0 0.8.5 0.8.4 0.8.3 0.8.2 0.8.1 0.8.0 0.7.0 0.6.5 0.6.4 0.6.3 0.6.2 0.6.1 0.6.0 0.5.10 0.5.9 0.5.8 0.5.7 0.5.6 0.5.5 0.5.4 0.5.3 0.5.2 0.5.1 0.5.0 0.4.0 0.3.0 0.2.3 0.2.2 0.2.1 0.2.0 0.1.4 0.1.3 0.1.2 0.1.1

Firebird Client

HexDocs

Current section

Files

Jump to
Readme
49 Versions
0 Dependencies
Dependants
Files
Activity
efirebirdsql src efirebirdsql_socket.erl
Raw

src/efirebirdsql_socket.erl 

%%% The MIT License (MIT)

%%% Copyright (c) 2019 Hajime Nakagami<nakagami@gmail.com>



-module(efirebirdsql_socket).



-export([send/2, recv/2, recv_align/2, recv_null_bitmap/2]).



-include("efirebirdsql.hrl").



send(State, Data) when State#state.write_state =:= undefined ->

    gen_tcp:send(State#state.sock, Data),

    State;

send(State, Message) ->

    {NewWriteState, Encrypted} = crypto:stream_encrypt(State#state.write_state, Message),

    gen_tcp:send(State#state.sock, Encrypted),

    State#state{write_state=NewWriteState}.



recv(State, Len) when Len =:= 0 ->

    {ok, [], State};

recv(State, Len) when State#state.read_state =:= undefined ->

    {T, V} = gen_tcp:recv(State#state.sock, Len),

    {T, V, State};

recv(State, Len) ->

    {T, Encrypted} = gen_tcp:recv(State#state.sock, Len),

    {NewReadState, Message} = crypto:stream_decrypt(State#state.read_state, Encrypted),

    {T, Message, State#state{read_state=NewReadState}}.



recv_align(State, Len) ->

    {T, V, S2} = recv(State, Len),

    S4 = case Len rem 4 of

        0 -> S2;

        1 -> {_, _, S3} = recv(S2, 3), S3;

        2 -> {_, _, S3} = recv(S2, 2), S3;

        3 -> {_, _, S3} = recv(S2, 1), S3

        end,

    {T, V, S4}.



recv_null_bitmap(State, BitLen) when BitLen =:= 0 ->

    {[], State};

recv_null_bitmap(State, BitLen) ->

    Div8 = BitLen div 8,

    Len = if

        BitLen rem 8 =:= 0 -> Div8;

        BitLen rem 8 =/= 0 -> Div8 + 1

        end,

    {ok, Buf, S2} = recv_align(State, Len),

    <<Bitmap:Len/little-unit:8>> = Buf,

    {Bitmap, S2}.