%%% @copyright 2012-2021 Michael Santos <michael.santos@gmail.com>
%%% All rights reserved.
%%%
%%% Redistribution and use in source and binary forms, with or without
%%% modification, are permitted provided that the following conditions
%%% are met:
%%%
%%% 1. Redistributions of source code must retain the above copyright notice,
%%% this list of conditions and the following disclaimer.
%%%
%%% 2. Redistributions in binary form must reproduce the above copyright
%%% notice, this list of conditions and the following disclaimer in the
%%% documentation and/or other materials provided with the distribution.
%%%
%%% 3. Neither the name of the copyright holder nor the names of its
%%% contributors may be used to endorse or promote products derived from
%%% this software without specific prior written permission.
%%%
%%% THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
%%% "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
%%% LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
%%% A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
%%% HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
%%% SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
%%% TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
%%% PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
%%% LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
%%% NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
%%% SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-module(ewpcap).
-include("ewpcap.hrl").
-export([
open/0, open/1, open/2,
close/1,
filter/2, filter/3,
read/1, read/2,
write/2,
getifaddrs/0,
dev/0,
stats/1,
dirty_scheduler_enabled/0
]).
-define(PCAP_NETMASK_UNKNOWN, 4294967295).
-define(DLT_EN10MB, 1).
% (1 bsl 31) - 1.
-define(INT32_MAX, 2147483647).
-type ewpcap_resource() :: #ewpcap_resource{}.
-type ewpcap_stat() :: #ewpcap_stat{}.
-export_type([ewpcap_resource/0, ewpcap_stat/0]).
-on_load({on_load, 0}).
%%--------------------------------------------------------------------
%%% NIF stubs
%%--------------------------------------------------------------------
on_load() ->
case erlang:system_info(smp_support) of
true -> erlang:load_nif(progname(), []);
false -> {error, "Requires smp support (-smp enable)"}
end.
pcap_compile(_, _, _, _) -> erlang:nif_error(not_implemented).
pcap_open_live(_, _, _, _, _, _, _, _) -> erlang:nif_error(not_implemented).
pcap_close(_) -> erlang:nif_error(not_implemented).
pcap_findalldevs() -> erlang:nif_error(not_implemented).
pcap_loop(_) -> erlang:nif_error(not_implemented).
pcap_sendpacket(_, _) -> erlang:nif_error(not_implemented).
pcap_stats(_) -> erlang:nif_error(not_implemented).
% @doc Indicates whether ewpcap is using the dirty scheduler
-spec dirty_scheduler_enabled() -> boolean().
dirty_scheduler_enabled() -> erlang:nif_error(not_implemented).
%%--------------------------------------------------------------------
%%% API
%%--------------------------------------------------------------------
-type time_unit() :: timestamp | microsecond.
-type open_options() :: [
{snaplen, non_neg_integer()}
| {promisc, boolean()}
| {to_ms, non_neg_integer()}
| {filter, iodata()}
| {buffer, non_neg_integer()}
| {monitor, boolean()}
| {time_unit, time_unit()}
| {immediate, boolean()}
| {timeout, immediate | infinity | non_neg_integer()}
].
% @doc Open a network interface using the default device.
%
% open/0 defaults to:
%
% * promiscuous mode disabled
%
% * a snaplen (packet length) of 65535 bytes
%
% * uses immediate mode
%
% * the time unit is an erlang timestamp in the same format
% as now/0
%
% * no filter (all packets are received)
%
% See open/2.
-spec open() -> {ok, ewpcap_resource()} | {error, string() | enomem}.
open() -> open(<<>>, []).
% @doc Open a network interface using the specified device.
%
% open/1 defaults to:
%
% * promiscuous mode disabled
%
% * a snaplen (packet length) of 65535 bytes
%
% * uses immediate mode
%
% * the time unit is an erlang timestamp in the same format
% as now/0
%
% * no filter (all packets are received)
%
% See open/2.
-spec open(iodata()) -> {ok, ewpcap_resource()} | {error, string() | enomem}.
open(Dev) -> open(Dev, []).
% @doc Open a network interface and begin receiving packets.
%
% The returned Socket in the 'ok' tuple must be kept by the
% process. When the socket goes out of scope, the pcap filter will
% be shut down and all resources associated with the socket will
% be freed. See also close/1.
%
% Dev is the name of the network device. If an empty binary (`<<>>')
% is passed in, pcap will select a default interface.
%
% If an error occurs, the PCAP string describing the error is
% returned to the caller.
%
% The `timeout` option sets `pcap_set_timeout(3PCAP)` and
% `pcap_set_immediate_mode(3PCAP)`. By default, ewpcap uses
% `immediate` mode and returns packets as they are received.
% Setting timeout to an integer value disables `immediate` mode
% and buffers any packets until either the timeout is reached or
% the buffer is filled.
%
% If ewpcap is dropping packets (see stats/1), the PCAP buffer
% size can be increased (should be some multiple of the snaplen).
%
% Wireless devices can be set to use monitor mode (rfmon) by
% passing in the 'monitor' option.
%
% The timestamp in the message can be formatted either as a now/0
% tuple or returned in microseconds.
%
% For filter options, see filter/3.
%
% Packets are returned as messages to the caller:
%
% {ewpcap, Ref, DatalinkType, Time, Length, Packet}
%
% Ref is a reference identifying the socket handle.
%
% The DataLinkType is an integer representing the link layer,
% e.g., ethernet, Linux cooked socket.
%
% The Time is a tuple in the same format as erlang:now/0, {MegaSecs,
% Secs, MicroSecs} or microseconds.
%
% The Length corresponds to the actual packet length on the
% wire. The captured packet may have been truncated. To get the
% captured packet length, use byte_size(Packet).
%
% The Packet is a binary holding the captured data.
%
% Errors will be sent to the caller and the pcap filter will
% be terminated:
%
% {ewpcap_error, Ref, Error}
-spec open(iodata(), open_options()) ->
{ok, ewpcap_resource()}
| {error, string() | enomem}.
open(<<>>, Options) ->
case pcap_lookupdev() of
{ok, Dev} -> open(Dev, Options);
Error -> Error
end;
open(Dev, Options) when is_list(Options) ->
To_ms0 = proplists:get_value(to_ms, Options, immediate),
Timeout = proplists:get_value(timeout, Options, To_ms0),
{Immediate0, To_ms} =
case Timeout of
% dummy value: timeout value is ignored
immediate -> {true, 500};
infinity -> {false, ?INT32_MAX};
_ when is_integer(Timeout) -> {false, Timeout}
end,
Snaplen = proplists:get_value(snaplen, Options, 65535),
Promisc = bool(proplists:get_value(promisc, Options, false)),
Filter = proplists:get_value(filter, Options, <<>>),
Buffer = proplists:get_value(buffer, Options, 0),
Monitor = bool(proplists:get_value(monitor, Options, false)),
TimeUnit = time_unit(proplists:get_value(time_unit, Options, timestamp)),
Immediate = bool(proplists:get_value(immediate, Options, Immediate0)),
case pcap_open_live(Dev, Snaplen, Promisc, To_ms, Buffer, Monitor, TimeUnit, Immediate) of
{ok, Socket} -> open_1(Socket, Options, Filter);
Error -> Error
end.
open_1(Socket, _Options, <<>>) ->
open_2(Socket);
open_1(Socket, Options, Filter) ->
case filter(Socket, Filter, Options) of
ok -> open_2(Socket);
Error -> Error
end.
open_2(Socket) ->
case loop(Socket) of
ok -> {ok, Socket};
Error -> Error
end.
% @doc Closes the pcap descriptor.
-spec close(ewpcap_resource()) -> ok.
close(#ewpcap_resource{res = Res}) -> pcap_close(Res).
-type filter_options() :: [
{optimize, boolean()}
| {netmask, non_neg_integer()}
].
% @doc Compile a PCAP filter and apply it to the PCAP descriptor.
%
% Since the library passes the filter string to pcap_compile(3PCAP)
% directly, any bugs in pcap_compile() may cause the Erlang VM
% to crash. Do not use filters from untrusted sources.
%
% Filters are limited to 8192 bytes by default since it may be
% possible for very large filters to cause a stack overflow. For
% example:
%
% ```
% ewpcap:open(<<>>, [{filter, string:copies("ip and ", 50000) ++ "ip"}, {limit, -1}])
% '''
-spec filter(ewpcap_resource(), iodata()) -> ok | {error, string() | enomem}.
filter(Res, Filter) -> filter(Res, Filter, []).
-spec filter(ewpcap_resource(), iodata(), filter_options()) ->
ok
| {error, string() | enomem}.
filter(#ewpcap_resource{res = Res}, Filter, Options) when is_binary(Filter); is_list(Filter) ->
Optimize = bool(proplists:get_value(optimize, Options, true)),
Netmask = mask(proplists:get_value(netmask, Options, ?PCAP_NETMASK_UNKNOWN)),
Limit = proplists:get_value(limit, Options, 8192),
case iolist_size(Filter) < Limit orelse Limit < 0 of
true -> pcap_compile(Res, Filter, Optimize, Netmask);
false -> {error, enomem}
end.
-spec loop(ewpcap_resource()) -> ok | {error, files:posix()}.
loop(#ewpcap_resource{res = Res}) -> pcap_loop(Res).
% @doc Convenience function wrapping receive, returning the packet
% contents.
-spec read(ewpcap_resource()) -> {ok, binary()} | {error, string()}.
read(Res) -> read(Res, infinity).
-spec read(ewpcap_resource(), infinity | non_neg_integer()) ->
{ok, binary()}
| {error, string() | eagain}.
read(#ewpcap_resource{ref = Ref}, Timeout) ->
receive
{ewpcap, Ref, _DatalinkType, _Time, _ActualLength, Packet} -> {ok, Packet};
{ewpcap_error, Ref, Error} -> {error, Error}
after Timeout -> {error, eagain}
end.
% @doc Write the packet to the network. See pcap_sendpacket(3PCAP).
-spec write(ewpcap_resource(), iodata()) -> ok | {error, string()}.
write(#ewpcap_resource{res = Res}, Data) when is_list(Data); is_binary(Data) ->
pcap_sendpacket(Res, Data).
% @doc Returns the default device used by PCAP.
-spec dev() -> {ok, string()} | {error, string()}.
dev() -> pcap_lookupdev().
% @doc Returns a list of interfaces. Ifname can be used as the first
% parameter to open/1 and open/2.
%
% This function is modelled on inet:getifaddrs/0 but uses
% pcap_findalldevs(3PCAP) to look up the interface attributes:
%
% * getifaddrs/0 may return pseudo devices, such as the "any"
% device on Linux
%
% * getifaddrs/0 will only return the list of devices that
% can be used with open/1 and open/2. An empty list ({ok,
% []}) may be returned if the user does not have permission
% to open any of the system interfaces
-spec getifaddrs() ->
{ok, [] | [{string(), [proplists:proplist()]}]}
| {error, string()}.
getifaddrs() ->
case pcap_findalldevs() of
{ok, Iflist} -> {ok, [iface(N) || N <- lists:reverse(Iflist)]};
Error -> Error
end.
% @doc To use the return value as a record, include the header:
%
% -include_lib("ewpcap/include/ewpcap.hrl").
%
% stats/1 returns statistics about dropped packets. See
% pcap_stats(3PCAP) for details.
%
% The ewpcap_stat records contains these fields:
%
% recv : number of packets received
%
% drop : number of packets dropped due to insufficient buffer
%
% ifdrop : number of packets dropped by the network interface
%
% capt : always 0 (was number of packets received by the application (Win32 only))
-spec stats(ewpcap_resource()) -> {ok, ewpcap_stat()} | {error, string()}.
stats(#ewpcap_resource{res = Res}) -> pcap_stats(Res).
iface({If, Attr}) -> {If, addr(Attr)}.
addr(Attr) -> addr(Attr, []).
addr([], Attr) ->
Attr;
addr([{Key, <<A, B, C, D>>} | T], Attr) ->
addr(T, [{Key, {A, B, C, D}} | Attr]);
addr([{Key, <<A:16, B:16, C:16, D:16, E:16, F:16, G:16, H:16>>} | T], Attr) ->
addr(T, [{Key, {A, B, C, D, E, F, G, H}} | Attr]);
addr([N | T], Attr) ->
addr(T, [N | Attr]).
%%--------------------------------------------------------------------
%%% Internal functions
%%--------------------------------------------------------------------
pcap_lookupdev() ->
case getifaddrs() of
{ok, [{If, _} | _]} ->
{ok, If};
{error, _} = Error ->
Error
end.
bool(true) -> 1;
bool(false) -> 0.
time_unit(timestamp) -> 0;
time_unit(microsecond) -> 1.
mask(N) when is_integer(N) -> N;
mask({A, B, C, D}) -> A bsl 24 bor (B bsl 16) bor (C bsl 8) bor D.
progname() ->
case code:priv_dir(?MODULE) of
{error, bad_name} ->
filename:join([filename:dirname(code:which(?MODULE)), "..", "priv", ?MODULE]);
Dir ->
filename:join([Dir, ?MODULE])
end.