/
malware_policy_action.go
213 lines (179 loc) · 7.58 KB
/
malware_policy_action.go
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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
package appsec
import (
"context"
"encoding/json"
"fmt"
"net/http"
validation "github.com/go-ozzo/ozzo-validation/v4"
)
type (
// The MalwarePolicyAction interface supports retrieving and modifying the action associated with
// a specified malware policy, or with all malware policies in a security policy.
MalwarePolicyAction interface {
// GetMalwarePolicyActions retrieves the actions for a specific malware protection policy.
//
// See: https://techdocs.akamai.com/application-security/reference/get-malware-policies-actions
GetMalwarePolicyActions(ctx context.Context, params GetMalwarePolicyActionsRequest) (*GetMalwarePolicyActionsResponse, error)
// UpdateMalwarePolicyAction modifies the actions for a specific malware protection policy.
//
// See: https://techdocs.akamai.com/application-security/reference/put-malware-policy-action
UpdateMalwarePolicyAction(ctx context.Context, params UpdateMalwarePolicyActionRequest) (*UpdateMalwarePolicyActionResponse, error)
// UpdateMalwarePolicyActions is for Akamai internal use only.
UpdateMalwarePolicyActions(ctx context.Context, params UpdateMalwarePolicyActionsRequest) (*UpdateMalwarePolicyActionsResponse, error)
}
// GetMalwarePolicyActionsRequest is used to retrieve a configuration's malware policies and their associated actions.
GetMalwarePolicyActionsRequest struct {
ConfigID int `json:"configID"`
Version int `json:"version"`
PolicyID string `json:"policyID"`
MalwarePolicyID int `json:"id"`
}
// MalwarePolicyActionBody defines the actions for a specific malware policy.
MalwarePolicyActionBody struct {
MalwarePolicyID int `json:"id"`
Action string `json:"action"`
UnscannedAction string `json:"unscannedAction"`
}
// GetMalwarePolicyActionsResponse is returned from a call to GetMalwarePolicyActions.
GetMalwarePolicyActionsResponse struct {
MalwarePolicyActions []MalwarePolicyActionBody `json:"malwarePolicyActions"`
}
// UpdateMalwarePolicyActionRequest is used to update the actions for a malware policy.
UpdateMalwarePolicyActionRequest struct {
ConfigID int `json:"configID"`
Version int `json:"version"`
PolicyID string `json:"policyID"`
MalwarePolicyID int `json:"id"`
Action string `json:"action"`
UnscannedAction string `json:"unscannedAction"`
}
// UpdateMalwarePolicyActionResponse is returned from a call to UpdateMalwarePolicy.
UpdateMalwarePolicyActionResponse struct {
Action string `json:"action"`
UnscannedAction string `json:"unscannedAction"`
}
// UpdateMalwarePolicyActionsRequest is used to update the actions for multiple malware policies.
UpdateMalwarePolicyActionsRequest struct {
ConfigID int
Version int
PolicyID string
MalwarePolicyActions json.RawMessage `json:"-"`
}
// UpdateMalwarePolicyActionsResponse is returned from a call to UpdateMalwarePolicyActions.
UpdateMalwarePolicyActionsResponse GetMalwarePolicyActionsResponse
)
// Validate validates a GetMalwarePolicyActionsRequest.
func (v GetMalwarePolicyActionsRequest) Validate() error {
return validation.Errors{
"ConfigID": validation.Validate(v.ConfigID, validation.Required),
"Version": validation.Validate(v.Version, validation.Required),
"PolicyID": validation.Validate(v.PolicyID, validation.Required),
}.Filter()
}
// Validate validates an UpdateMalwarePolicyActionRequest.
func (v UpdateMalwarePolicyActionRequest) Validate() error {
return validation.Errors{
"ConfigID": validation.Validate(v.ConfigID, validation.Required),
"Version": validation.Validate(v.Version, validation.Required),
"PolicyID": validation.Validate(v.PolicyID, validation.Required),
"MalwarePolicyID": validation.Validate(v.MalwarePolicyID, validation.Required),
"Action": validation.Validate(v.Action, validation.Required),
"UnscannedAction": validation.Validate(v.UnscannedAction, validation.Required),
}.Filter()
}
// Validate validates an UpdateMalwarePolicyActionsRequest.
func (v UpdateMalwarePolicyActionsRequest) Validate() error {
return validation.Errors{
"ConfigID": validation.Validate(v.ConfigID, validation.Required),
"Version": validation.Validate(v.Version, validation.Required),
"PolicyID": validation.Validate(v.PolicyID, validation.Required),
"MalwarePolicyActions": validation.Validate(v.MalwarePolicyActions, validation.Required),
}.Filter()
}
func (p *appsec) GetMalwarePolicyActions(ctx context.Context, params GetMalwarePolicyActionsRequest) (*GetMalwarePolicyActionsResponse, error) {
logger := p.Log(ctx)
logger.Debug("GetMalwarePolicyActions")
if err := params.Validate(); err != nil {
return nil, fmt.Errorf("%w: %s", ErrStructValidation, err.Error())
}
uri := fmt.Sprintf(
"/appsec/v1/configs/%d/versions/%d/security-policies/%s/malware-policies",
params.ConfigID,
params.Version,
params.PolicyID)
req, err := http.NewRequestWithContext(ctx, http.MethodGet, uri, nil)
if err != nil {
return nil, fmt.Errorf("failed to create GetMalwarePolicyActions request: %w", err)
}
var result GetMalwarePolicyActionsResponse
resp, err := p.Exec(req, &result)
if err != nil {
return nil, fmt.Errorf("get malware policy actions request failed: %w", err)
}
if resp.StatusCode != http.StatusOK {
return nil, p.Error(resp)
}
if params.MalwarePolicyID != 0 {
var filteredResult GetMalwarePolicyActionsResponse
for _, val := range result.MalwarePolicyActions {
if val.MalwarePolicyID == params.MalwarePolicyID {
filteredResult.MalwarePolicyActions = append(filteredResult.MalwarePolicyActions, val)
}
}
return &filteredResult, nil
}
return &result, nil
}
func (p *appsec) UpdateMalwarePolicyAction(ctx context.Context, params UpdateMalwarePolicyActionRequest) (*UpdateMalwarePolicyActionResponse, error) {
logger := p.Log(ctx)
logger.Debug("UpdateMalwarePolicyAction")
if err := params.Validate(); err != nil {
return nil, fmt.Errorf("%w: %s", ErrStructValidation, err.Error())
}
uri := fmt.Sprintf(
"/appsec/v1/configs/%d/versions/%d/security-policies/%s/malware-policies/%d",
params.ConfigID,
params.Version,
params.PolicyID,
params.MalwarePolicyID,
)
req, err := http.NewRequestWithContext(ctx, http.MethodPut, uri, nil)
if err != nil {
return nil, fmt.Errorf("failed to create UpdateMalwarePolicyAction request: %w", err)
}
var result UpdateMalwarePolicyActionResponse
resp, err := p.Exec(req, &result, params)
if err != nil {
return nil, fmt.Errorf("update malware policy action request failed: %w", err)
}
if resp.StatusCode != http.StatusOK {
return nil, p.Error(resp)
}
return &result, nil
}
func (p *appsec) UpdateMalwarePolicyActions(ctx context.Context, params UpdateMalwarePolicyActionsRequest) (*UpdateMalwarePolicyActionsResponse, error) {
logger := p.Log(ctx)
logger.Debug("UpdateMalwarePolicyActions")
if err := params.Validate(); err != nil {
return nil, fmt.Errorf("%w: %s", ErrStructValidation, err.Error())
}
uri := fmt.Sprintf(
"/appsec/v1/configs/%d/versions/%d/security-policies/%s/malware-policies",
params.ConfigID,
params.Version,
params.PolicyID,
)
req, err := http.NewRequestWithContext(ctx, http.MethodPut, uri, nil)
if err != nil {
return nil, fmt.Errorf("failed to create UpdateMalwarePolicyActions request: %w", err)
}
var result UpdateMalwarePolicyActionsResponse
resp, err := p.Exec(req, &result, params.MalwarePolicyActions)
if err != nil {
return nil, fmt.Errorf("update malware policy actions request failed: %w", err)
}
if resp.StatusCode != http.StatusOK {
return nil, p.Error(resp)
}
return &result, nil
}