scripts/ospf.rb at master · earlzero/scripts · GitHub

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
69
70
71
72
#!/usr/bin/ruby
require 'rubygems'
require 'snmp'
require 'mathn'
if ARGV.length>0
	dupes=Hash.new
        iptoid=Hash.new
	ridtoip=Hash.new
	f=File.open(ARGV[0],"r")
	out=File.new("result.gml","w")
	out.puts "Creator \"Selfmade\"\nVersion \"0.0\"\ngraph\n[\ndirected 1"
	ip=f.readlines
	i=0
        ip.each do |str|
                manager=SNMP::Manager.new(:Host=>str.chomp,:Version=> :SNMPv2c)
		iptoid[manager.get_value("1.3.6.1.2.1.14.1.1.0")]=i
		ridtoip[i]=str.chomp
		i+=1
                manager.walk("1.3.6.1.2.1.14.10.1.1") do |vb|
                        manager2=SNMP::Manager.new(:Host=>vb.value.to_s,:Version=> :SNMPv2c)
                        begin
                        printf ";%s",manager2.get_value("1.3.6.1.2.1.14.1.1.0").to_s
                        rescue => bang
                                puts "Fuck"+vb.value+":"+bang
                        end
                end
		puts
        end
	I=Array.new(i)
	I.map! {Array.new(i)}
	iptoid.each do |val1,val2|
		out.puts "node ["
		out.puts "id "+val2.to_s
		out.puts "label \""+ridtoip[val2].to_s+"\"\n]"
	end
        ip.each do |str|
                manager=SNMP::Manager.new(:Host=>str.chomp,:Version=> :SNMPv2c)
                manager.walk("1.3.6.1.2.1.14.10.1.1") do |vb|
                        manager2=SNMP::Manager.new(:Host=>vb.value.to_s,:Version=> :SNMPv2c)
                        begin
                        printf ";%s",manager2.get_value("1.3.6.1.2.1.14.1.1.0").to_s
			i=iptoid[manager.get_value("1.3.6.1.2.1.14.1.1.0")].to_i
			j=iptoid[manager2.get_value("1.3.6.1.2.1.14.1.1.0")].to_i
			if(iptoid[manager.get_value("1.3.6.1.2.1.14.1.1.0")].to_s.length>0 && iptoid[manager2.get_value("1.3.6.1.2.1.14.1.1.0")].to_s.length>0)
			I[i][j]=1
			I[j][i]=0
#			out.puts "edge\n[\nsource "+iptoid[manager.get_value("1.3.6.1.2.1.14.1.1.0")].to_s+"\ntarget "+iptoid[manager2.get_value("1.3.6.1.2.1.14.1.1.0")].to_s+"\n]"
			end
#			out.puts iptoid[manager.get_value("1.3.6.1.2.1.14.1.1.0")].to_s.length
#			out.puts iptoid[manager2.get_value("1.3.6.1.2.1.14.1.1.0")].to_s.length
                        rescue => bang
                                puts "Fuck"+vb.value+":"+bang
                        end
                end
		puts
	end
	for i in 0..I.size
		for j in i+1..I.size
			I[i][j]=0
		end
	end
	for i in 0..I.size-1
		for j in 0..I.size-1
			if(I[i][j]==1)
			out.puts "edge\n[\nsource "+i.to_s+"\ntarget "+j.to_s+"\n]"
			end
	end
	end
	out.puts "]"
	out.close
	puts ridtoip
end