fastnetmon-go/blackhole.go at main · FastNetMon/fastnetmon-go · 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
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
126
127
128
129
130
131
132
133
package fastnetmon

import (
	"fmt"

	"github.com/levigross/grequests"
)

// Blocks some specified blackhole host
func (client *FastNetMonClient) Blackhole(ip_address string) (bool, error) {
	resp, err := grequests.Put(client.Prefix+"/blackhole/"+ip_address, grequests.FromRequestOptions(client.Ro))

	if err != nil {
		return false, fmt.Errorf("Cannot connect to API: %w", err)
	}

	if !resp.Ok {
		return false, apiStatusError(resp)
	}

	response := ErrorJson{}
	err = decodeResponseJSON(resp, &response)
	if err != nil {
		return false, err
	}

	return response.Success, nil
}

// Returns all IPs blocked using blackhole
func (client *FastNetMonClient) GetBlackhole() ([]BlackholeAnnounces, error) {
	resp, err := grequests.Get(client.Prefix+"/blackhole", grequests.FromRequestOptions(client.Ro))

	if err != nil {
		return nil, fmt.Errorf("Cannot connect to API: %w", err)
	}

	if !resp.Ok {
		return nil, apiStatusError(resp)
	}

	ban_list_response := ResponseRemoteBlackholeListJson{}
	err = decodeResponseJSON(resp, &ban_list_response)
	if err != nil {
		return nil, err
	}

	return ban_list_response.Values, nil
}

// Removes blackhole entry using UUID
func (client *FastNetMonClient) RemoveBlackhole(mitigation_uuid string) (bool, error) {
	resp, err := grequests.Delete(client.Prefix+"/blackhole/"+mitigation_uuid, grequests.FromRequestOptions(client.Ro))

	if err != nil {
		return false, fmt.Errorf("Cannot connect to API: %w", err)
	}

	if !resp.Ok {
		return false, apiStatusError(resp)
	}

	response := ErrorJson{}
	err = decodeResponseJSON(resp, &response)
	if err != nil {
		return false, err
	}

	return response.Success, nil
}

// Blocks some specified remote blackhole host
func (client *FastNetMonClient) BlackholeRemote(ip_address string) (bool, error) {
	resp, err := grequests.Put(client.Prefix+"/remote_blackhole/"+ip_address, grequests.FromRequestOptions(client.Ro))

	if err != nil {
		return false, fmt.Errorf("Cannot connect to API: %w", err)
	}

	if !resp.Ok {
		return false, apiStatusError(resp)
	}

	response := ErrorJson{}
	err = decodeResponseJSON(resp, &response)
	if err != nil {
		return false, err
	}

	return response.Success, nil
}

// Returns all IPs blocked using remote blackhole
func (client *FastNetMonClient) GetRemoteBlackhole() ([]BlackholeAnnounces, error) {
	resp, err := grequests.Get(client.Prefix+"/remote_blackhole", grequests.FromRequestOptions(client.Ro))

	if err != nil {
		return nil, fmt.Errorf("Cannot connect to API: %w", err)
	}

	if !resp.Ok {
		return nil, apiStatusError(resp)
	}

	ban_list_response := ResponseRemoteBlackholeListJson{}
	err = decodeResponseJSON(resp, &ban_list_response)
	if err != nil {
		return nil, err
	}

	return ban_list_response.Values, nil
}

// Removes remote blackhole entry using UUID
func (client *FastNetMonClient) RemoveRemoteBlackhole(mitigation_uuid string) (bool, error) {
	resp, err := grequests.Delete(client.Prefix+"/remote_blackhole/"+mitigation_uuid, grequests.FromRequestOptions(client.Ro))

	if err != nil {
		return false, fmt.Errorf("Cannot connect to API: %w", err)
	}

	if !resp.Ok {
		return false, apiStatusError(resp)
	}

	response := ErrorJson{}
	err = decodeResponseJSON(resp, &response)
	if err != nil {
		return false, err
	}

	return response.Success, nil
}