feat: Starting work on daemon process

Author: sousandrei

.github/workflows/main.yaml

@@ -1,4 +1,4 @@
-name: Branches
+name: Main
 
 on:
   push:

.gitignore

@@ -3,7 +3,7 @@ target
 Cargo.lock
 assets
 
-# firecracker
+firecracker
 tmp
 logs
 todo

Cargo.toml

@@ -2,13 +2,30 @@
 authors = ["Andrei Sousa <sousandrei@gmail.com>"]
 edition = "2018"
 name = "firesquid"
+readme = "README.md"
 version = "0.1.0"
 
+
+[package.metadata.deb]
+assets = [
+    ["target/release/firesquid", "usr/bin/firesquid", "755"],
+    [
+        "debian/firesquid.service",
+        "/lib/systemd/system/firesquid.service",
+        "644",
+    ],
+]
+depends = "$auto, systemd"
+extended-description = "firesquid daemon"
+maintainer-scripts = "debian/scripts"
+priority = "optional"
+section = "admin"
+
 [dependencies]
 chrono = "0.4.19"
 clap = { version = "2.33.3", features = ["yaml"] }
 hyper = { version = "0.14.4", features = ["client", "server", "stream", "http1", "tcp"] }
-hyperlocal = { version = "0.8" }
+hyperlocal = "0.8"
 serde = { version = "1.0.123", features = ["derive"] }
 serde_json = "1.0.63"
 tokio = { version = "1.2.0", features = [

README.md

@@ -13,7 +13,7 @@
 
 <br />
 
-[![Build Status](https://github.com/sousandrei/firesquid/workflows/Branches/badge.svg)](https://github.com/sousandrei/firesquid/actions)
+[![Build Status](https://github.com/sousandrei/firesquid/workflows/Main/badge.svg)](https://github.com/sousandrei/firesquid/actions)
 
 ## Table of Contents
 
@@ -36,6 +36,12 @@ It abstracts the hard part making it a breeze to spawn your very own fleet of mi
 - Lightweight
 - Customizable
 
+Upcoming:
+
+- Networking
+- Proper package release and distribution
+- Choices between which kernel to use for which machine
+
 ## <a name="help-wanted"></a> Assets 📦
 
 Here are some assets to get you started. The default folder for assets is just called `assets` in the same folder as FireSquid.

debian/firesquid.service

@@ -0,0 +1,13 @@
+[Unit]
+AssertPathExists=/usr/bin/firesquid
+
+[Service]
+WorkingDirectory=~
+ExecStart=/usr/bin/firesquid
+Restart=always
+PrivateTmp=true
+NoNewPrivileges=true
+
+[Install]
+Alias=firesquid
+WantedBy=default.target

debian/scripts/postinst

@@ -0,0 +1,17 @@
+#!/bin/sh
+
+## This will only remove masks created by d-s-h on package removal.
+deb-systemd-helper unmask firesquid.service > /dev/null || true
+
+# was-enabled defaults to true, so new installations run enable.
+if deb-systemd-helper --quiet was-enabled firesquid.service
+then
+	# Enables the unit on first installation, creates new
+	# symlinks on upgrades if the unit file has changed.
+	deb-systemd-helper enable firesquid.service > /dev/null || true
+	deb-systemd-invoke start firesquid
+else
+	# Update the statefile to add new symlinks (if any), which need to be
+	# cleaned up on purge. Also remove old symlinks.
+	deb-systemd-helper update-state firesquid.service > /dev/null || true
+fi

debian/scripts/postrm

@@ -0,0 +1,13 @@
+#!/bin/sh
+
+# In case this system is running systemd, we make systemd reload the unit files
+# to pick up changes.
+if [ -d /run/systemd/system ] ; then
+	systemctl --system daemon-reload >/dev/null || true
+fi
+
+if [ "$1" = "remove" ]; then
+	if [ -x "/usr/bin/deb-systemd-helper" ]; then
+		deb-systemd-helper mask firesquid.service >/dev/null
+	fi
+fi

debian/scripts/prerm

@@ -0,0 +1,3 @@
+#!/bin/sh
+
+deb-systemd-invoke stop firesquid

src/main.rs

@@ -18,6 +18,9 @@ mod vm;
 use crate::cli::generate_cli;
 use crate::state::{State, StatePtr};
 
+use std::fs;
+use std::path;
+
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
     if env::var("RUST_LOG").is_err() {
@@ -26,6 +29,22 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
 
     tracing_subscriber::fmt::init();
 
+    //TODO: better way of finding out if it's being run by system
+    if !path::Path::new("/tmp/firesquid/lock.pid").exists() {
+        start_daemon().await?;
+        return Ok(());
+    }
+
+    println!("CLI GOES HERE");
+
+    Ok(())
+}
+
+async fn start_daemon() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
+    info!("Creating lock file");
+    fs::create_dir("/tmp/firesquid").ok();
+    fs::File::create("/tmp/firesquid/lock.pid").ok();
+
     let cli_options = match generate_cli() {
         Ok(options) => options,
         Err(e) => return Err(e),
@@ -75,6 +94,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
 
     terminate_all_vms(state_ptr).await;
 
+    info!("Removing lock file");
+    fs::remove_file("/tmp/firesquid/lock.pid").ok();
+
     Ok(())
 }