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
115
116
117
118
119
120
121
122
123
124
125
126
127
|
// Copyright 2017-2018 int08h LLC
//
// 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.
extern crate hex;
use std::fs::File;
use std::io::Read;
use std::net::SocketAddr;
use std::time::Duration;
use yaml_rust::YamlLoader;
use config::ServerConfig;
use config::{DEFAULT_BATCH_SIZE, DEFAULT_STATUS_INTERVAL};
use Error;
///
/// Read a Roughenough server configuration ([ServerConfig](trait.ServerConfig.html))
/// from a YAML file.
///
/// Example config:
///
/// ```yaml
/// interface: 127.0.0.1
/// port: 8686
/// seed: f61075c988feb9cb700a4a6a3291bfbc9cab11b9c9eca8c802468eb38a43d7d3
/// ```
///
pub struct FileConfig {
port: u16,
interface: String,
seed: Vec<u8>,
batch_size: u8,
status_interval: Duration,
}
impl FileConfig {
pub fn new(config_file: &str) -> Result<Self, Error> {
let mut infile = File::open(config_file).expect("failed to open config file");
let mut contents = String::new();
infile
.read_to_string(&mut contents)
.expect("could not read config file");
let cfg = YamlLoader::load_from_str(&contents).expect("could not parse config file");
if cfg.len() != 1 {
return Err(Error::InvalidConfiguration(
"Empty or malformed config file".to_string(),
));
}
let mut config = FileConfig {
port: 0,
interface: "".to_string(),
seed: Vec::new(),
batch_size: DEFAULT_BATCH_SIZE,
status_interval: DEFAULT_STATUS_INTERVAL,
};
for (key, value) in cfg[0].as_hash().unwrap() {
match key.as_str().unwrap() {
"port" => config.port = value.as_i64().unwrap() as u16,
"interface" => config.interface = value.as_str().unwrap().to_string(),
"batch_size" => config.batch_size = value.as_i64().unwrap() as u8,
"seed" => {
let val = value.as_str().unwrap().to_string();
config.seed = hex::decode(val)
.expect("seed value invalid; 'seed' should be 32 byte hex value");
}
"status_interval" => {
let val = value.as_i64().expect("status_interval value invalid");
config.status_interval = Duration::from_secs(val as u64)
}
unknown => {
return Err(Error::InvalidConfiguration(format!(
"unknown config key: {}",
unknown
)));
}
}
}
Ok(config)
}
}
impl ServerConfig for FileConfig {
fn interface(&self) -> &str {
self.interface.as_ref()
}
fn port(&self) -> u16 {
self.port
}
fn seed(&self) -> &[u8] {
&self.seed
}
fn batch_size(&self) -> u8 {
self.batch_size
}
fn status_interval(&self) -> Duration {
self.status_interval
}
fn socket_addr(&self) -> Result<SocketAddr, Error> {
let addr = format!("{}:{}", self.interface, self.port);
match addr.parse() {
Ok(v) => Ok(v),
Err(_) => Err(Error::InvalidConfiguration(addr)),
}
}
}
|
