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
73
74
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
package hisgle
import ( "fmt"
"log"
"os"
"strconv" )
// http://go-database-sql.org/importing.html
// _ "github.com/mattn/go-sqlite3"
// _ "code.google.com/p/gosqlite/sqlite3"
import (
"database/sql"
_ "github.com/mxk/go-sqlite/sqlite3"
"github.com/pborman/getopt"
)
func ethers(db *sql.DB, vlan int) {
var (
mac string
ipv4 string
)
// CREATE TABLE devices (name, mac, interface);
// CREATE TABLE addresses (mac, vlan, ipv4);
rows, err := db.Query("SELECT mac, ipv4 FROM addresses WHERE vlan=?", vlan)
if err != nil {
log.Fatal(err)
}
for rows.Next() {
err := rows.Scan(&mac, &ipv4)
if err != nil {
log.Fatal(err)
}
fmt.Println(mac, " ", ipv4)
}
}
func hosts(db *sql.DB, vlan int) {
var (
ipv4 string
name string
alias string
)
// CREATE TABLE devices (name, mac, interface);
// CREATE TABLE addresses (mac, vlan, ipv4);
rows, err := db.Query("SELECT a.ipv4, d.name FROM addresses AS a JOIN " +
"devices AS d ON a.mac=d.mac WHERE vlan=?", vlan)
if err != nil {
log.Fatal(err)
}
for rows.Next() {
err := rows.Scan(&ipv4, &name)
if err != nil {
log.Fatal(err)
}
fmt.Print(ipv4, "\t", name)
// CREATE TABLE aliases (alias, target);
rows, err := db.Query("SELECT alias FROM aliases WHERE target=?", name)
if err != nil {
log.Fatal(err)
}
for rows.Next() {
err := rows.Scan(&alias)
if err != nil {
log.Fatal(err)
}
fmt.Print(" ", alias)
}
fmt.Println()
}
}
func main() {
var (
description string
ipv4 string
)
optDatabase := getopt.StringLong("database", 'd', "",
"Path to SQLite3 database file with device addresses")
optHelp := getopt.BoolLong("help", 0, "Help")
getopt.SetParameters("{ethers|hosts} list_of_networks")
getopt.Parse()
if *optHelp {
getopt.Usage()
os.Exit(0)
}
args := getopt.Args()
db, err := sql.Open("sqlite3", *optDatabase)
if err != nil {
log.Fatal(err)
}
for _, arg := range args[1:] {
vlan, _ := strconv.Atoi(arg)
rows, err := db.Query(
"SELECT description, ipv4 FROM networks WHERE vlan=?", vlan)
if err != nil {
log.Fatal(err)
}
for rows.Next() {
err := rows.Scan(&description, &ipv4)
if err != nil {
log.Fatal(err)
}
fmt.Print("\n# ", description, ", ", ipv4, "\n")
fmt.Println("# vlan:", vlan)
switch args[0] {
case "ethers":
ethers(db, vlan)
case "hosts":
hosts(db, vlan)
default:
os.Exit(1)
}
}
}
db.Close()
}
|