sozu 0.9.0

sozu, a fast, reliable, hot reconfigurable HTTP reverse proxy
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

use nix::fcntl::{fcntl,FcntlArg,FdFlag};
use std::os::unix::io::RawFd;
use std::fs::File;
use std::io::Write;
use libc;

use logging;
use sozu_command::config::Config;
use sozu::network::metrics;

pub fn enable_close_on_exec(fd: RawFd) -> Option<i32> {
  fcntl(fd, FcntlArg::F_GETFD).map_err(|e| {
    error!("could not get file descriptor flags: {:?}", e);
  }).ok().and_then(FdFlag::from_bits).and_then(|mut new_flags| {
    new_flags.insert(FdFlag::FD_CLOEXEC);
    fcntl(fd, FcntlArg::F_SETFD(new_flags)).map_err(|e| {
      error!("could not set file descriptor flags: {:?}", e);
    }).ok()
  })
}

// FD_CLOEXEC is set by default on every fd in Rust standard lib,
// so we need to remove the flag on the client, otherwise
// it won't be accessible
pub fn disable_close_on_exec(fd: RawFd) -> Option<i32> {
  fcntl(fd, FcntlArg::F_GETFD).map_err(|e| {
    error!("could not get file descriptor flags: {:?}", e);
  }).ok().and_then(FdFlag::from_bits).and_then(|mut new_flags| {
    new_flags.remove(FdFlag::FD_CLOEXEC);
    fcntl(fd, FcntlArg::F_SETFD(new_flags)).map_err(|e| {
      error!("could not set file descriptor flags: {:?}", e);
    }).ok()
  })
}

pub fn setup_logging(config: &Config) {
  //FIXME: should have an id for the master too
  logging::setup("MASTER".to_string(), &config.log_level,
    &config.log_target, config.log_access_target.as_ref().map(|s| s.as_str()));
}

pub fn setup_metrics(config: &Config) {
  if let Some(ref metrics) = config.metrics.as_ref() {
    metrics::setup(&metrics.address, "MASTER", metrics.tagged_metrics, metrics.prefix.clone());
  }
}

pub fn write_pid_file(config: &Config) -> Result<(), String> {
  let pid_file_path = match config.pid_file_path {
    Some(ref pid_file_path) => Some(pid_file_path.as_ref()),
    None => option_env!("SOZU_PID_FILE_PATH")
  };

  if let Some(ref pid_file_path) = pid_file_path {
    File::create(pid_file_path)
      .and_then(|mut file| {
        let pid = unsafe { libc::getpid() };
        file
          .write_all(format!("{}", pid).as_bytes())
          .map(|()| file)
      })
      .and_then(|file| file.sync_all())
      .map(|()| ())
      .or_else(|err| Err(format!("Couldn't write the PID file to {}. Error: {:?}", pid_file_path, err)))
  } else {
    Ok(())
  }
}