amnesie 0.0.4 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 81455c8939114e8211b6c21844ad35e88d0422c29be33d36cafff2ea5210bfa6
-  data.tar.gz: 6dcc45de41fbfbf84b3e6f4a0af45fe139af8296ce5a444ecc3e4d5997baf243
+  metadata.gz: fee6ddb83b07b28a9196dfddbef713a7f3a6f9a5d32129e7f0952e65c4ed9857
+  data.tar.gz: 8e31162f07c03bb4fda8f9774d940adee9df0f9764002176a6c67f0b6687ab88
 SHA512:
-  metadata.gz: bc53a51bed717b6e931d35d0a1e19ba45c8b6ee250a811ed08e5956e6f64c419167112e1329f32019b0fa0204e0b98fc8ebb9d16f368923b959d45504769591c
-  data.tar.gz: 4a78f282cc125ce8dd2e7248ebfa9b3126dcff206a257a04157f7d01582728944fcdf0da62638dc16e079e700055b657bad0c043eef3d175e887d610552db288
+  metadata.gz: 5db556f79185ecd0098584d9ad2862d97f659461918beb39f625da1b4d804b5af5fe5c7cbd2d5245fe96280deaa5f5abb0dc2c362f73b59c0385c8d66ab4cb2f
+  data.tar.gz: 928261af07f2fc37e8f6801dc7b3480ee76854231b08e3ac45d8668fef917d145997788d8ccd87929b81c9eea409a7fdbffe47ec37d7c832f22039dab01d29bb

data/.gitignore

@@ -1,5 +1,6 @@
 *.gem
 *.rbc
+*.swp
 /.config
 /coverage/
 /InstalledFiles

data/CHANGELOG.md

@@ -1,3 +1,30 @@
+## 0.0.9, release 2020-11-06
+* Support for iwd
+* Support for wpa_supplicant
+* Config file (per user in ~/.config/amnesie/ or system wide /etc/amnesie/)
+* Can generate a random hostname between 8-25 characters
+
+## 0.0.8, release 2020-09-22
+* Enhance code
+* Update the class MAC, more ruby like
+* Use securerandom (>= ruby2.5)
+
+## 0.0.7, release 2020-05-29
+* Shortcut action for service (e/d) 
+* Correct permission on amnesie-mac@.service
+* Enhance lib/amnesie/helper
+
+## 0.0.6, release 2020-05-20
+* Init a doc with RDoc.
+* Add -i|--init to start init process properly.
+* Add dhcpcd to the mac@.service, dhclient doesn't seem necessary.
+* Correct the mac@.service.
+
+## 0.0.5, release 2020-05-19
+* Create a amnesie-mac@.service (work only if the gem is install system-wide)
+* Add -p|--persist
+* Replace Makefile by a Rakefile
+
 ## 0.0.4, release 2020-05-16
 * Debian10 need to reload ifup@card-name too
 * Correct path where search dhclient (/bin,/sbin)

data/README.md

@@ -2,15 +2,13 @@
 A tool to make your computer amnesic.
 
 Inspiration come from a post found on [Qubes-OS](https://www.qubes-os.org/doc/anonymizing-your-mac-address/) and the [Whonix](https://www.whonix.org/) project. 
-+ Anonymizing your MAC Address
-+ Randomize all Ethernet and Wifi connections
++ Anonymizing MAC Address (wifi, ethernet)
 + Randomize your hostname
-+ [Boot Clock Randomization](https://www.whonix.org/wiki/Boot_Clock_Randomization)
 
 ## Install
 Amnesie is cryptographically signed, so add my public key (if you haven’t already) as a trusted certificate.
 
-    $ gem cert --add <(curl -Ls https://raw.githubusercontent.com/szorfein/spior/master/certs/szorfein.pem)
+    $ gem cert --add <(curl -Ls https://raw.githubusercontent.com/szorfein/amnesie/master/certs/szorfein.pem)
 
 And install the gem
 
@@ -24,14 +22,22 @@ To be able to use the `persist mode` (with systemd for now), the gem should be i
 If you can, i recommend that you create a package for your distribution.  
 
 ## Usage
-To change the MAC address for eth0:
+To change the MAC address only on `eth0`:
 
     $ amnesie -n eth0 -m
 
+Create or Disable all systemd services for a network card:
+
+    $ amnesie -p
+
+Forge a random hostname with `-H`
+
+    $ amnesie -H
+
 ## Left Over
 
 ### Issues
-For any questions, comments, feedback or issues, submit a [new issue](https://github.com/szorfein/spior/issues/new).
+For any questions, comments, feedback or issues, submit a [new issue](https://github.com/szorfein/amnesie/issues/new).
 
 ### links
 + https://rubyreferences.github.io/rubyref

data/Rakefile

@@ -1,10 +1,21 @@
 # https://github.com/seattlerb/minitest#running-your-tests-
 require "rake/testtask"
+require File.dirname(__FILE__) + "/lib/amnesie/version"
 
 Rake::TestTask.new(:test) do |t|
-    t.libs << "test"
-    t.libs << "lib"
-    t.test_files = FileList["test/test_*.rb"]
+  t.libs << "test"
+  t.libs << "lib"
+  t.test_files = FileList["test/test_*.rb"]
+end
+
+# rake gem:build
+namespace :gem do
+  desc "build the gem"
+  task :build do
+    Dir["amnesie*.gem"].each {|f| File.unlink(f) }
+    system("gem build amnesie.gemspec")
+    system("gem install amnesie-#{Amnesie::VERSION}.gem -P MediumSecurity")
+  end
 end
 
 task :default => :test

data/amnesie.gemspec

@@ -1,6 +1,8 @@
+require File.dirname(__FILE__) + "/lib/amnesie/version"
+
 Gem::Specification.new do |s|
   s.name = "amnesie"
-  s.version = "0.0.4"
+  s.version = Amnesie::VERSION
   s.summary = "A tool to make your computer amnesic"
   s.description = <<-EOF 
     A tool to make your computer amnesic"
@@ -34,4 +36,5 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency('interfacez', '1.0.3')
   s.add_runtime_dependency('nomansland', '0.0.2')
   s.add_runtime_dependency('tty-which', '0.4.2')
+  s.add_runtime_dependency('highline', '2.0.3')
 end

data/bin/amnesie

@@ -1,6 +1,5 @@
 #!/usr/bin/env ruby
 
-require 'amnesie/runner'
+require 'amnesie'
 
-runner = Amnesie::Runner.new(ARGV)
-runner.run
+Amnesie::Main.new(ARGV)

data/lib/amnesie.rb

@@ -0,0 +1,99 @@
+require_relative 'amnesie/options'
+require_relative 'amnesie/persist'
+require_relative 'amnesie/process'
+require_relative 'amnesie/network'
+require_relative 'amnesie/mac'
+require_relative 'amnesie/host'
+require_relative 'amnesie/config'
+require_relative 'amnesie/helpers'
+
+module Amnesie
+
+  OPTIONS = {
+    mac: true,
+    hostname: false,
+    card_match: /^en/
+  }.freeze
+
+  def self.random_mac(network)
+    mac = Amnesie::MAC.new(network)
+    mac.set_addr
+    puts "New MAC for " + mac.to_s
+  end
+
+  # For wifi card, no need systemd
+  def self.persist_wifi
+    if TTY::Which.exist?('iwctl')
+      Amnesie::Persist::Iwd.new
+    elsif TTY::Which.exist?('wpa_supplicant')
+      Amnesie::Persist::WpaSupplicant.new
+    end
+  end
+
+  def self.services(network)
+    # For ethernet card
+    if TTY::Which.exist?('systemctl') && network.match(/^en/)
+      persist = Amnesie::Persist::Systemd.new(network)
+      if ! persist.mac_exist?
+        puts "Create service..."
+        persist.services
+      elsif persist.mac_exist?
+        puts "service exist"
+        persist.update_mac
+      end
+      persist.menu_mac
+    end
+  end
+
+  def self.random_mac_and_kill(network)
+    process = Amnesie::Process.new(network)
+    mac = Amnesie::MAC.new(network)
+
+    process.kill
+    mac.down
+    mac.set_addr
+    puts "New MAC for " + mac.to_s
+    mac.up
+    process.restart
+
+  end
+
+  class Main
+    def initialize(argv)
+      @argv = argv
+      run
+    end
+
+    private
+
+    def run
+      options = Options.new(@argv)
+      networks = Network.new(options.card_match, options.net_dev).search
+      puts "cards #{networks}"
+
+      if options.init
+        networks.each { |net|
+          Amnesie.random_mac(net)
+        }
+      end
+
+      if options.persist
+        networks.each { |net|
+          Amnesie.services(net)
+        }
+        Amnesie.persist_wifi
+        exit
+      end
+
+      if options.mac
+        networks.each { |net|
+          Amnesie.random_mac_and_kill(net)
+        }
+      end
+
+      if options.hostname
+        Amnesie::Host.new
+      end
+    end
+  end
+end

data/lib/amnesie/config.rb

@@ -0,0 +1,34 @@
+require 'yaml'
+
+module Amnesie
+  class Config
+    def initialize
+      @file = File.join(find_conf)
+      #puts "Config file in #{@file}"
+    end
+
+    def load
+      if !File.exist? @file
+        puts "[+] Config file created at #{@file}"
+        save
+      end
+      YAML.load_file @file
+    end
+
+    private
+
+    def find_conf
+      if !ENV["HOME"] || ENV["HOME"] == '/root'
+        "/etc/conf.d/amnesie.yaml"
+      else
+        "#{ENV['HOME']}/.config/amnesie/amnesie.yaml"
+      end
+    end
+
+    def save
+      dir = File.dirname @file
+      Nito::Mkdir.new(dir)
+      File.open(@file, 'w') { |f| YAML::dump(OPTIONS, f) }
+    end
+  end
+end

data/lib/amnesie/helpers.rb

@@ -1,13 +1,15 @@
+require 'fileutils'
+require 'tempfile'
+
 module Helpers
   class Exec
     def initialize(name)
-      @search_uid=`id -u`.chomp
-      @search_uid ||= 1000 unless $?.success?
+      @search_uid = Process::Sys.getuid
       @name = name
     end
 
     def run(args)
-      if @search_uid == '0' then
+      if @search_uid == 0 then
         #puts "found root - uid #{@search_uid}"
         system(@name + " " + args)
       else
@@ -16,4 +18,53 @@ module Helpers
       end
     end
   end
+
+  class NewFile
+    def initialize(string, name, dest = "/tmp")
+      @string = string
+      @name = name
+      @dest = dest + "/" + @name
+    end
+    
+    def add
+      @mv = Helpers::Exec.new("mv")
+      tmp = Tempfile.new(@name)
+      File.open(tmp.path, 'w') do |file|
+        file.puts @string
+      end
+      @mv.run("#{tmp.path} #{@dest}")
+    end
+
+    def perm(user, perm)
+      chown = Helpers::Exec.new("chown")
+      chmod = Helpers::Exec.new("chmod")
+      chown.run("#{user}:#{user} #{@dest}")
+      chmod.run("#{perm} #{@dest}")
+    end
+  end
+
+  class NewSystemd < NewFile
+    def initialize(string, name)
+      super
+      @systemd_dir = search_systemd_dir
+      @dest = @systemd_dir + "/" + @name
+    end
+
+    def add
+      @systemctl = Helpers::Exec.new("systemctl")
+      super
+      @systemctl.run("daemon-reload")
+    end
+
+    private
+    def search_systemd_dir
+      if Dir.exist? "/lib/systemd/system"
+        "/lib/systemd/system"
+      elsif Dir.exist? "/usr/lib/systemd/system"
+        "/usr/lib/systemd/system"
+      else
+        raise "Systemd is no found..."
+      end
+    end
+  end
 end

data/lib/amnesie/host.rb

@@ -0,0 +1,16 @@
+require 'securerandom'
+
+module Amnesie
+  class Host
+    def initialize
+      @nb = rand(8..25)
+      @hostname = SecureRandom.alphanumeric(@ng)
+      Nito::Hostname.new(@hostname)
+      puts to_s
+    end
+
+    def to_s
+      "Your hostname will become #{@hostname}"
+    end
+  end
+end

data/lib/amnesie/mac.rb

@@ -1,47 +1,50 @@
-require_relative 'helpers'
+require 'securerandom'
 
 module Amnesie
   class MAC
     def initialize(card)
       @card = card
       @ip = Helpers::Exec.new("ip")
+      save_origin
+      new_mac
     end
 
-    # TODO yep...
-    def rand
-      first=`printf ""; dd bs=1 count=1 if=/dev/urandom 2>/dev/null | hexdump -v -e '/1 "%02X"'`
-      last=`printf ""; dd bs=1 count=5 if=/dev/urandom 2>/dev/null | hexdump -v -e '/1 ":%02X"'`
-      mac="#{first}#{last}"
-      lastfive=`echo #{mac} | cut -d: -f 2-6`.chomp
-      firstbyte=`echo #{mac} | cut -d: -f 1`.chomp
-      firstbyte=`printf '%02X' $(( 0x#{firstbyte} & 254 | 2))`
-      @mac="#{firstbyte}:#{lastfive}"
+    def set_addr
+      @ip.run("link set dev #{@card} address #{@mac}")
+    end
+
+    def down
+      @ip.run("link set dev #{@card} down")
+    end
+
+    def up
+      @ip.run("link set dev #{@card} up")
     end
 
     def to_s
-      rand
       @card + " " + @mac
     end
 
-    def search_curr_mac
-      @curr = `ip addr show dev #{@card} | grep -i ether | awk '{print $2}'`
+    private
+
+    def new_mac
+      first = SecureRandom.hex(1)
+      last = SecureRandom.hex(5)
+      lastfive = last.split(//).each_slice(2).to_a.map(&:join).join(':')
+      firstbyte = `printf '%02X' $(( 0x#{first} & 254 | 2))`
+      @mac = "#{firstbyte}:#{lastfive}".downcase
     end
 
+    def search_curr_mac
+      @curr = `ip addr show dev #{@card} | grep -i ether | awk '{print $2}'`.chomp
+    end
+    
     def save_origin
+      return if File.exist? "/tmp/mac_#{@card}"
       search_curr_mac
-      txt="/tmp/mac_#{@card}"
-      if not File.exist?(txt)
-        file = File.open(txt, 'a+')
-        file.puts(@curr)
-        file.close
-        puts "Origin saved"
-      end
-    end
-
-    def apply
-      @ip.run("link set dev #{@card} down")
-      @ip.run("link set dev #{@card} address #{@mac}")
-      @ip.run("link set dev #{@card} up")
+      filename="/tmp/mac_#{@card}"
+      File.write(filename, @curr)
+      puts "Origin saved"
     end
   end
 end

data/lib/amnesie/network.rb

@@ -2,43 +2,39 @@ require 'interfacez'
 
 module Amnesie
   class Network
-    attr_accessor :card
-
-    def initialize(name = false)
+    def initialize(card_match, name = nil)
+      @card_match = card_match
       @name = name
+      @devs = []
       @check = false
     end
 
-    def card
-      verify_card
-      if @check == false then
-        ask_for_card
+    def search
+      if @name
+        verify_card
+        @devs << @name
+      else
+        search_cards
+        @devs
       end
-      @name
     end
 
     private
 
     def verify_card
-      return if @check or not @name
       Interfacez.all do |interface|
         if interface == @name then
           @check = true
         end
       end
-      if not @check then
-        puts "Your interface #{@name} is no found"
+      if !@check then
+        raise ArgumentError, "Interface no found" if !@check
       end
     end
 
-    def ask_for_card
-      until @check == true
-        Interfacez.all do |interface|
-          print interface + " "
-        end
-        printf "\nWhat is the name of the card to be used? "
-        @name = gets.chomp
-        verify_card
+    def search_cards
+      Interfacez.all do |interface|
+        @devs << interface if interface.match(@card_match)
       end
     end
   end

data/lib/amnesie/options.rb

@@ -2,21 +2,39 @@ require 'optparse'
 
 module Amnesie
   class Options
-    attr_reader :mac, :netcard
+    attr_reader :init, :mac, :net_dev, :persist, :hostname, :card_match
 
     def initialize(argv)
+      @default = Config.new.load
+      @mac = @default[:mac]
+      @hostname = @default[:hostname]
+      @card_match = @default[:card_match]
       parse(argv)
     end
 
+    private
+
     def parse(argv)
       OptionParser.new do |opts|
 
-        opts.on("-m", "--mac", "What is my MAC address?") do
+        opts.on("-i", "--init", "When used with a init process (systemd, etc...)") do
+          @init = true
+        end
+
+        opts.on("-m", "--mac", "Forge a random MAC address.") do
           @mac = true
         end
 
-        opts.on("-n", "--net-card NAME", "The name of the card to use") do |net|
-          @netcard = net
+        opts.on("-n", "--net-card NAME", "Card to use, default use card_match from the config file.") do |net|
+          @net_dev = net
+        end
+
+        opts.on("-p", "--persist", "Enable systemd service") do |net|
+          @persist = true
+        end
+
+        opts.on("-H", "--hostname", "Generate a new random hostname") do |host|
+          @hostname = true
         end
 
         opts.on("-h", "--help", "Show this message") do
@@ -25,7 +43,6 @@ module Amnesie
         end
 
         begin
-          argv = ["-h"] if argv.empty?
           opts.parse!(argv)
         rescue OptionParser::ParseError => e
           STDERR.puts e.message, "\n", opts

data/lib/amnesie/persist.rb

@@ -0,0 +1,18 @@
+module Amnesie
+  module Persist
+    def self.grep?(file, regex)
+      is_found = false
+      return is_found if ! File.exist? file
+      File.open(file) do |f|
+        f.each do |line|
+          is_found = true if line.match(regex)
+        end
+      end
+      is_found
+    end
+  end
+end
+
+require_relative 'persist/iwd'
+require_relative 'persist/wpa_supplicant'
+require_relative 'persist/systemd'

data/lib/amnesie/persist/iwd.rb

@@ -0,0 +1,34 @@
+require 'nito'
+require 'tempfile'
+
+module Amnesie
+  module Persist
+    class Iwd
+      def initialize
+        @tmp = Tempfile.new("main.conf")
+        apply
+      end
+
+      def apply
+        File.write(@tmp, iwd_conf)
+        if ! File.exist? "/etc/iwd/main.conf" ||
+            ! grep?("/etc/iwd/main.conf", /AddressRandomization/)
+          puts "Add iwd/main.conf"
+          Nito::Cp.new(@tmp.path, "/etc/iwd/main.conf")
+        else
+          puts "MAC random on iwd seem enable."
+        end
+      end
+
+      private
+
+      def iwd_conf
+        <<EOF
+[General]
+AddressRandomization=network
+AddressRandomizationRange=full
+EOF
+      end
+    end
+  end
+end

data/lib/amnesie/persist/systemd.rb

@@ -0,0 +1,88 @@
+module Amnesie
+  module Persist
+    class Systemd
+      def initialize(card = nil)
+        @card = card
+        @systemd_dir = search_systemd_dir
+        @systemctl = Helpers::Exec.new("systemctl")
+      end
+
+      def mac_exist?
+        File.exist? "#{@systemd_dir}/amnesie-mac@.service"
+      end
+
+      def to_s
+        @systemd_dir
+      end
+
+      def services
+        mac_service
+        new_service = Helpers::NewSystemd.new(@string, "amnesie-mac@.service")
+        new_service.add
+        new_service.perm("root", "644")
+      end
+
+      def update_mac
+        print "Found a old amnesie-mac@.service, update? (y|n) "
+        answer = gets.chomp
+        case answer
+        when /^y|^Y/
+          services
+        end
+      end
+
+      def menu_mac
+        print "Action on amnesie-mac@.service for #{@card} (enable/disable) ? (e/d) "
+        answer = gets.chomp
+        case answer
+        when /^e|^E/
+          mac_enable
+        when /^d|^D/
+          mac_disable
+        end
+      end
+
+      private
+
+      def mac_enable
+        @systemctl.run("enable amnesie-mac@#{@card}.service")
+      end
+
+      def mac_disable
+        @systemctl.run("disable amnesie-mac@#{@card}.service")
+      end
+
+      def search_systemd_dir
+        if Dir.exist? "/lib/systemd/system"
+          "/lib/systemd/system"
+        elsif Dir.exist? "/usr/lib/systemd/system"
+          "/usr/lib/systemd/system"
+        else
+          raise "No directory systemd found"
+          exit
+        end
+      end
+
+      def mac_service
+        dhcp = TTY::Which.exist?("dhcpcd") ? 'dhcpcd.service' : ''
+        @string=<<EOF
+[Unit]
+Description=Spoof MAC Address on %I
+Wants=network-pre.target
+Before=network-pre.target #{dhcp}
+BindsTo=sys-subsystem-net-devices-%i.device
+After=sys-subsystem-net-devices-%i.device
+
+[Service]
+Type=oneshot
+ExecStart=/usr/bin/env bash -lc "amnesie -i -n %I"
+ExecReload=/usr/bin/env bash -lc "amnesie -m -n %I"
+TimeoutSec=30
+
+[Install]
+WantedBy=multi-user.target
+EOF
+      end
+    end
+  end
+end

data/lib/amnesie/persist/wpa_supplicant.rb

@@ -0,0 +1,41 @@
+require 'nito'
+require 'tempfile'
+
+module Amnesie
+  module Persist
+    class WpaSupplicant
+      def initialize
+        @cards = Network.new(/wl^/, nil).search
+        @tmp = Tempfile.new("main.conf")
+        apply_cards
+      end
+
+      def apply_cards
+        @cards.each { |card|
+          apply(card)
+        }
+      end
+
+      private
+
+      def apply(card)
+        file = "/etc/wpa_supplicant/wpa_supplicant-#{card}.conf"
+        if ! File.exist? file ||
+            ! grep?(file, /gas_rand_mac/)
+          puts "Add #{file}"
+          Nito::Cat.new(file, wpa_conf)
+        else
+          puts "MAC random on wpa_supplicant seem enable."
+        end
+      end
+
+      def wpa_conf
+        <<EOF
+mac_addr=1
+preassoc_mac_addr=1
+gas_rand_mac_addr=1
+EOF
+      end
+    end
+  end
+end

data/lib/amnesie/process.rb

@@ -1,11 +1,10 @@
 require 'tty-which'
-require_relative 'helpers'
 
 module Amnesie
   class Process
     def initialize(card)
       @systemctl = Helpers::Exec.new("systemctl")
-      @pkill = Helpers::Exec.new("pkill")
+      @kill = Helpers::Exec.new("kill")
       @rm = Helpers::Exec.new("rm")
       @card = card
     end
@@ -25,15 +24,16 @@ module Amnesie
 
     def kill_dhcpcd
       return if not TTY::Which.exist?('dhcpcd')
-      `pgrep -x dhcpcd`
-      @pkill.run("dhcpcd") if $?.success?
+      pids=`pgrep -i dhcpcd`.chomp
+      kill_pids(pids) if $?.success?
       puts "Killed dhcpcd"
     end
 
     def kill_dhclient
-      return if not TTY::Which.exist?('dhclient', paths: ['/bin, /sbin'])
-      `pgrep -x dhclient`
-      @pkill.run("dhclient") if $?.success?
+      return if not TTY::Which.exist?('dhclient', paths: ['/sbin'])
+      pids=`pgrep -i dhclient`.chomp
+      kill_pids(pids) if $?.success?
+
       @rm.run("/run/dhclient.#{@card}.pid") if File.exist? "/run/dhclient.#{@card}.pid"
       @rm.run("/var/lib/dhcp/dhclient.#{@card}.leases") if File.exist? "/var/lib/dhcp/dhclient.#{@card}.leases"
       puts "Killed dhclient"
@@ -51,7 +51,7 @@ module Amnesie
     end
 
     def restart_dhclient
-      return if not TTY::Which.exist?('dhclient', paths: ['/bin, /sbin'])
+      return if not TTY::Which.exist?('dhclient', paths: ['/sbin'])
       dhclient = Helpers::Exec.new("dhclient")
       # command tested on debian, not try on another system yet...
       dhclient.run("-4 -v -i -pf /run/dhclient.#{@card}.pid -lf /var/lib/dhcp/dhclient.#{@card}.leases -I -df /var/lib/dhcp/dhclient6.#{@card}.leases #{@card}")
@@ -70,5 +70,12 @@ module Amnesie
         puts "Restarted tor"
       end
     end
+
+    def kill_pids(pids)
+      pids.lines.each { |p|
+        ps = p.gsub(/\n/, '')
+        @kill.run("-9 #{ps}")
+      }
+    end
   end
 end

data/lib/amnesie/version.rb

@@ -0,0 +1,3 @@
+module Amnesie
+  VERSION = '0.0.9'.freeze
+end

data/lib/nito.rb

@@ -0,0 +1,11 @@
+require_relative 'nito/pass'
+require_relative 'nito/sudo'
+require_relative 'nito/cp'
+require_relative 'nito/cat'
+require_relative 'nito/sed'
+require_relative 'nito/hostname'
+require_relative 'nito/mkdir'
+
+module Nito
+  ID = `id -u`.chomp.freeze
+end

data/lib/nito/cat.rb

@@ -0,0 +1,21 @@
+module Nito
+  class Cat
+    def initialize(conf, string)
+      @conf = conf
+      @string = string
+      write_file
+    end
+
+    private
+    def write_file
+      tmp = Tempfile.new(@conf)
+      if File.exist? @conf
+        File.open(@conf).each { |l|
+          File.write(tmp, l, mode: 'a')
+        }
+      end
+      File.write(tmp, @string, mode: 'a')
+      Nito::Cp.new(tmp.path, @conf)
+    end
+  end
+end

data/lib/nito/cp.rb

@@ -0,0 +1,35 @@
+require 'fileutils'
+
+module Nito
+  class Cp
+    #@@pass = nil
+
+    def initialize(src, dst, perm = 0644)
+      @src = src
+      @dst = dst
+      @perm = perm
+      if ID == "0"
+        root
+      else
+        #@@pass = Pass.new if ! @@pass
+        sudo
+      end
+    end
+
+    private
+
+    def root
+      FileUtils.copy_file(@src, @dst)
+      FileUtils.chmod(@perm, @dst)
+    end
+
+    def sudo
+      #Sudo.run("cp #{@src} #{@dst}", @@pass.secret)
+      Sudo.run("cp #{@src} #{@dst}")
+      perm = sprintf "%o", @perm
+      #puts "Applying perm #{perm}"
+      #Sudo.run("chmod #{perm} #{@dst}", @@pass.secret)
+      Sudo.run("chmod #{perm} #{@dst}")
+    end
+  end
+end

data/lib/nito/hostname.rb

@@ -0,0 +1,24 @@
+require 'tempfile'
+
+module Nito
+  class Hostname
+    def initialize(hostname)
+      @hostname = hostname
+      @hostname_file = '/etc/hostname'
+      @hosts_file = '/etc/hosts'
+      new
+    end
+
+    private
+
+    def new
+      tmp = Tempfile.new('hostname')
+      File.write(tmp, @hostname)
+      Nito::Cp.new(tmp.path, @hostname_file)
+      reg_1 = /^127.0.0.1[\s]+localhost/
+      reg_2 = /^::1[\s]+localhost/
+      Nito::Sed.new(@hosts_file, reg_1, "127.0.0.1 localhost #{@hostname}")
+      Nito::Sed.new(@hosts_file, reg_2, "::1 localhost #{@hostname}")
+    end
+  end
+end

data/lib/nito/mkdir.rb

@@ -0,0 +1,19 @@
+require 'fileutils'
+
+module Nito
+  class Mkdir
+    def initialize(dir)
+      @dir = dir
+      mkdir
+    end
+
+    private
+    def mkdir
+      begin
+        FileUtils.mkdir_p @dir if ! Dir.exist? @dir
+      rescue Errno::EACCES
+        FileUtils.mkdir_p @dir if ! Dir.exist? @dir
+      end
+    end
+  end
+end

data/lib/nito/pass.rb

@@ -0,0 +1,14 @@
+require 'highline/import'
+
+module Nito
+  class Pass 
+    attr_reader :secret
+    def initialize
+      get
+    end
+    private
+    def get(prompt = 'Password: ')
+      @secret = ask(prompt) { |q| q.echo = false }
+    end
+  end
+end

data/lib/nito/sed.rb

@@ -0,0 +1,25 @@
+require 'tempfile'
+
+module Nito
+  class Sed
+    def initialize(file, regex, change)
+      @file = file
+      @regex = regex
+      @change = change
+      apply
+    end
+
+    def apply
+      raise ArgumentError "No file #{@file} exist" if ! File.exist? @file
+      tmp = Tempfile.new('sed')
+      File.open(@file).each { |l|
+        if l.match(@regex)
+          File.write(tmp, "#{@change}\n", mode: 'a')
+        else
+          File.write(tmp, l, mode: 'a')
+        end
+      }
+      Nito::Cp.new(tmp.path, @file)
+    end
+  end
+end

data/lib/nito/sudo.rb

@@ -0,0 +1,16 @@
+module Nito
+  module Sudo
+    def self.run(command, input = nil)
+      #IO.popen("sudo -S #{command}", 'r+') do |io|
+      IO.popen("sudo #{command}", 'r+') do |io|
+        begin
+          io.puts input
+          io.close_write
+          io.read
+        rescue Interrupt
+          puts "\nInterrupt"
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: amnesie
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.9
 platform: ruby
 authors:
 - szorfein
@@ -35,7 +35,7 @@ cert_chain:
   J/zT/q2Ac7BWpSLbv6p9lChBiEnD9j24x463LR5QQjDNS5SsjzRQfFuprsa9Nqf2
   Tw==
   -----END CERTIFICATE-----
-date: 2020-05-15 00:00:00.000000000 Z
+date: 2020-11-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -113,6 +113,20 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 0.4.2
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.0.3
 description: '    A tool to make your computer amnesic"
 
   '
@@ -132,12 +146,27 @@ files:
 - Rakefile
 - amnesie.gemspec
 - bin/amnesie
+- lib/amnesie.rb
+- lib/amnesie/config.rb
 - lib/amnesie/helpers.rb
+- lib/amnesie/host.rb
 - lib/amnesie/mac.rb
 - lib/amnesie/network.rb
 - lib/amnesie/options.rb
+- lib/amnesie/persist.rb
+- lib/amnesie/persist/iwd.rb
+- lib/amnesie/persist/systemd.rb
+- lib/amnesie/persist/wpa_supplicant.rb
 - lib/amnesie/process.rb
-- lib/amnesie/runner.rb
+- lib/amnesie/version.rb
+- lib/nito.rb
+- lib/nito/cat.rb
+- lib/nito/cp.rb
+- lib/nito/hostname.rb
+- lib/nito/mkdir.rb
+- lib/nito/pass.rb
+- lib/nito/sed.rb
+- lib/nito/sudo.rb
 - test/test_mac.rb
 homepage: https://github.com/szorfein/amnesie
 licenses:
@@ -161,7 +190,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: A tool to make your computer amnesic

data/lib/amnesie/runner.rb

@@ -1,31 +0,0 @@
-require_relative 'options'
-require_relative 'mac'
-require_relative 'process'
-require_relative 'network'
-
-module Amnesie
-  class Runner
-    
-    def initialize(argv)
-      @options = Options.new(argv)
-      @network = false
-    end
-    def run
-      if @options.mac then
-        if not @network
-          @network = Amnesie::Network.new(@options.netcard)
-        end
-        puts @network.card
-        process = Amnesie::Process.new(@network.card)
-        card = Amnesie::MAC.new(@network.card)
-
-        process.kill
-        card.save_origin
-        card.rand
-        card.apply
-        puts card.to_s
-        process.restart
-      end
-    end
-  end
-end