summaryrefslogtreecommitdiff
path: root/src/pgsql/pgsql_util.erl
diff options
context:
space:
mode:
Diffstat (limited to 'src/pgsql/pgsql_util.erl')
-rw-r--r--src/pgsql/pgsql_util.erl321
1 files changed, 0 insertions, 321 deletions
diff --git a/src/pgsql/pgsql_util.erl b/src/pgsql/pgsql_util.erl
deleted file mode 100644
index d562f297..00000000
--- a/src/pgsql/pgsql_util.erl
+++ /dev/null
@@ -1,321 +0,0 @@
-%%% File : pgsql_util.erl
-%%% Author : Christian Sunesson
-%%% Description : utility functions used in implementation of
-%%% postgresql driver.
-%%% Created : 11 May 2005 by Blah <cos@local>
-
--module(pgsql_util).
-
-%% Key-Value handling
--export([option/3]).
-
-%% Networking
--export([socket/1]).
--export([send/2, send_int/2, send_msg/3]).
--export([recv_msg/2, recv_msg/1, recv_byte/2, recv_byte/1]).
-
-%% Protocol packing
--export([string/1, make_pair/2, split_pair/2]).
--export([split_pair_rec/2]).
--export([count_string/1, to_string/2]).
--export([oids/2, coldescs/3, datacoldescs/3]).
--export([decode_row/3, decode_descs/2]).
--export([errordesc/2]).
--export([to_integer/1, to_atom/1]).
-
--export([zip/2]).
-
-%% Constructing authentication messages.
--export([pass_plain/1, pass_md5/3]).
--import(erlang, [md5/1]).
--export([hexlist/2]).
-
-%% Lookup key in a plist stored in process dictionary under 'options'.
-%% Default is returned if there is no value for Key in the plist.
-option(Opts, Key, Default) ->
- case proplists:get_value(Key, Opts, Default) of
- Default ->
- Default;
- Value when is_binary(Value) ->
- binary_to_list(Value);
- Value ->
- Value
- end.
-
-
-%% Open a TCP connection
-socket({tcp, Host, Port}) ->
- gen_tcp:connect(Host, Port, [{active, false}, binary, {packet, raw}], 5000).
-
-send(Sock, Packet) ->
- gen_tcp:send(Sock, Packet).
-send_int(Sock, Int) ->
- Packet = <<Int:32/integer>>,
- gen_tcp:send(Sock, Packet).
-
-send_msg(Sock, Code, Packet) when is_binary(Packet) ->
- Len = size(Packet) + 4,
- Msg = <<Code:8/integer, Len:4/integer-unit:8, Packet/binary>>,
- gen_tcp:send(Sock, Msg).
-
-recv_msg(Sock, Timeout) ->
- {ok, Head} = gen_tcp:recv(Sock, 5, Timeout),
- <<Code:8/integer, Size:4/integer-unit:8>> = Head,
- %%io:format("Code: ~p, Size: ~p~n", [Code, Size]),
- if
- Size > 4 ->
- {ok, Packet} = gen_tcp:recv(Sock, Size-4, Timeout),
- {ok, Code, Packet};
- true ->
- {ok, Code, <<>>}
- end.
-recv_msg(Sock) ->
- recv_msg(Sock, infinity).
-
-
-recv_byte(Sock) ->
- recv_byte(Sock, infinity).
-recv_byte(Sock, Timeout) ->
- case gen_tcp:recv(Sock, 1, Timeout) of
- {ok, <<Byte:1/integer-unit:8>>} ->
- {ok, Byte};
- E={error, _Reason} ->
- throw(E)
- end.
-
-%% Convert String to binary
-string(String) when is_list(String) ->
- Bin = list_to_binary(String),
- <<Bin/binary, 0/integer>>;
-string(Bin) when is_binary(Bin) ->
- <<Bin/binary, 0/integer>>.
-
-%%% Two zero terminated strings.
-make_pair(Key, Value) when is_atom(Key) ->
- make_pair(atom_to_list(Key), Value);
-make_pair(Key, Value) when is_atom(Value) ->
- make_pair(Key, atom_to_list(Value));
-make_pair(Key, Value) when is_list(Key), is_list(Value) ->
- BinKey = list_to_binary(Key),
- BinValue = list_to_binary(Value),
- make_pair(BinKey, BinValue);
-make_pair(Key, Value) when is_binary(Key), is_binary(Value) ->
- <<Key/binary, 0/integer,
- Value/binary, 0/integer>>.
-
-split_pair(Bin, AsBin) when is_binary(Bin) ->
- split_pair(binary_to_list(Bin), AsBin);
-split_pair(Str, AsBin) ->
- split_pair_rec(Str, norec, AsBin).
-
-split_pair_rec(Bin, AsBin) when is_binary(Bin) ->
- split_pair_rec(binary_to_list(Bin), AsBin);
-split_pair_rec(Arg, AsBin) ->
- split_pair_rec(Arg,[], AsBin).
-
-split_pair_rec([], Acc, _AsBin) ->
- lists:reverse(Acc);
-split_pair_rec([0], Acc, _AsBin) ->
- lists:reverse(Acc);
-split_pair_rec(S, Acc, AsBin) ->
- Fun = fun(C) -> C /= 0 end,
- {K, [0|S1]} = lists:splitwith(Fun, S),
- {V, [0|Tail]} = lists:splitwith(Fun, S1),
- {Key, Value} = if AsBin ->
- {list_to_binary(K), list_to_binary(V)};
- true ->
- {K, V}
- end,
- case Acc of
- norec -> {Key, Value};
- _ ->
- split_pair_rec(Tail, [{Key, Value}| Acc], AsBin)
- end.
-
-
-count_string(Bin) when is_binary(Bin) ->
- count_string(Bin, 0).
-
-count_string(<<>>, N) ->
- {N, <<>>};
-count_string(<<0/integer, Rest/binary>>, N) ->
- {N, Rest};
-count_string(<<_C/integer, Rest/binary>>, N) ->
- count_string(Rest, N+1).
-
-to_string(Bin, AsBin) when is_binary(Bin) ->
- {Count, _} = count_string(Bin, 0),
- <<String:Count/binary, _/binary>> = Bin,
- if AsBin ->
- {String, Count};
- true ->
- {binary_to_list(String), Count}
- end.
-
-oids(<<>>, Oids) ->
- lists:reverse(Oids);
-oids(<<Oid:32/integer, Rest/binary>>, Oids) ->
- oids(Rest, [Oid|Oids]).
-
-coldescs(<<>>, Descs, _AsBin) ->
- lists:reverse(Descs);
-coldescs(Bin, Descs, AsBin) ->
- {Name, Count} = to_string(Bin, AsBin),
- <<_:Count/binary, 0/integer,
- TableOID:32/integer,
- ColumnNumber:16/integer,
- TypeId:32/integer,
- TypeSize:16/integer-signed,
- TypeMod:32/integer-signed,
- FormatCode:16/integer,
- Rest/binary>> = Bin,
- Format = case FormatCode of
- 0 -> text;
- 1 -> binary
- end,
- Desc = {Name, Format, ColumnNumber,
- TypeId, TypeSize, TypeMod,
- TableOID},
- coldescs(Rest, [Desc|Descs], AsBin).
-
-datacoldescs(N, <<16#ffffffff:32, Rest/binary>>, Descs) when N >= 0 ->
- datacoldescs(N-1, Rest, [null|Descs]);
-datacoldescs(N,
- <<Len:32/integer, Data:Len/binary, Rest/binary>>,
- Descs) when N >= 0 ->
- datacoldescs(N-1, Rest, [Data|Descs]);
-datacoldescs(_N, _, Descs) ->
- lists:reverse(Descs).
-
-decode_descs(OidMap, Cols) ->
- decode_descs(OidMap, Cols, []).
-decode_descs(_OidMap, [], Descs) ->
- {ok, lists:reverse(Descs)};
-decode_descs(OidMap, [Col|ColTail], Descs) ->
- {Name, Format, ColNumber, Oid, _, _, _} = Col,
- OidName = dict:fetch(Oid, OidMap),
- decode_descs(OidMap, ColTail, [{Name, Format, ColNumber, OidName, [], [], []}|Descs]).
-
-decode_row(Types, Values, AsBin) ->
- decode_row(Types, Values, [], AsBin).
-decode_row([], [], Out, _AsBin) ->
- {ok, lists:reverse(Out)};
-decode_row([Type|TypeTail], [Value|ValueTail], Out0, AsBin) ->
- Out1 = decode_col(Type, Value, AsBin),
- decode_row(TypeTail, ValueTail, [Out1|Out0], AsBin).
-
-decode_col({_, text, _, _, _, _, _}, Value, AsBin) ->
- if AsBin -> Value;
- true -> binary_to_list(Value)
- end;
-decode_col({_Name, _Format, _ColNumber, varchar, _Size, _Modifier, _TableOID}, Value, AsBin) ->
- if AsBin -> Value;
- true -> binary_to_list(Value)
- end;
-decode_col({_Name, _Format, _ColNumber, int4, _Size, _Modifier, _TableOID}, Value, _AsBin) ->
- <<Int4:32/integer>> = Value,
- Int4;
-decode_col({_Name, _Format, _ColNumber, Oid, _Size, _Modifier, _TableOID}, Value, _AsBin) ->
- {Oid, Value}.
-
-errordesc(Bin, AsBin) ->
- errordesc(Bin, [], AsBin).
-
-errordesc(<<0/integer, _Rest/binary>>, Lines, _AsBin) ->
- lists:reverse(Lines);
-errordesc(<<Code/integer, Rest/binary>>, Lines, AsBin) ->
- {String, Count} = to_string(Rest, AsBin),
- <<_:Count/binary, 0, Rest1/binary>> = Rest,
- Msg = case Code of
- $S ->
- {severity, to_atom(String)};
- $C ->
- {code, String};
- $M ->
- {message, String};
- $D ->
- {detail, String};
- $H ->
- {hint, String};
- $P ->
- {position, to_integer(String)};
- $p ->
- {internal_position, to_integer(String)};
- $W ->
- {where, String};
- $F ->
- {file, String};
- $L ->
- {line, to_integer(String)};
- $R ->
- {routine, String};
- Unknown ->
- {Unknown, String}
- end,
- errordesc(Rest1, [Msg|Lines]).
-
-%%% Zip two lists together
-zip(List1, List2) ->
- zip(List1, List2, []).
-zip(List1, List2, Result) when List1 =:= [];
- List2 =:= [] ->
- lists:reverse(Result);
-zip([H1|List1], [H2|List2], Result) ->
- zip(List1, List2, [{H1, H2}|Result]).
-
-%%% Authentication utils
-
-pass_plain(Password) ->
- Pass = [Password, 0],
- list_to_binary(Pass).
-
-%% MD5 authentication patch from
-%% Juhani Rankimies <juhani@juranki.com>
-%% (patch slightly rewritten, new bugs are mine :] /Christian Sunesson)
-
-%%
-%% MD5(MD5(password + user) + salt)
-%%
-
-pass_md5(User, Password, Salt) ->
- Digest = hex(md5([Password, User])),
- Encrypt = hex(md5([Digest, Salt])),
- Pass = ["md5", Encrypt, 0],
- list_to_binary(Pass).
-
-to_integer(B) when is_binary(B) ->
- to_integer(binary_to_list(B));
-to_integer(S) ->
- list_to_integer(S).
-
-to_atom(B) when is_binary(B) ->
- to_atom(binary_to_list(B));
-to_atom(S) ->
- list_to_atom(S).
-
-hex(B) when is_binary(B) ->
- hexlist(binary_to_list(B), []).
-
-hexlist([], Acc) ->
- lists:reverse(Acc);
-hexlist([N|Rest], Acc) ->
- HighNibble = (N band 16#f0) bsr 4,
- LowNibble = (N band 16#0f),
- hexlist(Rest, [hexdigit(LowNibble), hexdigit(HighNibble)|Acc]).
-
-hexdigit(0) -> $0;
-hexdigit(1) -> $1;
-hexdigit(2) -> $2;
-hexdigit(3) -> $3;
-hexdigit(4) -> $4;
-hexdigit(5) -> $5;
-hexdigit(6) -> $6;
-hexdigit(7) -> $7;
-hexdigit(8) -> $8;
-hexdigit(9) -> $9;
-hexdigit(10) -> $a;
-hexdigit(11) -> $b;
-hexdigit(12) -> $c;
-hexdigit(13) -> $d;
-hexdigit(14) -> $e;
-hexdigit(15) -> $f.
generated by cgit v1.2.3 (git 2.25.1) at 2025-09-05 04:54:21 +0000