zlown 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 19a2c7df1dee8add69e4ebf2f10c5d695196957b
-  data.tar.gz: ccdc99c8c26528d5e2b5c79f96c06f1f64615aac
+  metadata.gz: 7b6ece0b0fb285bc512cae0a12d7a7455b14071a
+  data.tar.gz: ea1f3d87fe1404dbb06bc8db43197e5531174ed9
 SHA512:
-  metadata.gz: bb981a35454d833cef5f91f4fb7a26f5a5319c6d945bd544a55829d447fcbf60fcfd407b911c3f12a889f9f46be730841bbfffda25c04f891f3ebb0d67b608ad
-  data.tar.gz: 810175e2003cf970e08dbe36e4d1223f1e66dcb59aaeb06ae8a0ce47ed1db47851acf3fda1c0601dc971934d91e8b1b2fa4c7271d88628d96be14923087f5189
+  metadata.gz: ea460cde4a975bb2dd6d375827af0782f83fd80f7312353e2c52b9614fe594e088a2c28800a6c1c5dd3a49c72c059487535a391453f3b379facc36ee159f6a61
+  data.tar.gz: fbf83a149f6c70d12895b9b6db844f89041e0ebf84f7c109334cd78b38cfb8e3cb2600412e3fad787c4deae583e29fcaed83d90194ab4038570524258e397afc

data/etc/dnsmasq.conf

@@ -1,6 +1,6 @@
 # EVIL STARTS HERE
 log-facility=/var/log/dnsmasq.log
-interface=wlan1
+interface=${IFACE_AP}
 dhcp-range=10.0.0.10,10.0.0.250,12h
 dhcp-option=3,10.0.0.1
 dhcp-option=6,10.0.0.1

data/etc/hostapd/hostapd.conf

@@ -1,7 +1,6 @@
-interface=wlan1
-driver=nl80211
-ssid=FreeWifi
-channel=6
+interface=${IFACE}
+driver=${DRIVER}
+ssid=${SSID}
+channel=${CHANNEL}
 # Yes, we support the Karma attack.
 # enable_karma=1
-

data/etc/rc.local

@@ -11,6 +11,6 @@
 #
 # By default this script does nothing.
 
-exec /root/scripts/enable-rogue.sh
+exec /root/.zlown/enable-forwarding.sh
 
 exit 0

data/lib/zlown/cli/cmd/update_cmd.rb

@@ -0,0 +1,12 @@
+require_relative '../shared'
+
+require_relative '../../core/core'
+
+module Zlown::CLI
+  desc 'Update all required scripts based on config'
+  command 'update' do |c|
+    c.action do |global_options, options, args|
+      Core.update_configs(args, options)
+    end
+  end
+end

data/lib/zlown/config.rb

@@ -0,0 +1,42 @@
+# encoding: UTF-8
+#
+# Copyright (c) 2016 Tomas Korcak <korczis@gmail.com>. All rights reserved.
+# This source code is licensed under the MIT-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+module Zlown
+  module Config
+    APP_DIR = File.expand_path('~/.zlown')
+    APP_BINARY = File.expand_path('../../../bin/zlown', __FILE__)
+
+    CONFIG_FILE = File.join(APP_DIR, 'zlown.yml')
+
+    BOOT_SCRIPT = File.join(APP_DIR, 'enable-forwarding.sh')
+    BOOT_SCRIPT_TEMPLATE = File.expand_path('../../../scripts/enable-forwarding.sh', __FILE__)
+
+    DNSMASQ_CONFIG = '/etc/dnsmasq.conf'
+    DNSMASQ_TEMPLATE = File.expand_path('../../../etc/dnsmasq.conf', __FILE__)
+
+    HOSTAPD_CONFIG = '/etc/hostapd/hostapd.conf'
+    HOSTAPD_TEMPLATE = File.expand_path('../../../etc/hostapd/hostapd.conf', __FILE__)
+
+    HTTPRY_PID_FILE = File.join(APP_DIR, 'run', 'httpry.pid')
+
+    DATA_DIR = File.join(APP_DIR, 'data')
+    RUN_DIR = File.join(APP_DIR, 'run')
+
+    SERVICE_TEMPLATE = File.expand_path('../../../etc/systemd/system/zlown.service', __FILE__)
+    SERVICE_FILE = File.expand_path("#{APP_DIR}/zlown.service")
+
+    RCLOCAL_CONFIG = '/etc/rc.local'
+    RCLOCAL_TEMPLATE = File.expand_path('../../../etc/rc.local', __FILE__)
+
+    RUN_CMD = "#{APP_BINARY} run"
+
+    DNSMASQ_SERVICE = 'dnsmasq.service'
+    HOSTAPD_SERVICE = 'hostapd.service'
+
+    ZLOWN_SERVICE_FILE = File.expand_path('~/.zlown/zlown.service')
+    ZLOWN_SERVICE_NAME = 'zlown'
+  end
+end

data/lib/zlown/core/core.rb

@@ -8,23 +8,23 @@ require 'fileutils'
 require 'highline'
 require 'yaml'
 
+require_relative '../config'
+
 module Zlown
   class Core
-    APP_DIR = File.expand_path('~/.zlown')
-    APP_BINARY = File.expand_path('../../../../bin/zlown', __FILE__)
-
-    CONFIG_FILE = File.join(APP_DIR, 'zlown.yml')
-
-    DATA_DIR = File.join(APP_DIR, 'data')
-    RUN_DIR = File.join(APP_DIR, 'run')
-
-    SERVICE_TEMPLATE = File.expand_path('../../../../etc/systemd/system/zlown.service', __FILE__)
-    SERVICE_FILE = File.expand_path("#{APP_DIR}/zlown.service")
-
-    RUN_CMD = "#{APP_BINARY} run"
-
-    DNSMASQ_SERVICE = 'dnsmasq.service'
-    HOSTAPD_SERVICE = 'hostapd.service'
+    def self.load_config(args = [], opts = {})
+      config = {
+        upstream: 'eth0',
+        ap: 'wlan0',
+        driver: 'nl80211',
+        ssid: 'FreeWifi',
+        channel: '6'
+      }
+
+      if File.exist?(Zlown::Config::CONFIG_FILE)
+        config = config.merge(YAML.load(File.open(Zlown::Config::CONFIG_FILE)))
+      end
+    end
 
     def self.install(args = [], opts = {})
       cmd = 'apt-get install -y hostapd dnsmasq wireless-tools iw wvdial'
@@ -32,67 +32,159 @@ module Zlown
       system cmd
     end
 
-    def self.init(args = [], opts = {})
-      unless File.directory?(APP_DIR)
-        puts "Creating directory #{APP_DIR}"
-        FileUtils.mkdir_p(APP_DIR)
+    def self.init_dirs(args = [], opts = {})
+      unless File.directory?(Zlown::Config::APP_DIR)
+        puts "Creating directory #{Zlown::Config::APP_DIR}"
+        FileUtils.mkdir_p(Zlown::Config::APP_DIR)
       end
 
-      unless File.directory?(DATA_DIR)
-        puts "Creating directory #{DATA_DIR}"
-        FileUtils.mkdir_p(DATA_DIR)
+      unless File.directory?(Zlown::Config::DATA_DIR)
+        puts "Creating directory #{Zlown::Config::DATA_DIR}"
+        FileUtils.mkdir_p(Zlown::Config::DATA_DIR)
       end
 
-      unless File.directory?(RUN_DIR)
-        puts "Creating directory #{RUN_DIR}"
-        FileUtils.mkdir_p(RUN_DIR)
+      unless File.directory?(Zlown::Config::RUN_DIR)
+        puts "Creating directory #{Zlown::Config::RUN_DIR}"
+        FileUtils.mkdir_p(Zlown::Config::RUN_DIR)
       end
+    end
 
-      template = File.read(SERVICE_TEMPLATE)
-      content = template.gsub('#{RUN_CMD}', RUN_CMD)
+    def self.init_service_template(args = [], opts = {})
+      template = File.read(Zlown::Config::SERVICE_TEMPLATE)
+      content = template.gsub('#{RUN_CMD}', Zlown::Config::RUN_CMD)
 
       # To write changes to the file, use:
-      File.open(SERVICE_FILE, 'w') do |file|
-        puts "Writting file #{SERVICE_FILE}"
+      File.open(Zlown::Config::SERVICE_FILE, 'w') do |file|
+        puts "Writting file #{Zlown::Config::SERVICE_FILE}"
         file.puts content
       end
+    end
 
-      config = {}
-      if File.exist?(CONFIG_FILE)
-        config = YAML.load(File.open(CONFIG_FILE))
-      end
+    def self.init_config_file(args = [], opts = {})
+      config = Core.load_config(args, opts)
 
       cli = HighLine.new
-      config[:upstream] = cli.ask('upstream interface?') { |q| q.default = config[:upstream] || 'eth0' }
-      config[:ap] = cli.ask('wifi ap interface?') { |q| q.default = config[:ap] || 'wlan0pa' }
-
-      puts "Writting config to #{CONFIG_FILE}"
-      File.open(CONFIG_FILE, 'w') do |f|
+      config[:upstream] = cli.ask('Upstream Interface?') { |q| q.default = config[:upstream] }
+      config[:ap] = cli.ask('Wi-Fi ap Interface?') { |q| q.default = config[:ap] }
+      config[:driver] = cli.ask('Wi-Fi Driver?') { |q| q.default = config[:driver] }
+      config[:ssid] = cli.ask('Wi-Fi SSID?') { |q| q.default = config[:ssid] }
+      config[:channel] = cli.ask('Wi-Fi Channel?') { |q| q.default = config[:channel] }
+
+      puts "Writting config to #{Zlown::Config::CONFIG_FILE}"
+      File.open(Zlown::Config::CONFIG_FILE, 'w') do |f|
         f.write config.to_yaml
       end
 
-      # See https://www.offensive-security.com/kali-linux/kali-linux-evil-wireless-access-point/
-      cmd = "sed -i 's#^DAEMON_CONF=.*#DAEMON_CONF=/etc/hostapd/hostapd.conf#' /etc/init.d/hostapd"
+      config
+    end
+
+    def self.init_systemctl(args = [], opts = {})
+      # TODO: Process dnsmasq.conf and hostapd.conf
+
+      cmd = "systemctl enable #{Zlown::Config::HOSTAPD_SERVICE}"
       puts cmd
       system cmd
 
-      # TODO: Process dnsmasq.conf and hostapd.conf
+      cmd = "systemctl enable #{Zlown::Config::DNSMASQ_SERVICE}"
+      puts cmd
+      system cmd
 
-      cmd = "systemctl enable #{HOSTAPD_SERVICE}"
+      cmd = "systemctl start #{Zlown::Config::HOSTAPD_SERVICE}"
       puts cmd
       system cmd
 
-      cmd = "systemctl enable #{DNSMASQ_SERVICE}"
+      cmd = "systemctl start #{Zlown::Config::DNSMASQ_SERVICE}"
       puts cmd
       system cmd
+    end
+
+    def self.init_dnsmaq(args = [], opts = {})
+      config = Core.load_config(args, opts)
 
-      cmd = "systemctl start #{HOSTAPD_SERVICE}"
+      template = File.read(Zlown::Config::DNSMASQ_TEMPLATE)
+      content = template.gsub('${IFACE_AP}', config[:ap])
+
+      # To write changes to the file, use:
+      File.open(Zlown::Config::DNSMASQ_CONFIG, 'w') do |file|
+        puts "Writting file #{Zlown::Config::DNSMASQ_CONFIG}"
+        file.puts content
+      end
+    end
+
+    def self.init_hostapd(args = [], opts = {})
+      # See https://www.offensive-security.com/kali-linux/kali-linux-evil-wireless-access-point/
+      cmd = "sed -i 's#^DAEMON_CONF=.*#DAEMON_CONF=/etc/hostapd/hostapd.conf#' /etc/init.d/hostapd"
       puts cmd
       system cmd
 
-      cmd = "systemctl start #{DNSMASQ_SERVICE}"
+      config = Core.load_config(args, opts)
+
+      template = File.read(Zlown::Config::HOSTAPD_TEMPLATE)
+      content = template
+                  .gsub('${IFACE}', config[:ap])
+                  .gsub('${DRIVER}', config[:driver])
+                  .gsub('${SSID}', config[:ssid])
+                  .gsub('${CHANNEL}', config[:channel])
+
+      # To write changes to the file, use:
+      File.open(Zlown::Config::HOSTAPD_CONFIG, 'w') do |file|
+        puts "Writting file #{Zlown::Config::HOSTAPD_CONFIG}"
+        file.puts content
+      end
+    end
+
+    def self.init_boot_script(args = [], opts = {})
+      config = Core.load_config(args, opts)
+
+      template = File.read(Zlown::Config::BOOT_SCRIPT_TEMPLATE)
+      content = template
+                  .gsub('${IFACE_AP}', config[:ap])
+                  .gsub('${IFACE_UPSTREAM}', config[:upstream])
+
+      # To write changes to the file, use:
+      File.open(Zlown::Config::BOOT_SCRIPT, 'w') do |file|
+        puts "Writting file #{Zlown::Config::BOOT_SCRIPT}"
+        file.puts content
+      end
+
+      cmd = "chmod +x #{Zlown::Config::BOOT_SCRIPT}"
       puts cmd
       system cmd
     end
+
+    def self.init_rc_local(args = [], opts = {})
+      Core.init_boot_script(args, opts)
+
+      config = Core.load_config(args, opts)
+
+      template = File.read(Zlown::Config::RCLOCAL_TEMPLATE)
+      content = template
+
+      # To write changes to the file, use:
+      File.open(Zlown::Config::RCLOCAL_CONFIG, 'w') do |file|
+        puts "Writting file #{Zlown::Config::RCLOCAL_CONFIG}"
+        file.puts content
+      end
+    end
+
+    def self.update_configs(args = [], opts = {})
+      Core.init_dnsmaq(args, opts)
+
+      Core.init_hostapd(args, opts)
+
+      Core.init_rc_local(args, opts)
+
+      Core.init_systemctl(args, opts)
+    end
+
+    def self.init(args = [], opts = {})
+      Core.init_dirs(args, opts)
+
+      Core.init_service_template(args, opts)
+
+      Core.init_config_file(args, opts)
+
+      Core.update_configs(args, opts)
+    end
   end
 end

data/lib/zlown/script/script.rb

@@ -4,18 +4,18 @@
 # This source code is licensed under the MIT-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+require_relative '../config'
+
 module Zlown
   class Script
-    HTTPRY_PID_FILE = '/root/.zlown/run/httpry.pid'
-
     def self.httpry_start(args = [], opts = {})
-      cmd = "httpry -d -P #{HTTPRY_PID_FILE} -i wlan1 -o /root/.zlown/data/httpry.log -b /root/.zlown/data/httpry.bin"
+      cmd = "httpry -d -P #{Zlown::Config::HTTPRY_PID_FILE} -i wlan1 -o /root/.zlown/data/httpry.log -b /root/.zlown/data/httpry.bin"
       puts cmd
       system cmd
     end
 
     def self.httpry_stop(args = [], opts = {})
-      pid = File.open(HTTPRY_PID_FILE).read.to_i
+      pid = File.open(Zlown::Config::HTTPRY_PID_FILE).read.to_i
       cmd = "kill #{pid}"
       puts cmd
       system cmd

data/lib/zlown/systemctl/systemctl.rb

@@ -4,27 +4,26 @@
 # This source code is licensed under the MIT-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+require_relative '../config'
+
 module Zlown
   class Systemctl
-    SERVICE_FILE = File.expand_path('~/.zlown/zlown.service')
-    SERVICE_NAME = 'zlown'
-
     def self.enable(args = [], opts = {})
       puts 'Enabling systemctl service'
-      cmd = "systemctl enable #{SERVICE_FILE}"
+      cmd = "systemctl enable #{Zlown::Config::ZLOWN_SERVICE_FILE}"
       puts cmd
       system cmd
     end
 
     def self.disable(args = [], opts = {})
       puts 'Disabling systemctl service'
-      cmd = "systemctl disable #{SERVICE_NAME}"
+      cmd = "systemctl disable #{Zlown::Config::ZLOWN_SERVICE_NAME}"
       system cmd
     end
 
     def self.start(args = [], opts = {})
       puts 'Starting systemctl service'
-      cmd = "systemctl start #{SERVICE_NAME}"
+      cmd = "systemctl start #{Zlown::Config::ZLOWN_SERVICE_NAME}"
       system cmd
     end
 
@@ -35,12 +34,12 @@ module Zlown
     end
 
     def self.status(args = [], opts = {})
-      cmd = "systemctl status #{SERVICE_NAME}"
+      cmd = "systemctl status #{Zlown::Config::ZLOWN_SERVICE_NAME}"
       system cmd
     end
 
     def self.log(args = [], opts = {})
-      cmd = "journalctl -u #{SERVICE_NAME}"
+      cmd = "journalctl -u #{Zlown::Config::ZLOWN_SERVICE_NAME}"
       system cmd
     end
   end

data/lib/zlown/version.rb

@@ -5,5 +5,5 @@
 # LICENSE file in the root directory of this source tree.
 
 module Zlown
-  VERSION = '0.0.2'.freeze
+  VERSION = '0.0.3'.freeze
 end

data/scripts/enable-forwarding.sh

@@ -0,0 +1,8 @@
+#! /usr/bin/env bash
+
+iptables -t nat -F
+iptables -F
+iptables -t nat -A POSTROUTING -o ${IFACE_UPSTREAM} -j MASQUERADE
+iptables -A FORWARD -i ${IFACE_AP} -o ${IFACE_UPSTREAM} -j ACCEPT
+
+echo '1' > /proc/sys/net/ipv4/ip_forward

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: zlown
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Tomas Korcak
@@ -134,14 +134,16 @@ files:
 - lib/zlown/cli/cmd/run_cmd.rb
 - lib/zlown/cli/cmd/script_cmd.rb
 - lib/zlown/cli/cmd/systemctl_cmd.rb
+- lib/zlown/cli/cmd/update_cmd.rb
 - lib/zlown/cli/cmd/version_cmd.rb
 - lib/zlown/cli/shared.rb
+- lib/zlown/config.rb
 - lib/zlown/core/core.rb
 - lib/zlown/daemon/daemon.rb
 - lib/zlown/script/script.rb
 - lib/zlown/systemctl/systemctl.rb
 - lib/zlown/version.rb
-- scripts/enable-rogue.sh
+- scripts/enable-forwarding.sh
 - scripts/sniff-httpry.sh
 - scripts/sniff-ngrep.sh
 - scripts/wifi-power.sh

data/scripts/enable-rogue.sh

@@ -1,12 +0,0 @@
-#! /usr/bin/env bash
-
-IF_WIFI=wlan1
-IF_LTE=eth1
-
-iptables -t nat -F
-iptables -F
-iptables -t nat -A POSTROUTING -o ${IF_LTE} -j MASQUERADE
-iptables -A FORWARD -i ${IF_WIFI} -o ${IF_LTE} -j ACCEPT
-
-# echo '1' > /proc/sys/net/ipv4/ip_forward
-