diff --git a/server/api/flukso/src/flukso.app b/server/api/flukso/src/flukso.app index 0fe0969..166c852 100644 --- a/server/api/flukso/src/flukso.app +++ b/server/api/flukso/src/flukso.app @@ -6,6 +6,7 @@ flukso_app, flukso_sup, flukso_deps, + flukso_sensor, flukso_sensor_xyz ]}, {registered, []}, diff --git a/server/api/flukso/src/flukso.erl b/server/api/flukso/src/flukso.erl index 10ffaf6..7b1391c 100644 --- a/server/api/flukso/src/flukso.erl +++ b/server/api/flukso/src/flukso.erl @@ -16,7 +16,8 @@ ensure_started(App) -> end. mysql_prepare() -> - mysql:prepare(permissions, <<"SELECT permissions FROM logger_tokens WHERE meter = ? AND token = ?">>). + mysql:prepare(permissions, <<"SELECT permissions FROM logger_tokens WHERE meter = ? AND token = ?">>), + mysql:prepare(device_key, <<"SELECT sha FROM logger_devices WHERE device = ?">>). %% @spec start_link() -> {ok,Pid::pid()} %% @doc Starts the app for inclusion in a supervisor tree diff --git a/server/api/flukso/src/flukso_sensor.erl b/server/api/flukso/src/flukso_sensor.erl new file mode 100644 index 0000000..4039781 --- /dev/null +++ b/server/api/flukso/src/flukso_sensor.erl @@ -0,0 +1,105 @@ +%% @author icarus75 +%% @copyright 2009-2010 flukso.net +%% @doc Flukso webmachine_resource. + +-module(flukso_sensor). +-export([init/1, + allowed_methods/2, + malformed_request/2, + is_authorized/2, + process_post/2]). + +-include_lib("webmachine/include/webmachine.hrl"). + +-record(state, + {rrdSensor}). + +init([]) -> + {ok, undefined}. + +% debugging +%init(Config) -> +% {{trace, "/tmp"}, Config}. + +allowed_methods(ReqData, State) -> + {['POST'], ReqData, State}. + +malformed_request(ReqData, State) -> + {_Version, ValidVersion} = check_version(wrq:get_req_header("X-Version", ReqData), wrq:get_qs_value("version", ReqData)), +% TODO: check validity of X-Device and X-Digest headers + + {case {ValidVersion} of + {true} -> false; + _ -> true + end, + ReqData, State}. + +is_authorized(ReqData, State) -> + {data, Result} = mysql:execute(pool, device_key, [wrq:get_req_header("X-Device", ReqData)]), + [[Key]] = mysql:get_result_rows(Result), + Data = wrq:req_body(ReqData), + <> = crypto:sha_mac(Key, Data), + Digest = list_to_binary(io_lib:format("~40.16.0b", [X])), + + {case list_to_binary(wrq:get_req_header("X-Digest", ReqData)) of + Digest -> true; + _WrongDigest -> "access refused" + end, + ReqData, State}. + +% JSON: {"measurements":{"":[[,],...,[,]], +% ..., +% "":[[,],...,[,]]}} +% +% Mochijson2: {struct,[{<<"measurements">>, {struct, [{<<"">>, [[,],...,[,]]}, +% ..., +% {<<"">>, [[,],...,[,]]}]}}]} +% +process_post(ReqData, State) -> + {struct, JsonData} = mochijson2:decode(wrq:req_body(ReqData)), + {struct, Measurements} = proplists:get_value(<<"measurements">>, JsonData), + Ids = proplists:get_keys(Measurements), + RrdResponse = [update(RrdSensor, proplists:get_value(RrdSensor, Measurements)) || RrdSensor <- Ids], + + JsonResponse = mochijson2:encode({struct, [{<<"response">>, {struct, RrdResponse}}]}), + {true , wrq:set_resp_body(JsonResponse, ReqData), State}. + +update(RrdSensor, TimeSeries) -> + Path = "var/data/base/", + RrdData = [[integer_to_list(Time), ":", integer_to_list(Counter), " "] || [Time, Counter] <- TimeSeries], + +%debugging: io:format("~s~n", [[Path, [binary_to_list(RrdSensor)|".rrd"], " ", RrdData]]), + + case erlrrd:update([Path, [binary_to_list(RrdSensor)|".rrd"], " ", RrdData]) of + {ok, _RrdResponse} -> {RrdSensor, <<"ok">>}; + {error, RrdResponse} -> {RrdSensor, list_to_binary(RrdResponse)} + end. + + +%% checks +check_version(undefined, undefined) -> + {false, false}; +check_version(Version, undefined) -> + case Version of + "1.0" -> {Version, true}; + _ -> {false, false} + end; +check_version(undefined, Version) -> + check_version(Version, undefined); +check_version(_, _) -> + {false, false}. + +check_sensor(Sensor) -> + case re:run(Sensor, "[0-9a-f]+", []) of + {match, [{0,32}]} -> {Sensor, true}; + _ -> {false, false} + end. + +check_token(undefined, undefined) -> + {false, false}; +check_token(Token, undefined) -> + check_sensor(Token); +check_token(undefined, Token) -> + check_sensor(Token); +check_token(_, _) -> + {false, false}.