1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
%%--------------------------------------------------------------------
%% %CopyrightBegin%
%%
%% Copyright Ericsson AB 2010-2016. All Rights Reserved.
%%
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%% http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.
%%
%% %CopyrightEnd%
%%----------------------------------------------------------------------
%% File : ct_config_plain.erl
%% Description : CT callback module for reading configs from text files
%%
%% Created : 15 February 2010
%%----------------------------------------------------------------------
-module(ct_config_plain).
-export([read_config/1, check_parameter/1]).
read_config(ConfigFile) ->
case file:consult(ConfigFile) of
{ok,Config} ->
{ok, Config};
{error,enoent} ->
{error,{config_file_error,file:format_error(enoent)}};
{error,Reason} ->
Key =
case application:get_env(common_test, decrypt) of
{ok,KeyOrFile} ->
case KeyOrFile of
{key,K} ->
K;
{file,F} ->
ct_config:get_crypt_key_from_file(F)
end;
_ ->
ct_config:get_crypt_key_from_file()
end,
case Key of
{error,no_crypt_file} ->
{error,{config_file_error,
lists:flatten(
io_lib:format("~ts",[file:format_error(Reason)]))}};
{error,CryptError} ->
{error,{decrypt_file_error,CryptError}};
_ when is_list(Key) ->
case ct_config:decrypt_config_file(ConfigFile,
undefined,
{key,Key}) of
{ok,CfgBin} ->
case read_config_terms(CfgBin) of
{error,ReadFail} ->
{error,{config_file_error,ReadFail}};
Config ->
{ok,Config}
end;
{error,DecryptFail} ->
{error,{decrypt_config_error,DecryptFail}}
end;
_ ->
{error,{bad_decrypt_key,Key}}
end
end.
% check if config file exists
check_parameter(File)->
case filelib:is_file(File) of
true->
{ok,{file,File}};
false->
{error,{nofile,File}}
end.
read_config_terms(Bin) when is_binary(Bin) ->
case catch binary_to_list(Bin) of
{'EXIT',_} ->
{error,invalid_textfile};
Lines ->
read_config_terms(Lines)
end;
read_config_terms(Lines) when is_list(Lines) ->
read_config_terms1(erl_scan:tokens([], Lines, 0), 1, [], []).
read_config_terms1({done,{ok,Ts,EL},Rest}, L, Terms, _) ->
case erl_parse:parse_term(Ts) of
{ok,Term} when Rest == [] ->
lists:reverse([Term|Terms]);
{ok,Term} ->
read_config_terms1(erl_scan:tokens([], Rest, 0),
EL+1, [Term|Terms], Rest);
_ ->
{error,{bad_term,{L,EL}}}
end;
read_config_terms1({done,{eof,_},_}, _, Terms, Rest) when Rest == [] ->
lists:reverse(Terms);
read_config_terms1({done,{eof,EL},_}, L, _, _) ->
{error,{bad_term,{L,EL}}};
read_config_terms1({done,{error,Info,EL},_}, L, _, _) ->
{error,{Info,{L,EL}}};
read_config_terms1({more,_}, L, Terms, Rest) ->
case string:lexemes(Rest, [$\n,[$\r,$\n],$\t]) of
[] ->
lists:reverse(Terms);
_ ->
{error,{bad_term,L}}
end.
|
