Class: THC::RC

Inherits:
Object
  • Object
show all
Defined in:
lib/thc/rc.rb

Constant Summary collapse

DEFAULT_TORRC = 
"/etc/tor/torrc"
HEADER = 
"# THC"

Class Method Summary collapse

Class Method Details

.add_hs(dport, sport, id) ⇒ Object 



64
65
66
67
68
69
70
71
72
73
 
# File 'lib/thc/rc.rb', line 64

def self.add_hs(dport,sport,id)
  c,p=RC::check(dport.to_i,sport.to_i)
  if c<0
    File.open(@torrc, 'a+') do |f|
        f.puts "HiddenServiceDir /var/tor/.tor_hidden_"+id.to_s+"/\nHiddenServicePort "+dport+" 127.0.0.1:"+sport  
    end 
  else
    raise "Hidden Service already exist on "+p+" port: "+c.to_s
  end
end

.check(dp, sp) ⇒ Object 



42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/thc/rc.rb', line 42

def self.check(dp,sp)
  c=-1
  p=""
  File.open(@torrc, 'r') do |f|
    f.each_line do |line|
      if (line.include?("HiddenServicePort")) && (c == -1)
        if line.split(" ")[1].to_i == dp
          c=dp
          p="destination"
          break
        end
        if line.split(" ").last.split(":").last.to_i == sp
          c=sp
          p="source"
          break
        end
      end
    end
  end
  return c,p
end

.init(path) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 
# File 'lib/thc/rc.rb', line 9

def self.init(path)
  @torrc = path
  y=false
  File.open(@torrc, 'r') do |f|
    f.each_line do |line|
      if line.include?(HEADER)
        y=true
      end
    end
  end
  if !y 
    FileUtils.cp(@torrc, @torrc+"-old")
    File.open(@torrc, 'a+') do |f|
        f.puts HEADER
    end 
  end
end

.load(config) ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/thc/rc.rb', line 27

def self.load(config)
  count=-1;
  File.open(@torrc, 'r') do |f|
    f.each_line do |line|
      if line.include?("HiddenServiceDir")
        config << {'path' => line.split(" ").last, 'service' => []}
        count+=1;
      elsif line.include?("HiddenServicePort")
        config[count]['service'] << {'dport' => line.split(" ")[1].to_i, 'sport' => line.split(" ").last.split(":").last.to_i}
        #puts config.to_s
      end
    end
  end
end

.rm_hs(id) ⇒ Object 



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
 
# File 'lib/thc/rc.rb', line 75

def self.rm_hs(id)
  y=false
  c=0
  File.open(@torrc+"_temp", "w") do |out_file|
    File.foreach(@torrc) do |line|
      if line.include?(HEADER)
       y=true
       out_file.puts line
      elsif y && line.include?("HiddenServiceDir")
       c+=1
        if id != c
          out_file.puts line
        end
      elsif y && line.include?("HiddenServicePort")
        if id != c
          out_file.puts line
        end
      else 
       out_file.puts line
      end
    end
  end

  FileUtils.mv(@torrc+"_temp", @torrc)
end