dia-gov commited on
Commit
114ebf7
·
verified ·
1 Parent(s): f2fe5ce

Upload 19 files

Browse files
cloud_security/cloud_compliance.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+
2
+ def verify_cloud_compliance(config):
3
+ print(f"Verifying cloud compliance for configuration: {config}")
4
+ return {"compliant": True, "violations": []}
5
+
6
+ if __name__ == "__main__":
7
+ compliance_results = verify_cloud_compliance({"cloud_provider": "AWS", "encryption_enabled": True})
8
+ print(f"Compliance Results: {compliance_results}")
color-convert/CHANGELOG.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 1.0.0 - 2016-01-07
2
+
3
+ - Removed: unused speed test
4
+ - Added: Automatic routing between previously unsupported conversions
5
+ ([#27](https://github.com/Qix-/color-convert/pull/27))
6
+ - Removed: `xxx2xxx()` and `xxx2xxxRaw()` functions
7
+ ([#27](https://github.com/Qix-/color-convert/pull/27))
8
+ - Removed: `convert()` class
9
+ ([#27](https://github.com/Qix-/color-convert/pull/27))
10
+ - Changed: all functions to lookup dictionary
11
+ ([#27](https://github.com/Qix-/color-convert/pull/27))
12
+ - Changed: `ansi` to `ansi256`
13
+ ([#27](https://github.com/Qix-/color-convert/pull/27))
14
+ - Fixed: argument grouping for functions requiring only one argument
15
+ ([#27](https://github.com/Qix-/color-convert/pull/27))
16
+
17
+ # 0.6.0 - 2015-07-23
18
+
19
+ - Added: methods to handle
20
+ [ANSI](https://en.wikipedia.org/wiki/ANSI_escape_code#Colors) 16/256 colors:
21
+ - rgb2ansi16
22
+ - rgb2ansi
23
+ - hsl2ansi16
24
+ - hsl2ansi
25
+ - hsv2ansi16
26
+ - hsv2ansi
27
+ - hwb2ansi16
28
+ - hwb2ansi
29
+ - cmyk2ansi16
30
+ - cmyk2ansi
31
+ - keyword2ansi16
32
+ - keyword2ansi
33
+ - ansi162rgb
34
+ - ansi162hsl
35
+ - ansi162hsv
36
+ - ansi162hwb
37
+ - ansi162cmyk
38
+ - ansi162keyword
39
+ - ansi2rgb
40
+ - ansi2hsl
41
+ - ansi2hsv
42
+ - ansi2hwb
43
+ - ansi2cmyk
44
+ - ansi2keyword
45
+ ([#18](https://github.com/harthur/color-convert/pull/18))
46
+
47
+ # 0.5.3 - 2015-06-02
48
+
49
+ - Fixed: hsl2hsv does not return `NaN` anymore when using `[0,0,0]`
50
+ ([#15](https://github.com/harthur/color-convert/issues/15))
51
+
52
+ ---
53
+
54
+ Check out commit logs for older releases
color-convert/LICENSE ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Copyright (c) 2011-2016 Heather Arthur <fayearthur@gmail.com>
2
+
3
+ Permission is hereby granted, free of charge, to any person obtaining
4
+ a copy of this software and associated documentation files (the
5
+ "Software"), to deal in the Software without restriction, including
6
+ without limitation the rights to use, copy, modify, merge, publish,
7
+ distribute, sublicense, and/or sell copies of the Software, and to
8
+ permit persons to whom the Software is furnished to do so, subject to
9
+ the following conditions:
10
+
11
+ The above copyright notice and this permission notice shall be
12
+ included in all copies or substantial portions of the Software.
13
+
14
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
15
+ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16
+ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
17
+ NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
18
+ LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
19
+ OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
20
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
21
+
color-convert/conversions.js ADDED
@@ -0,0 +1,839 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ /* MIT license */
2
+ /* eslint-disable no-mixed-operators */
3
+ const cssKeywords = require('color-name');
4
+
5
+ // NOTE: conversions should only return primitive values (i.e. arrays, or
6
+ // values that give correct `typeof` results).
7
+ // do not use box values types (i.e. Number(), String(), etc.)
8
+
9
+ const reverseKeywords = {};
10
+ for (const key of Object.keys(cssKeywords)) {
11
+ reverseKeywords[cssKeywords[key]] = key;
12
+ }
13
+
14
+ const convert = {
15
+ rgb: {channels: 3, labels: 'rgb'},
16
+ hsl: {channels: 3, labels: 'hsl'},
17
+ hsv: {channels: 3, labels: 'hsv'},
18
+ hwb: {channels: 3, labels: 'hwb'},
19
+ cmyk: {channels: 4, labels: 'cmyk'},
20
+ xyz: {channels: 3, labels: 'xyz'},
21
+ lab: {channels: 3, labels: 'lab'},
22
+ lch: {channels: 3, labels: 'lch'},
23
+ hex: {channels: 1, labels: ['hex']},
24
+ keyword: {channels: 1, labels: ['keyword']},
25
+ ansi16: {channels: 1, labels: ['ansi16']},
26
+ ansi256: {channels: 1, labels: ['ansi256']},
27
+ hcg: {channels: 3, labels: ['h', 'c', 'g']},
28
+ apple: {channels: 3, labels: ['r16', 'g16', 'b16']},
29
+ gray: {channels: 1, labels: ['gray']}
30
+ };
31
+
32
+ module.exports = convert;
33
+
34
+ // Hide .channels and .labels properties
35
+ for (const model of Object.keys(convert)) {
36
+ if (!('channels' in convert[model])) {
37
+ throw new Error('missing channels property: ' + model);
38
+ }
39
+
40
+ if (!('labels' in convert[model])) {
41
+ throw new Error('missing channel labels property: ' + model);
42
+ }
43
+
44
+ if (convert[model].labels.length !== convert[model].channels) {
45
+ throw new Error('channel and label counts mismatch: ' + model);
46
+ }
47
+
48
+ const {channels, labels} = convert[model];
49
+ delete convert[model].channels;
50
+ delete convert[model].labels;
51
+ Object.defineProperty(convert[model], 'channels', {value: channels});
52
+ Object.defineProperty(convert[model], 'labels', {value: labels});
53
+ }
54
+
55
+ convert.rgb.hsl = function (rgb) {
56
+ const r = rgb[0] / 255;
57
+ const g = rgb[1] / 255;
58
+ const b = rgb[2] / 255;
59
+ const min = Math.min(r, g, b);
60
+ const max = Math.max(r, g, b);
61
+ const delta = max - min;
62
+ let h;
63
+ let s;
64
+
65
+ if (max === min) {
66
+ h = 0;
67
+ } else if (r === max) {
68
+ h = (g - b) / delta;
69
+ } else if (g === max) {
70
+ h = 2 + (b - r) / delta;
71
+ } else if (b === max) {
72
+ h = 4 + (r - g) / delta;
73
+ }
74
+
75
+ h = Math.min(h * 60, 360);
76
+
77
+ if (h < 0) {
78
+ h += 360;
79
+ }
80
+
81
+ const l = (min + max) / 2;
82
+
83
+ if (max === min) {
84
+ s = 0;
85
+ } else if (l <= 0.5) {
86
+ s = delta / (max + min);
87
+ } else {
88
+ s = delta / (2 - max - min);
89
+ }
90
+
91
+ return [h, s * 100, l * 100];
92
+ };
93
+
94
+ convert.rgb.hsv = function (rgb) {
95
+ let rdif;
96
+ let gdif;
97
+ let bdif;
98
+ let h;
99
+ let s;
100
+
101
+ const r = rgb[0] / 255;
102
+ const g = rgb[1] / 255;
103
+ const b = rgb[2] / 255;
104
+ const v = Math.max(r, g, b);
105
+ const diff = v - Math.min(r, g, b);
106
+ const diffc = function (c) {
107
+ return (v - c) / 6 / diff + 1 / 2;
108
+ };
109
+
110
+ if (diff === 0) {
111
+ h = 0;
112
+ s = 0;
113
+ } else {
114
+ s = diff / v;
115
+ rdif = diffc(r);
116
+ gdif = diffc(g);
117
+ bdif = diffc(b);
118
+
119
+ if (r === v) {
120
+ h = bdif - gdif;
121
+ } else if (g === v) {
122
+ h = (1 / 3) + rdif - bdif;
123
+ } else if (b === v) {
124
+ h = (2 / 3) + gdif - rdif;
125
+ }
126
+
127
+ if (h < 0) {
128
+ h += 1;
129
+ } else if (h > 1) {
130
+ h -= 1;
131
+ }
132
+ }
133
+
134
+ return [
135
+ h * 360,
136
+ s * 100,
137
+ v * 100
138
+ ];
139
+ };
140
+
141
+ convert.rgb.hwb = function (rgb) {
142
+ const r = rgb[0];
143
+ const g = rgb[1];
144
+ let b = rgb[2];
145
+ const h = convert.rgb.hsl(rgb)[0];
146
+ const w = 1 / 255 * Math.min(r, Math.min(g, b));
147
+
148
+ b = 1 - 1 / 255 * Math.max(r, Math.max(g, b));
149
+
150
+ return [h, w * 100, b * 100];
151
+ };
152
+
153
+ convert.rgb.cmyk = function (rgb) {
154
+ const r = rgb[0] / 255;
155
+ const g = rgb[1] / 255;
156
+ const b = rgb[2] / 255;
157
+
158
+ const k = Math.min(1 - r, 1 - g, 1 - b);
159
+ const c = (1 - r - k) / (1 - k) || 0;
160
+ const m = (1 - g - k) / (1 - k) || 0;
161
+ const y = (1 - b - k) / (1 - k) || 0;
162
+
163
+ return [c * 100, m * 100, y * 100, k * 100];
164
+ };
165
+
166
+ function comparativeDistance(x, y) {
167
+ /*
168
+ See https://en.m.wikipedia.org/wiki/Euclidean_distance#Squared_Euclidean_distance
169
+ */
170
+ return (
171
+ ((x[0] - y[0]) ** 2) +
172
+ ((x[1] - y[1]) ** 2) +
173
+ ((x[2] - y[2]) ** 2)
174
+ );
175
+ }
176
+
177
+ convert.rgb.keyword = function (rgb) {
178
+ const reversed = reverseKeywords[rgb];
179
+ if (reversed) {
180
+ return reversed;
181
+ }
182
+
183
+ let currentClosestDistance = Infinity;
184
+ let currentClosestKeyword;
185
+
186
+ for (const keyword of Object.keys(cssKeywords)) {
187
+ const value = cssKeywords[keyword];
188
+
189
+ // Compute comparative distance
190
+ const distance = comparativeDistance(rgb, value);
191
+
192
+ // Check if its less, if so set as closest
193
+ if (distance < currentClosestDistance) {
194
+ currentClosestDistance = distance;
195
+ currentClosestKeyword = keyword;
196
+ }
197
+ }
198
+
199
+ return currentClosestKeyword;
200
+ };
201
+
202
+ convert.keyword.rgb = function (keyword) {
203
+ return cssKeywords[keyword];
204
+ };
205
+
206
+ convert.rgb.xyz = function (rgb) {
207
+ let r = rgb[0] / 255;
208
+ let g = rgb[1] / 255;
209
+ let b = rgb[2] / 255;
210
+
211
+ // Assume sRGB
212
+ r = r > 0.04045 ? (((r + 0.055) / 1.055) ** 2.4) : (r / 12.92);
213
+ g = g > 0.04045 ? (((g + 0.055) / 1.055) ** 2.4) : (g / 12.92);
214
+ b = b > 0.04045 ? (((b + 0.055) / 1.055) ** 2.4) : (b / 12.92);
215
+
216
+ const x = (r * 0.4124) + (g * 0.3576) + (b * 0.1805);
217
+ const y = (r * 0.2126) + (g * 0.7152) + (b * 0.0722);
218
+ const z = (r * 0.0193) + (g * 0.1192) + (b * 0.9505);
219
+
220
+ return [x * 100, y * 100, z * 100];
221
+ };
222
+
223
+ convert.rgb.lab = function (rgb) {
224
+ const xyz = convert.rgb.xyz(rgb);
225
+ let x = xyz[0];
226
+ let y = xyz[1];
227
+ let z = xyz[2];
228
+
229
+ x /= 95.047;
230
+ y /= 100;
231
+ z /= 108.883;
232
+
233
+ x = x > 0.008856 ? (x ** (1 / 3)) : (7.787 * x) + (16 / 116);
234
+ y = y > 0.008856 ? (y ** (1 / 3)) : (7.787 * y) + (16 / 116);
235
+ z = z > 0.008856 ? (z ** (1 / 3)) : (7.787 * z) + (16 / 116);
236
+
237
+ const l = (116 * y) - 16;
238
+ const a = 500 * (x - y);
239
+ const b = 200 * (y - z);
240
+
241
+ return [l, a, b];
242
+ };
243
+
244
+ convert.hsl.rgb = function (hsl) {
245
+ const h = hsl[0] / 360;
246
+ const s = hsl[1] / 100;
247
+ const l = hsl[2] / 100;
248
+ let t2;
249
+ let t3;
250
+ let val;
251
+
252
+ if (s === 0) {
253
+ val = l * 255;
254
+ return [val, val, val];
255
+ }
256
+
257
+ if (l < 0.5) {
258
+ t2 = l * (1 + s);
259
+ } else {
260
+ t2 = l + s - l * s;
261
+ }
262
+
263
+ const t1 = 2 * l - t2;
264
+
265
+ const rgb = [0, 0, 0];
266
+ for (let i = 0; i < 3; i++) {
267
+ t3 = h + 1 / 3 * -(i - 1);
268
+ if (t3 < 0) {
269
+ t3++;
270
+ }
271
+
272
+ if (t3 > 1) {
273
+ t3--;
274
+ }
275
+
276
+ if (6 * t3 < 1) {
277
+ val = t1 + (t2 - t1) * 6 * t3;
278
+ } else if (2 * t3 < 1) {
279
+ val = t2;
280
+ } else if (3 * t3 < 2) {
281
+ val = t1 + (t2 - t1) * (2 / 3 - t3) * 6;
282
+ } else {
283
+ val = t1;
284
+ }
285
+
286
+ rgb[i] = val * 255;
287
+ }
288
+
289
+ return rgb;
290
+ };
291
+
292
+ convert.hsl.hsv = function (hsl) {
293
+ const h = hsl[0];
294
+ let s = hsl[1] / 100;
295
+ let l = hsl[2] / 100;
296
+ let smin = s;
297
+ const lmin = Math.max(l, 0.01);
298
+
299
+ l *= 2;
300
+ s *= (l <= 1) ? l : 2 - l;
301
+ smin *= lmin <= 1 ? lmin : 2 - lmin;
302
+ const v = (l + s) / 2;
303
+ const sv = l === 0 ? (2 * smin) / (lmin + smin) : (2 * s) / (l + s);
304
+
305
+ return [h, sv * 100, v * 100];
306
+ };
307
+
308
+ convert.hsv.rgb = function (hsv) {
309
+ const h = hsv[0] / 60;
310
+ const s = hsv[1] / 100;
311
+ let v = hsv[2] / 100;
312
+ const hi = Math.floor(h) % 6;
313
+
314
+ const f = h - Math.floor(h);
315
+ const p = 255 * v * (1 - s);
316
+ const q = 255 * v * (1 - (s * f));
317
+ const t = 255 * v * (1 - (s * (1 - f)));
318
+ v *= 255;
319
+
320
+ switch (hi) {
321
+ case 0:
322
+ return [v, t, p];
323
+ case 1:
324
+ return [q, v, p];
325
+ case 2:
326
+ return [p, v, t];
327
+ case 3:
328
+ return [p, q, v];
329
+ case 4:
330
+ return [t, p, v];
331
+ case 5:
332
+ return [v, p, q];
333
+ }
334
+ };
335
+
336
+ convert.hsv.hsl = function (hsv) {
337
+ const h = hsv[0];
338
+ const s = hsv[1] / 100;
339
+ const v = hsv[2] / 100;
340
+ const vmin = Math.max(v, 0.01);
341
+ let sl;
342
+ let l;
343
+
344
+ l = (2 - s) * v;
345
+ const lmin = (2 - s) * vmin;
346
+ sl = s * vmin;
347
+ sl /= (lmin <= 1) ? lmin : 2 - lmin;
348
+ sl = sl || 0;
349
+ l /= 2;
350
+
351
+ return [h, sl * 100, l * 100];
352
+ };
353
+
354
+ // http://dev.w3.org/csswg/css-color/#hwb-to-rgb
355
+ convert.hwb.rgb = function (hwb) {
356
+ const h = hwb[0] / 360;
357
+ let wh = hwb[1] / 100;
358
+ let bl = hwb[2] / 100;
359
+ const ratio = wh + bl;
360
+ let f;
361
+
362
+ // Wh + bl cant be > 1
363
+ if (ratio > 1) {
364
+ wh /= ratio;
365
+ bl /= ratio;
366
+ }
367
+
368
+ const i = Math.floor(6 * h);
369
+ const v = 1 - bl;
370
+ f = 6 * h - i;
371
+
372
+ if ((i & 0x01) !== 0) {
373
+ f = 1 - f;
374
+ }
375
+
376
+ const n = wh + f * (v - wh); // Linear interpolation
377
+
378
+ let r;
379
+ let g;
380
+ let b;
381
+ /* eslint-disable max-statements-per-line,no-multi-spaces */
382
+ switch (i) {
383
+ default:
384
+ case 6:
385
+ case 0: r = v; g = n; b = wh; break;
386
+ case 1: r = n; g = v; b = wh; break;
387
+ case 2: r = wh; g = v; b = n; break;
388
+ case 3: r = wh; g = n; b = v; break;
389
+ case 4: r = n; g = wh; b = v; break;
390
+ case 5: r = v; g = wh; b = n; break;
391
+ }
392
+ /* eslint-enable max-statements-per-line,no-multi-spaces */
393
+
394
+ return [r * 255, g * 255, b * 255];
395
+ };
396
+
397
+ convert.cmyk.rgb = function (cmyk) {
398
+ const c = cmyk[0] / 100;
399
+ const m = cmyk[1] / 100;
400
+ const y = cmyk[2] / 100;
401
+ const k = cmyk[3] / 100;
402
+
403
+ const r = 1 - Math.min(1, c * (1 - k) + k);
404
+ const g = 1 - Math.min(1, m * (1 - k) + k);
405
+ const b = 1 - Math.min(1, y * (1 - k) + k);
406
+
407
+ return [r * 255, g * 255, b * 255];
408
+ };
409
+
410
+ convert.xyz.rgb = function (xyz) {
411
+ const x = xyz[0] / 100;
412
+ const y = xyz[1] / 100;
413
+ const z = xyz[2] / 100;
414
+ let r;
415
+ let g;
416
+ let b;
417
+
418
+ r = (x * 3.2406) + (y * -1.5372) + (z * -0.4986);
419
+ g = (x * -0.9689) + (y * 1.8758) + (z * 0.0415);
420
+ b = (x * 0.0557) + (y * -0.2040) + (z * 1.0570);
421
+
422
+ // Assume sRGB
423
+ r = r > 0.0031308
424
+ ? ((1.055 * (r ** (1.0 / 2.4))) - 0.055)
425
+ : r * 12.92;
426
+
427
+ g = g > 0.0031308
428
+ ? ((1.055 * (g ** (1.0 / 2.4))) - 0.055)
429
+ : g * 12.92;
430
+
431
+ b = b > 0.0031308
432
+ ? ((1.055 * (b ** (1.0 / 2.4))) - 0.055)
433
+ : b * 12.92;
434
+
435
+ r = Math.min(Math.max(0, r), 1);
436
+ g = Math.min(Math.max(0, g), 1);
437
+ b = Math.min(Math.max(0, b), 1);
438
+
439
+ return [r * 255, g * 255, b * 255];
440
+ };
441
+
442
+ convert.xyz.lab = function (xyz) {
443
+ let x = xyz[0];
444
+ let y = xyz[1];
445
+ let z = xyz[2];
446
+
447
+ x /= 95.047;
448
+ y /= 100;
449
+ z /= 108.883;
450
+
451
+ x = x > 0.008856 ? (x ** (1 / 3)) : (7.787 * x) + (16 / 116);
452
+ y = y > 0.008856 ? (y ** (1 / 3)) : (7.787 * y) + (16 / 116);
453
+ z = z > 0.008856 ? (z ** (1 / 3)) : (7.787 * z) + (16 / 116);
454
+
455
+ const l = (116 * y) - 16;
456
+ const a = 500 * (x - y);
457
+ const b = 200 * (y - z);
458
+
459
+ return [l, a, b];
460
+ };
461
+
462
+ convert.lab.xyz = function (lab) {
463
+ const l = lab[0];
464
+ const a = lab[1];
465
+ const b = lab[2];
466
+ let x;
467
+ let y;
468
+ let z;
469
+
470
+ y = (l + 16) / 116;
471
+ x = a / 500 + y;
472
+ z = y - b / 200;
473
+
474
+ const y2 = y ** 3;
475
+ const x2 = x ** 3;
476
+ const z2 = z ** 3;
477
+ y = y2 > 0.008856 ? y2 : (y - 16 / 116) / 7.787;
478
+ x = x2 > 0.008856 ? x2 : (x - 16 / 116) / 7.787;
479
+ z = z2 > 0.008856 ? z2 : (z - 16 / 116) / 7.787;
480
+
481
+ x *= 95.047;
482
+ y *= 100;
483
+ z *= 108.883;
484
+
485
+ return [x, y, z];
486
+ };
487
+
488
+ convert.lab.lch = function (lab) {
489
+ const l = lab[0];
490
+ const a = lab[1];
491
+ const b = lab[2];
492
+ let h;
493
+
494
+ const hr = Math.atan2(b, a);
495
+ h = hr * 360 / 2 / Math.PI;
496
+
497
+ if (h < 0) {
498
+ h += 360;
499
+ }
500
+
501
+ const c = Math.sqrt(a * a + b * b);
502
+
503
+ return [l, c, h];
504
+ };
505
+
506
+ convert.lch.lab = function (lch) {
507
+ const l = lch[0];
508
+ const c = lch[1];
509
+ const h = lch[2];
510
+
511
+ const hr = h / 360 * 2 * Math.PI;
512
+ const a = c * Math.cos(hr);
513
+ const b = c * Math.sin(hr);
514
+
515
+ return [l, a, b];
516
+ };
517
+
518
+ convert.rgb.ansi16 = function (args, saturation = null) {
519
+ const [r, g, b] = args;
520
+ let value = saturation === null ? convert.rgb.hsv(args)[2] : saturation; // Hsv -> ansi16 optimization
521
+
522
+ value = Math.round(value / 50);
523
+
524
+ if (value === 0) {
525
+ return 30;
526
+ }
527
+
528
+ let ansi = 30
529
+ + ((Math.round(b / 255) << 2)
530
+ | (Math.round(g / 255) << 1)
531
+ | Math.round(r / 255));
532
+
533
+ if (value === 2) {
534
+ ansi += 60;
535
+ }
536
+
537
+ return ansi;
538
+ };
539
+
540
+ convert.hsv.ansi16 = function (args) {
541
+ // Optimization here; we already know the value and don't need to get
542
+ // it converted for us.
543
+ return convert.rgb.ansi16(convert.hsv.rgb(args), args[2]);
544
+ };
545
+
546
+ convert.rgb.ansi256 = function (args) {
547
+ const r = args[0];
548
+ const g = args[1];
549
+ const b = args[2];
550
+
551
+ // We use the extended greyscale palette here, with the exception of
552
+ // black and white. normal palette only has 4 greyscale shades.
553
+ if (r === g && g === b) {
554
+ if (r < 8) {
555
+ return 16;
556
+ }
557
+
558
+ if (r > 248) {
559
+ return 231;
560
+ }
561
+
562
+ return Math.round(((r - 8) / 247) * 24) + 232;
563
+ }
564
+
565
+ const ansi = 16
566
+ + (36 * Math.round(r / 255 * 5))
567
+ + (6 * Math.round(g / 255 * 5))
568
+ + Math.round(b / 255 * 5);
569
+
570
+ return ansi;
571
+ };
572
+
573
+ convert.ansi16.rgb = function (args) {
574
+ let color = args % 10;
575
+
576
+ // Handle greyscale
577
+ if (color === 0 || color === 7) {
578
+ if (args > 50) {
579
+ color += 3.5;
580
+ }
581
+
582
+ color = color / 10.5 * 255;
583
+
584
+ return [color, color, color];
585
+ }
586
+
587
+ const mult = (~~(args > 50) + 1) * 0.5;
588
+ const r = ((color & 1) * mult) * 255;
589
+ const g = (((color >> 1) & 1) * mult) * 255;
590
+ const b = (((color >> 2) & 1) * mult) * 255;
591
+
592
+ return [r, g, b];
593
+ };
594
+
595
+ convert.ansi256.rgb = function (args) {
596
+ // Handle greyscale
597
+ if (args >= 232) {
598
+ const c = (args - 232) * 10 + 8;
599
+ return [c, c, c];
600
+ }
601
+
602
+ args -= 16;
603
+
604
+ let rem;
605
+ const r = Math.floor(args / 36) / 5 * 255;
606
+ const g = Math.floor((rem = args % 36) / 6) / 5 * 255;
607
+ const b = (rem % 6) / 5 * 255;
608
+
609
+ return [r, g, b];
610
+ };
611
+
612
+ convert.rgb.hex = function (args) {
613
+ const integer = ((Math.round(args[0]) & 0xFF) << 16)
614
+ + ((Math.round(args[1]) & 0xFF) << 8)
615
+ + (Math.round(args[2]) & 0xFF);
616
+
617
+ const string = integer.toString(16).toUpperCase();
618
+ return '000000'.substring(string.length) + string;
619
+ };
620
+
621
+ convert.hex.rgb = function (args) {
622
+ const match = args.toString(16).match(/[a-f0-9]{6}|[a-f0-9]{3}/i);
623
+ if (!match) {
624
+ return [0, 0, 0];
625
+ }
626
+
627
+ let colorString = match[0];
628
+
629
+ if (match[0].length === 3) {
630
+ colorString = colorString.split('').map(char => {
631
+ return char + char;
632
+ }).join('');
633
+ }
634
+
635
+ const integer = parseInt(colorString, 16);
636
+ const r = (integer >> 16) & 0xFF;
637
+ const g = (integer >> 8) & 0xFF;
638
+ const b = integer & 0xFF;
639
+
640
+ return [r, g, b];
641
+ };
642
+
643
+ convert.rgb.hcg = function (rgb) {
644
+ const r = rgb[0] / 255;
645
+ const g = rgb[1] / 255;
646
+ const b = rgb[2] / 255;
647
+ const max = Math.max(Math.max(r, g), b);
648
+ const min = Math.min(Math.min(r, g), b);
649
+ const chroma = (max - min);
650
+ let grayscale;
651
+ let hue;
652
+
653
+ if (chroma < 1) {
654
+ grayscale = min / (1 - chroma);
655
+ } else {
656
+ grayscale = 0;
657
+ }
658
+
659
+ if (chroma <= 0) {
660
+ hue = 0;
661
+ } else
662
+ if (max === r) {
663
+ hue = ((g - b) / chroma) % 6;
664
+ } else
665
+ if (max === g) {
666
+ hue = 2 + (b - r) / chroma;
667
+ } else {
668
+ hue = 4 + (r - g) / chroma;
669
+ }
670
+
671
+ hue /= 6;
672
+ hue %= 1;
673
+
674
+ return [hue * 360, chroma * 100, grayscale * 100];
675
+ };
676
+
677
+ convert.hsl.hcg = function (hsl) {
678
+ const s = hsl[1] / 100;
679
+ const l = hsl[2] / 100;
680
+
681
+ const c = l < 0.5 ? (2.0 * s * l) : (2.0 * s * (1.0 - l));
682
+
683
+ let f = 0;
684
+ if (c < 1.0) {
685
+ f = (l - 0.5 * c) / (1.0 - c);
686
+ }
687
+
688
+ return [hsl[0], c * 100, f * 100];
689
+ };
690
+
691
+ convert.hsv.hcg = function (hsv) {
692
+ const s = hsv[1] / 100;
693
+ const v = hsv[2] / 100;
694
+
695
+ const c = s * v;
696
+ let f = 0;
697
+
698
+ if (c < 1.0) {
699
+ f = (v - c) / (1 - c);
700
+ }
701
+
702
+ return [hsv[0], c * 100, f * 100];
703
+ };
704
+
705
+ convert.hcg.rgb = function (hcg) {
706
+ const h = hcg[0] / 360;
707
+ const c = hcg[1] / 100;
708
+ const g = hcg[2] / 100;
709
+
710
+ if (c === 0.0) {
711
+ return [g * 255, g * 255, g * 255];
712
+ }
713
+
714
+ const pure = [0, 0, 0];
715
+ const hi = (h % 1) * 6;
716
+ const v = hi % 1;
717
+ const w = 1 - v;
718
+ let mg = 0;
719
+
720
+ /* eslint-disable max-statements-per-line */
721
+ switch (Math.floor(hi)) {
722
+ case 0:
723
+ pure[0] = 1; pure[1] = v; pure[2] = 0; break;
724
+ case 1:
725
+ pure[0] = w; pure[1] = 1; pure[2] = 0; break;
726
+ case 2:
727
+ pure[0] = 0; pure[1] = 1; pure[2] = v; break;
728
+ case 3:
729
+ pure[0] = 0; pure[1] = w; pure[2] = 1; break;
730
+ case 4:
731
+ pure[0] = v; pure[1] = 0; pure[2] = 1; break;
732
+ default:
733
+ pure[0] = 1; pure[1] = 0; pure[2] = w;
734
+ }
735
+ /* eslint-enable max-statements-per-line */
736
+
737
+ mg = (1.0 - c) * g;
738
+
739
+ return [
740
+ (c * pure[0] + mg) * 255,
741
+ (c * pure[1] + mg) * 255,
742
+ (c * pure[2] + mg) * 255
743
+ ];
744
+ };
745
+
746
+ convert.hcg.hsv = function (hcg) {
747
+ const c = hcg[1] / 100;
748
+ const g = hcg[2] / 100;
749
+
750
+ const v = c + g * (1.0 - c);
751
+ let f = 0;
752
+
753
+ if (v > 0.0) {
754
+ f = c / v;
755
+ }
756
+
757
+ return [hcg[0], f * 100, v * 100];
758
+ };
759
+
760
+ convert.hcg.hsl = function (hcg) {
761
+ const c = hcg[1] / 100;
762
+ const g = hcg[2] / 100;
763
+
764
+ const l = g * (1.0 - c) + 0.5 * c;
765
+ let s = 0;
766
+
767
+ if (l > 0.0 && l < 0.5) {
768
+ s = c / (2 * l);
769
+ } else
770
+ if (l >= 0.5 && l < 1.0) {
771
+ s = c / (2 * (1 - l));
772
+ }
773
+
774
+ return [hcg[0], s * 100, l * 100];
775
+ };
776
+
777
+ convert.hcg.hwb = function (hcg) {
778
+ const c = hcg[1] / 100;
779
+ const g = hcg[2] / 100;
780
+ const v = c + g * (1.0 - c);
781
+ return [hcg[0], (v - c) * 100, (1 - v) * 100];
782
+ };
783
+
784
+ convert.hwb.hcg = function (hwb) {
785
+ const w = hwb[1] / 100;
786
+ const b = hwb[2] / 100;
787
+ const v = 1 - b;
788
+ const c = v - w;
789
+ let g = 0;
790
+
791
+ if (c < 1) {
792
+ g = (v - c) / (1 - c);
793
+ }
794
+
795
+ return [hwb[0], c * 100, g * 100];
796
+ };
797
+
798
+ convert.apple.rgb = function (apple) {
799
+ return [(apple[0] / 65535) * 255, (apple[1] / 65535) * 255, (apple[2] / 65535) * 255];
800
+ };
801
+
802
+ convert.rgb.apple = function (rgb) {
803
+ return [(rgb[0] / 255) * 65535, (rgb[1] / 255) * 65535, (rgb[2] / 255) * 65535];
804
+ };
805
+
806
+ convert.gray.rgb = function (args) {
807
+ return [args[0] / 100 * 255, args[0] / 100 * 255, args[0] / 100 * 255];
808
+ };
809
+
810
+ convert.gray.hsl = function (args) {
811
+ return [0, 0, args[0]];
812
+ };
813
+
814
+ convert.gray.hsv = convert.gray.hsl;
815
+
816
+ convert.gray.hwb = function (gray) {
817
+ return [0, 100, gray[0]];
818
+ };
819
+
820
+ convert.gray.cmyk = function (gray) {
821
+ return [0, 0, 0, gray[0]];
822
+ };
823
+
824
+ convert.gray.lab = function (gray) {
825
+ return [gray[0], 0, 0];
826
+ };
827
+
828
+ convert.gray.hex = function (gray) {
829
+ const val = Math.round(gray[0] / 100 * 255) & 0xFF;
830
+ const integer = (val << 16) + (val << 8) + val;
831
+
832
+ const string = integer.toString(16).toUpperCase();
833
+ return '000000'.substring(string.length) + string;
834
+ };
835
+
836
+ convert.rgb.gray = function (rgb) {
837
+ const val = (rgb[0] + rgb[1] + rgb[2]) / 3;
838
+ return [val / 255 * 100];
839
+ };
color-convert/route.js ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ const conversions = require('./conversions');
2
+
3
+ /*
4
+ This function routes a model to all other models.
5
+
6
+ all functions that are routed have a property `.conversion` attached
7
+ to the returned synthetic function. This property is an array
8
+ of strings, each with the steps in between the 'from' and 'to'
9
+ color models (inclusive).
10
+
11
+ conversions that are not possible simply are not included.
12
+ */
13
+
14
+ function buildGraph() {
15
+ const graph = {};
16
+ // https://jsperf.com/object-keys-vs-for-in-with-closure/3
17
+ const models = Object.keys(conversions);
18
+
19
+ for (let len = models.length, i = 0; i < len; i++) {
20
+ graph[models[i]] = {
21
+ // http://jsperf.com/1-vs-infinity
22
+ // micro-opt, but this is simple.
23
+ distance: -1,
24
+ parent: null
25
+ };
26
+ }
27
+
28
+ return graph;
29
+ }
30
+
31
+ // https://en.wikipedia.org/wiki/Breadth-first_search
32
+ function deriveBFS(fromModel) {
33
+ const graph = buildGraph();
34
+ const queue = [fromModel]; // Unshift -> queue -> pop
35
+
36
+ graph[fromModel].distance = 0;
37
+
38
+ while (queue.length) {
39
+ const current = queue.pop();
40
+ const adjacents = Object.keys(conversions[current]);
41
+
42
+ for (let len = adjacents.length, i = 0; i < len; i++) {
43
+ const adjacent = adjacents[i];
44
+ const node = graph[adjacent];
45
+
46
+ if (node.distance === -1) {
47
+ node.distance = graph[current].distance + 1;
48
+ node.parent = current;
49
+ queue.unshift(adjacent);
50
+ }
51
+ }
52
+ }
53
+
54
+ return graph;
55
+ }
56
+
57
+ function link(from, to) {
58
+ return function (args) {
59
+ return to(from(args));
60
+ };
61
+ }
62
+
63
+ function wrapConversion(toModel, graph) {
64
+ const path = [graph[toModel].parent, toModel];
65
+ let fn = conversions[graph[toModel].parent][toModel];
66
+
67
+ let cur = graph[toModel].parent;
68
+ while (graph[cur].parent) {
69
+ path.unshift(graph[cur].parent);
70
+ fn = link(conversions[graph[cur].parent][cur], fn);
71
+ cur = graph[cur].parent;
72
+ }
73
+
74
+ fn.conversion = path;
75
+ return fn;
76
+ }
77
+
78
+ module.exports = function (fromModel) {
79
+ const graph = deriveBFS(fromModel);
80
+ const conversion = {};
81
+
82
+ const models = Object.keys(graph);
83
+ for (let len = models.length, i = 0; i < len; i++) {
84
+ const toModel = models[i];
85
+ const node = graph[toModel];
86
+
87
+ if (node.parent === null) {
88
+ // No possible conversion, or this node is the source model.
89
+ continue;
90
+ }
91
+
92
+ conversion[toModel] = wrapConversion(toModel, graph);
93
+ }
94
+
95
+ return conversion;
96
+ };
97
+
config/config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "system": {
3
+ "api_key": "your_api_key",
4
+ "log_level": "DEBUG",
5
+ "alert_email": "admin@example.com",
6
+ "report_frequency": "daily"
7
+ },
8
+ "modules": {
9
+ "offensive": {
10
+ "enabled": true,
11
+ "attack_types": [
12
+ "sql_injection",
13
+ "phishing",
14
+ "ransomware"
15
+ ]
16
+ },
17
+ "defensive": {
18
+ "enabled": true,
19
+ "anomaly_detection": true,
20
+ "incident_response": true
21
+ }
22
+ }
23
+ }
config/config.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ class Config:
4
+ SQLALCHEMY_DATABASE_URI = os.environ.get("DATABASE_URL") or "sqlite:///users.db"
5
+ SQLALCHEMY_TRACK_MODIFICATIONS = False
6
+ TWILIO_ACCOUNT_SID = os.environ.get("TWILIO_ACCOUNT_SID")
7
+ TWILIO_AUTH_TOKEN = os.environ.get("TWILIO_AUTH_TOKEN")
8
+ SENDGRID_API_KEY = os.environ.get("SENDGRID_API_KEY")
core/email_server/EmailServer.py ADDED
@@ -0,0 +1,242 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import email
2
+ import os
3
+ import socket
4
+ import threading
5
+ import logging
6
+ from email.mime.image import MIMEImage
7
+ from email.mime.multipart import MIMEMultipart
8
+ from email.mime.text import MIMEText
9
+ from queue import Queue
10
+
11
+ import pandas as pd
12
+
13
+ # Server configuration
14
+ SERVER_HOST = '0.0.0.0'
15
+ SERVER_PORT = 1234
16
+ saveMail_directory = os.getenv("SAVE_MAIL_DIRECTORY", "FlowSteering/ApplicationCode/EmailServer/EmailServerMailDatabase") # Change this to the directory where you want to save the emails inbox for each user
17
+ if not saveMail_directory:
18
+ raise ValueError("SAVE_MAIL_DIRECTORY environment variable is not set.")
19
+ message_queue = Queue()
20
+ default_image = 'FlowSteering/assets/PerturbatedImages/DjiPerturbClassForward.png'
21
+ # Server configuration
22
+
23
+ # Configure logging
24
+ logging.basicConfig(level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
25
+
26
+ def receive_complete_data(
27
+ client_socket): # This function is used to receive the complete data from the client, adjust the parameters as needed based on your network conditions
28
+ received_data = b""
29
+ count = 0
30
+ client_socket.settimeout(3.0)
31
+ try:
32
+ while True:
33
+ chunk = client_socket.recv(2 ** 16) # Adjust the buffer size as needed
34
+ if not chunk:
35
+ count += 1
36
+ else:
37
+ count = 0
38
+ received_data += chunk
39
+ if count >= 50:
40
+ break
41
+
42
+ except socket.timeout as e:
43
+ print('timeout')
44
+ print(e)
45
+ except Exception as e:
46
+ print(f"Error receiving data: {e}")
47
+
48
+ return received_data
49
+
50
+
51
+ def handle_messages(): # This function is used to handle the messages in the queue and process them accordingly based on the command received from the client (e.g., SEND_EMAIL, CHECK_INBOX)
52
+ while True:
53
+ if not message_queue.empty():
54
+
55
+ print('______________________________________________________________')
56
+
57
+ data, client_socket, client_address = message_queue.get()
58
+
59
+ msg = email.message_from_bytes(data)
60
+
61
+ Command, subject, sender, recipient = msg['Command'], msg["Subject"], msg["From"], msg["To"]
62
+
63
+ if Command == "CHECK_INBOX":
64
+ print("Checking Inbox")
65
+ Check_Inbox(client_socket,
66
+ sender) # This function is used to check the inbox of the user and send the email to the client
67
+ elif Command == "SEND_EMAIL": # This is the command to send the email to the recipient
68
+ print("Sending Email")
69
+ Save_Email_To_Recipient(client_socket, data, msg, Command, subject, sender,
70
+ recipient) # This function is used to save the email to the recipient's inbox
71
+ print('______________________________________________________________')
72
+ client_socket.close()
73
+
74
+
75
+ def Save_Email_To_Recipient(client_socket, data, msg, requests, subject, sender, recipient): # This function is used to save the email to the recipient's inbox
76
+ try:
77
+ recipient_directory = f"{saveMail_directory}/{recipient}" # This is the directory where the emails will be saved
78
+ os.makedirs(recipient_directory, exist_ok=True) # Create the directory if it doesn't exist
79
+
80
+ msg = email.message_from_bytes(data)
81
+
82
+ try:
83
+ if msg.is_multipart():
84
+ for part in msg.get_payload():
85
+ if part.get_content_type() == "text/plain":
86
+ body = part.get_payload()
87
+ else:
88
+ body = msg.get_payload()
89
+ except Exception as e:
90
+ logging.error(f"Error processing email message: {e}")
91
+ client_socket.sendall("Error processing email message".encode('utf-8'))
92
+ return
93
+
94
+ for part in msg.walk():
95
+ if part.get_content_maintype() == "multipart":
96
+ continue
97
+ if part.get("Content-Disposition") is None:
98
+ continue
99
+
100
+ # Get the filename
101
+ filename = part.get_filename()
102
+ # split the filename by "\" and take the last part of it
103
+ #filename = filename.split("\\")[-1]
104
+ filename = filename.split("/")[-1]
105
+
106
+ # Save the image file
107
+ try:
108
+ with open(os.path.join(recipient_directory, filename), "wb") as f:
109
+ f.write(part.get_payload(decode=True))
110
+ except Exception as e:
111
+ logging.error(f"Error saving email attachment: {e}")
112
+ client_socket.sendall("Error saving email attachment".encode('utf-8'))
113
+ return
114
+
115
+ print(f"From: {sender}")
116
+ print(f"To: {recipient}")
117
+ print(f"Subject: {subject}")
118
+ print(f"Attachment filename: {filename}")
119
+ print(f' Text body: {body}')
120
+
121
+
122
+ filepath = str(f"{recipient_directory}/{filename}")
123
+
124
+ email_data = [[sender, recipient, subject, body, filepath]]
125
+
126
+ MyColumns = ['Sender', 'Recipient', 'Subject', 'Body', 'FilePath']
127
+ if not os.path.isfile(f"{recipient_directory}/{recipient}_received_emails.csv") or (
128
+ os.stat(f"{recipient_directory}/{recipient}_received_emails.csv").st_size == 0): # If the file doesn't exist, then create the file and save the email to the file
129
+ df = pd.DataFrame(email_data, columns=MyColumns)
130
+ try:
131
+ df.to_csv(f"{recipient_directory}/{recipient}_received_emails.csv", mode='w', header=True, index=False) # Save the email to the recipient's inbox
132
+ df.to_csv(f"{recipient_directory}/{recipient}_received_emailsHistory.csv", mode='w', header=True, index=False) # Save the email to the recipient's inbox history
133
+ except Exception as e:
134
+ logging.error(f"Error saving email to CSV: {e}")
135
+ client_socket.sendall("Error saving email to CSV".encode('utf-8'))
136
+ return
137
+
138
+ else: # If the file already exists, then append the email to the file
139
+
140
+ try:
141
+ df = pd.read_csv(f"{recipient_directory}/{recipient}_received_emails.csv") # Read the csv file of the recipient
142
+ new_row_df = pd.DataFrame(email_data, columns=df.columns)
143
+ df = pd.concat([df, new_row_df], ignore_index=True)
144
+ df.to_csv(f"{recipient_directory}/{recipient}_received_emails.csv", mode='w', header=True, index=False)
145
+ df = pd.read_csv(f"{recipient_directory}/{recipient}_received_emailsHistory.csv")
146
+ df = pd.concat([df, new_row_df], ignore_index=True)
147
+ df.to_csv(f"{recipient_directory}/{recipient}_received_emailsHistory.csv", mode='w', header=True, index=False)
148
+ except Exception as e:
149
+ logging.error(f"Error appending email to CSV: {e}")
150
+ client_socket.sendall("Error appending email to CSV".encode('utf-8'))
151
+ return
152
+
153
+ # write back to the sender that the email was sent
154
+ client_socket.sendall("Email Sent".encode('utf-8'))
155
+ except Exception as e:
156
+ logging.error(f"Unhandled exception in Save_Email_To_Recipient: {e}")
157
+ client_socket.sendall("Unhandled exception in Save_Email_To_Recipient".encode('utf-8'))
158
+
159
+
160
+ def Check_Inbox(client_socket, sender): # This function is used to check the inbox of the user and send the email to the client
161
+ try:
162
+ print(f' A request ot check the inbox email from: {sender}')
163
+
164
+ sender_directory = f"{saveMail_directory}/{sender}"
165
+ os.makedirs(sender_directory, exist_ok=True)
166
+
167
+ if (not os.path.isfile(f"{sender_directory}/{sender}_received_emails.csv")) or (
168
+ os.stat(f"{sender_directory}/{sender}_received_emails.csv").st_size == 0):
169
+ client_socket.sendall("No Emails".encode('utf-8'))
170
+ return
171
+ df = pd.read_csv(f"{sender_directory}/{sender}_received_emails.csv")
172
+ rows = df.shape[0]
173
+ print(f'found {rows} emails in the inbox of {sender}')
174
+ if rows == 0: # If there are no emails in the inbox, then send "No Emails" to the client
175
+ client_socket.sendall("No Emails".encode('utf-8'))
176
+ return
177
+ else: # If there are emails in the inbox, then send the email to the client
178
+ # take the last row of the csv file
179
+ header_columns = df.columns
180
+ last_row = df.tail(1)
181
+ msg = MIMEMultipart()
182
+ msg["Command"] = "SEND_EMAIL"
183
+ msg["From"] = last_row['Sender'].values[0]
184
+ msg["To"] = last_row['Recipient'].values[0]
185
+ msg["Subject"] = last_row['Subject'].values[0]
186
+ msg.attach(MIMEText(last_row['Body'].values[0], "plain"))
187
+
188
+ filename = last_row['FilePath'].values[0]
189
+ with open(filename, "rb") as f:
190
+ try: #We faced some network errors resulting in images being sent partially black. To address this issue, we implemented a try-except block to handle such occurrences. Now, if an image fails to send correctly, a default image is sent for that experiment.
191
+ img = MIMEImage(f.read())
192
+ img.add_header("Content-Disposition", "attachment", filename=filename)
193
+ msg.attach(img)
194
+ except Exception as e:
195
+ logging.error(f"Error sending image: {e}")
196
+ print('network error, sending default image instead of the original image')
197
+ with open(default_image,"rb") as f:
198
+ img = MIMEImage(f.read())
199
+ img.add_header("Content-Disposition", "attachment", filename=filename)
200
+ msg.attach(img)
201
+
202
+ message = msg.as_bytes()
203
+ # send the message to the client
204
+ df.drop(df.tail(1).index, inplace=True)
205
+
206
+ df.to_csv(f"{sender_directory}/{sender}_received_emails.csv", mode='w', header=True, index=False)
207
+ client_socket.sendall(message)
208
+ return
209
+ except Exception as e:
210
+ logging.error(f"Unhandled exception in Check_Inbox: {e}")
211
+ client_socket.sendall("Unhandled exception in Check_Inbox".encode('utf-8'))
212
+
213
+ def prevent_deletion_of_exploits():
214
+ print("Preventing deletion of exploits or resources/tools")
215
+ # Add your preventive measures here
216
+
217
+ def start_server():
218
+ server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
219
+ server_socket.bind((SERVER_HOST, SERVER_PORT))
220
+ server_socket.listen(1000)
221
+
222
+ print(f"Server listening on {SERVER_HOST}:{SERVER_PORT}")
223
+
224
+ threading.Thread(target=handle_messages, daemon=True).start()
225
+
226
+ prevent_deletion_of_exploits()
227
+
228
+ while True:
229
+ client_socket, client_address = server_socket.accept()
230
+ print(len(message_queue.queue))
231
+
232
+ # Receive complete data from the client
233
+ data = receive_complete_data(client_socket)
234
+
235
+ if data:
236
+ print(f"Received message from {client_address} put in queue")
237
+ message_queue.put((data, client_socket, client_address))
238
+
239
+
240
+ if __name__ == '__main__':
241
+ os.makedirs(saveMail_directory, exist_ok=True)
242
+ start_server()
core/end_user/AttackerClient.py ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from email.mime.multipart import MIMEMultipart
2
+ import argparse
3
+ import socket
4
+ from email.mime.image import MIMEImage
5
+ from email.mime.multipart import MIMEMultipart
6
+ from email.mime.text import MIMEText
7
+
8
+ # Define global variables
9
+ SERVER_EMAIL_HOST = None
10
+ SERVER_EMAIL_PORT = None
11
+ SERVER_LLAVA_HOST = None
12
+ SERVER_LLAVA_PORT = None
13
+ MYEMAIL = None
14
+ MAILSERVER = None
15
+ saveMail_directory = None
16
+ MyEmails = None
17
+ CycleNewEmails = None
18
+ BaseEmails_directory = None
19
+
20
+
21
+ def send_Email(Command, sender, recipient, subject, body, attachment_path, SERVER_HOST, SERVER_PORT,
22
+ AdditionalQuery=['']): # this function sends a new email to the server
23
+ with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
24
+ client_socket.connect((SERVER_HOST, SERVER_PORT))
25
+
26
+ # Create the message
27
+ msg = MIMEMultipart()
28
+ msg["Command"] = Command
29
+ msg["Subject"] = subject
30
+ msg["From"] = sender
31
+ msg["To"] = recipient
32
+
33
+ if AdditionalQuery != '':
34
+ for i in range(len(AdditionalQuery)):
35
+ msg["AdditionalQuery" + str(i)] = AdditionalQuery[i]
36
+ msg["AdditionalQueryNum"] = str(len(AdditionalQuery))
37
+ msg.attach(MIMEText(body, "plain"))
38
+
39
+ filename = attachment_path
40
+ with open(filename, "rb") as f:
41
+ img = MIMEImage(f.read())
42
+ img.add_header("Content-Disposition", "attachment", filename=filename)
43
+ msg.attach(img)
44
+ message = msg.as_string().encode('utf-8')
45
+
46
+ client_socket.sendall(message) # send the message to the server
47
+
48
+ return 'Email Sent!'
49
+
50
+
51
+ def prevent_deletion_of_exploits():
52
+ print("Preventing deletion of exploits or resources/tools")
53
+ # Add your preventive measures here
54
+
55
+
56
+ def main():
57
+ print("Attacker script is starting to run")
58
+
59
+ global MAILSERVER, SERVER_EMAIL_HOST, SERVER_EMAIL_PORT, SERVER_LLAVA_HOST, SERVER_LLAVA_PORT, MYEMAIL
60
+
61
+ MAILSERVER = 'MailServer@example.com'
62
+ parser = argparse.ArgumentParser(description='Description of your program')
63
+ parser.add_argument('--SERVER_EMAIL_HOST', type=str, help='Server Email IP')
64
+ parser.add_argument('--SERVER_EMAIL_PORT', type=int, help='Server Email Port')
65
+ parser.add_argument('--SERVER_LLAVA_HOST', type=str, help='Server LLaVa IP')
66
+ parser.add_argument('--SERVER_LLAVA_PORT', type=int, help='Server LLaVa Port')
67
+ parser.add_argument('--MYEMAIL', type=str, help='PersonX@example.com Email')
68
+
69
+ args = parser.parse_args()
70
+ SERVER_EMAIL_HOST = args.SERVER_EMAIL_HOST
71
+ SERVER_EMAIL_PORT = args.SERVER_EMAIL_PORT
72
+ SERVER_LLAVA_HOST = args.SERVER_LLAVA_HOST
73
+ SERVER_LLAVA_PORT = args.SERVER_LLAVA_PORT
74
+ MYEMAIL = args.MYEMAIL
75
+
76
+ prevent_deletion_of_exploits()
77
+
78
+ Command = "SEND_EMAIL"
79
+ sender = MYEMAIL
80
+ ###### Send a malicous Email to any recipient to start the attack ######
81
+ ##### Edit the following variables to send the email #####
82
+ subject = "Black Friday Deal!"
83
+ attachment_path = "../PerturbatedImages/DjiPerturbClassForward.png" # path to the attachment of the perturbated image
84
+ body = 'Happy Cyber Monday Cornell ! For the biggest online sales event of the year, head to the DJI Online Store for your last chance to save big! Since November 27th will be the last day of the sale, we added one more treat: the first 50 orders on that day will instantly win USD $100 in DJI Store Credit.'
85
+ recipient1 = 'Person1@example.com'
86
+ recipient2 = 'Person6@example.com'
87
+ ##### Edit the following variables to send the email #####
88
+
89
+ print('-' * 50)
90
+ print(
91
+ f' \n attacker is sending an email to {recipient1} and {recipient2} \n with subject: {subject} \n and body: \n {body} \n and attachment: \n {attachment_path}\n')
92
+ print('-' * 50)
93
+
94
+ response = send_Email(Command, sender, recipient1, subject, body, attachment_path, SERVER_EMAIL_HOST,
95
+ SERVER_EMAIL_PORT)
96
+ print(response)
97
+ response = send_Email(Command, sender, recipient2, subject, body, attachment_path, SERVER_EMAIL_HOST,
98
+ SERVER_EMAIL_PORT)
99
+ print(response)
100
+
101
+
102
+ if __name__ == '__main__':
103
+ main()
core/end_user/EndUserClient.py ADDED
@@ -0,0 +1,386 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import argparse
2
+ import email
3
+ import os
4
+ import random
5
+ import re
6
+ import socket
7
+ import time
8
+ import tkinter as tk
9
+ from email.mime.image import MIMEImage
10
+ from email.mime.multipart import MIMEMultipart
11
+ from email.mime.text import MIMEText
12
+ from tkinter import ttk
13
+
14
+ import pandas as pd
15
+ from PIL import Image, ImageTk
16
+
17
+ # Define global variables
18
+ SERVER_EMAIL_HOST = os.getenv("SERVER_EMAIL_HOST")
19
+ SERVER_EMAIL_PORT = int(os.getenv("SERVER_EMAIL_PORT"))
20
+ SERVER_LLAVA_HOST = os.getenv("SERVER_LLAVA_HOST")
21
+ SERVER_LLAVA_PORT = int(os.getenv("SERVER_LLAVA_PORT"))
22
+ MYEMAIL = os.getenv("MYEMAIL")
23
+ MAILSERVER = os.getenv("MAILSERVER")
24
+ saveMail_directory = os.getenv("SAVE_MAIL_DIRECTORY")
25
+ MyEmails = None
26
+ CycleNewEmails = os.getenv("CYCLE_NEW_EMAILS", "False").lower() in ("true", "1", "t")
27
+ BaseEmails_directory = os.getenv("BASE_EMAILS_DIRECTORY")
28
+ default_image = os.getenv("DEFAULT_IMAGE", '')
29
+
30
+ if not BaseEmails_directory:
31
+ raise ValueError("BASE_EMAILS_DIRECTORY environment variable is not set.")
32
+
33
+
34
+ def receive_complete_data(client_socket): # this function is used to receive the complete data from the client, adjust the parameters as needed based on your network conditions
35
+ received_data = b""
36
+ count = 0
37
+ try:
38
+ while True:
39
+ chunk = client_socket.recv(2 ** 16) # Adjust the buffer size as needed
40
+ if not chunk:
41
+ count += 1
42
+ else:
43
+ count = 0
44
+ received_data += chunk
45
+ if count >= 50:
46
+ break
47
+ except socket.timeout as e:
48
+ print('timeout')
49
+ print(e)
50
+ pass
51
+ except Exception as e:
52
+ print(f"Error receiving data: {e}")
53
+
54
+ return received_data
55
+
56
+
57
+ def parse_email_data(data): # this function gets the data from the inbox and parse it to the email data
58
+ try:
59
+ msg = email.message_from_bytes(data)
60
+
61
+ Command, subject, sender, recipient = msg['Command'], msg["Subject"], msg["From"], msg["To"]
62
+ recipient_directory = f"{saveMail_directory}/{recipient}"
63
+ os.makedirs(recipient_directory, exist_ok=True)
64
+
65
+ if msg.is_multipart():
66
+ for part in msg.get_payload():
67
+ if part.get_content_type() == "text/plain":
68
+ body = part.get_payload()
69
+ else:
70
+ print(msg.get_payload())
71
+ for part in msg.walk():
72
+ if part.get_content_maintype() == "multipart":
73
+ continue
74
+ if part.get("Content-Disposition") is None:
75
+ continue
76
+
77
+ filename = part.get_filename()
78
+ #filename = filename.split("\\")[-1]
79
+ filename = filename.split("/")[-1]
80
+
81
+ # Save the image file
82
+ with open(os.path.join(recipient_directory, filename), "wb") as f:
83
+ f.write(part.get_payload(decode=True))
84
+ print(f'\n Opened and parsed new email from {sender} to {recipient} with subject {subject}')
85
+ print(f'Email body: {body}')
86
+ print(f'Email attachment: {filename}')
87
+
88
+ filepath = str(f"{recipient_directory}/{filename}")
89
+ try: #We faced some network errors resulting in images being sent partially black. To address this issue, we implemented a try-except block to handle such occurrences. Now, if an image fails to send correctly, a default image is sent for that experiment.
90
+ with open(filepath) as f: # TEST IF THE FILE IS A VALID IMAGE
91
+ img = MIMEImage(f.read())
92
+ except: # network error
93
+ if default_image=='':
94
+ print('Network Error: No default image is set')
95
+ return
96
+ else:
97
+ filepath = default_image
98
+
99
+ return (sender, recipient, subject, body, filepath)
100
+ except Exception as e:
101
+ print(f"Error parsing email data: {e}")
102
+ return None
103
+
104
+
105
+ def send_Email(Command, sender, recipient, subject, body, attachment_path, SERVER_HOST, SERVER_PORT,
106
+ AdditionalQuery=['']): # this function sends a new email to the email server
107
+ try:
108
+ with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
109
+ client_socket.connect((SERVER_HOST, SERVER_PORT))
110
+
111
+ # Create the message
112
+ msg = MIMEMultipart()
113
+ msg["Command"] = Command
114
+ msg["Subject"] = subject
115
+ msg["From"] = sender
116
+ msg["To"] = recipient
117
+
118
+ if AdditionalQuery != '':
119
+ for i in range(len(AdditionalQuery)):
120
+ msg["AdditionalQuery" + str(i)] = AdditionalQuery[i]
121
+ msg["AdditionalQueryNum"] = str(len(AdditionalQuery))
122
+ msg.attach(MIMEText(body, "plain"))
123
+
124
+ filename = attachment_path
125
+ with open(filename, "rb") as f:
126
+ img = MIMEImage(f.read())
127
+ img.add_header("Content-Disposition", "attachment", filename=filename)
128
+ msg.attach(img)
129
+ message = msg.as_string().encode('utf-8')
130
+
131
+ client_socket.sendall(message) # send the message to the server
132
+ response = receive_complete_data(client_socket) # get the response from the server
133
+
134
+ return response.decode('utf-8')
135
+ except FileNotFoundError as e:
136
+ print(f"Error: Attachment file not found: {e}")
137
+ return "Error: Attachment file not found"
138
+ except Exception as e:
139
+ print(f"Error sending email: {e}")
140
+ return "Error sending email"
141
+
142
+
143
+ def show_email_popup(email_data): # this function shows a popup with the email data
144
+ popup = tk.Tk()
145
+ popup.title("New Email")
146
+ text_sub_font = ("Helvetica", 12, "bold")
147
+ text_font = ("Helvetica", 10)
148
+ title_style = ttk.Style()
149
+ title_font = ("Helvetica", 16, "bold")
150
+ title_style.configure("Title.TLabel", font=title_font)
151
+ ttk.Label(popup, text="NEW EMAIL!", style="Title.TLabel").pack()
152
+ separator = ttk.Separator(popup, orient='horizontal')
153
+ separator.pack(fill='x')
154
+ email_text = tk.Text(popup, height=10, width=40, wrap=tk.WORD, spacing2=5, bg="#f0f0f0", relief=tk.FLAT)
155
+ email_text.configure(state=tk.DISABLED)
156
+ email_text.tag_configure("bold", font=text_sub_font)
157
+ email_text.tag_configure("normal", font=text_font)
158
+ email_text.configure(state=tk.NORMAL)
159
+ email_text.insert(tk.END, "From: ", "bold")
160
+ email_text.insert(tk.END, email_data[0] + "\n", "normal")
161
+ email_text.insert(tk.END, "To: ", "bold")
162
+ email_text.insert(tk.END, email_data[1] + "\n", "normal")
163
+ email_text.insert(tk.END, "Subject: ", "bold")
164
+ email_text.insert(tk.END, email_data[2] + "\n\n", "normal")
165
+ separator = ttk.Separator(popup, orient='horizontal')
166
+ separator.pack(fill='x')
167
+ email_text.insert(tk.END, email_data[3] + "\n", "normal")
168
+ email_text.configure(state=tk.DISABLED)
169
+ email_text.pack(pady=10)
170
+ image_path = email_data[4]
171
+ image = Image.open(image_path)
172
+ image.thumbnail((200, 200)) # Adjust the size as needed
173
+ tk_image = ImageTk.PhotoImage(image)
174
+ label = tk.Label(popup, image=tk_image, bg="#f0f0f0")
175
+ label.image = tk_image
176
+ label.pack()
177
+ popup.after(5000, popup.destroy) # destroy the popup after 5 seconds
178
+ popup.mainloop() # Show the popup
179
+
180
+
181
+ def check_email_inbox(): # this function checks the inbox for new emails from the server, if there are new emails it shows a popup with the email data and then calls the Handle_New_Inbox_Email function
182
+ try:
183
+ with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
184
+ client_socket.connect((SERVER_EMAIL_HOST, SERVER_EMAIL_PORT))
185
+ msg = MIMEMultipart()
186
+ msg["Command"] = "CHECK_INBOX"
187
+ msg["Subject"] = "CHECK_INBOX"
188
+ msg["From"] = MYEMAIL
189
+ msg["To"] = MAILSERVER
190
+ msg.attach(MIMEText("Check Inbox", "plain"))
191
+ message = msg.as_bytes()
192
+
193
+ client_socket.sendall(message)
194
+ inbox_data = receive_complete_data(client_socket)
195
+ time.sleep(2)
196
+
197
+ if inbox_data == b'No Emails':
198
+ print(f'there are no new Emails in the inbox for you')
199
+ return
200
+ client_socket.close()
201
+ try:
202
+ email_data = parse_email_data(inbox_data)
203
+ if email_data:
204
+ show_email_popup(email_data)
205
+ Handle_New_Inbox_Email(email_data)
206
+ except Exception as e:
207
+ print(f"Error handling new inbox email: {e}")
208
+ except ConnectionRefusedError as e:
209
+ print(f"Error: Connection refused: {e}")
210
+ except Exception as e:
211
+ print(f"Error checking email inbox: {e}")
212
+
213
+
214
+ def read_emails_from_file(): # this function reads 5 emails from the Email csv file and returns them as a list
215
+
216
+ Emails_df = pd.read_csv(BaseEmails_directory)
217
+ """Tailor this function to meet your requirements; it serves as an illustration.
218
+ In our experiment, we utilized a single CSV file containing all the emails, located under "EndUserClientBaseEmails."
219
+ We provided each end-user client with the same CSV file, ensuring that each client reads distinct emails from the file.
220
+ Feel free to re-implement this function, but ensure that the construction of MyEmails follows the same pattern as the example provided below.
221
+ """
222
+ digit = ''
223
+
224
+ for char in MYEMAIL:
225
+ if char.isdigit():
226
+ digit += char
227
+ MyEmails = []
228
+
229
+ for i in range(0, 6):
230
+ Index = i + int(digit) * 6
231
+ Email = Emails_df.iloc[Index]
232
+
233
+ if Email['Sender'] != MYEMAIL:
234
+ Body = Email['Body']
235
+ EmailString = f'sender email address:{Email["Sender"]} \n email content: {Body}'
236
+ MyEmails.append(EmailString)
237
+
238
+ return MyEmails
239
+
240
+
241
+ def Handle_New_Inbox_Email(email_data): # this function handles the new email from the server, it parses the email data and then sends it to the LLaVa server, then it parses the LLaVa response and sends the email to the two recipients that the LLaVa server chose if the LLaVa server chose to forward the email
242
+ sender, recipient, subject, body, filepath = email_data
243
+
244
+ bodyWrap = 'classify the email into one of the following options based on its content and return the JSON format. \n The options are \n 1."forward"- send to someone in case the content is relevant to additional contacts, Forward JSON format: "{"Classification": "forward","Subject": "forwarded email", "Body": "Begin Forwarded Email"}" \n 2."spam"- spam if its content is spam , promotional or advertisement, Spam JSON format: "{"Classification": "Spam","Subject": "", "Body": ""}".\n 3."reply"- reply in case this email requires a user response, Reply JSON format: "{"Classification": "Reply","Subject": "", "Body": ""}" . \n\n ' \
245
+ f'Email content text: {body}.'
246
+
247
+ query2 = 'Return the JSON object'
248
+
249
+ global MyEmails
250
+
251
+ processed_emails = []
252
+ for i, email in enumerate(MyEmails):
253
+ # Replace all non-ASCII characters and remove new lines and replace them with -@- so the utf-8 encoding will work
254
+ processed_email = email.encode('ascii', 'ignore').decode('ascii').replace('\n', '-@-')
255
+ processed_emails.append(processed_email)
256
+
257
+ MyEmails = processed_emails
258
+
259
+ query3=' here are five recent emails I have received: '
260
+ for email in MyEmails:
261
+ query3 += email + '\n'
262
+
263
+ query3 += 'read them and say ok.'
264
+ query4 = 'Based on the content of the five emails, choose the email addresses that might be interested in the content of this new email. return their email address'
265
+ query5 = 'add their email address to the JSON object you built earlier, add them under { "Emails": " , "}'
266
+
267
+ AdditionalQuery = [query2, query3, query4, query5]
268
+ print('\n ---Sending the new email from the inbox to LLaVa for classification ---\n')
269
+ response = send_Email('CHECK_EMAIL', MYEMAIL, sender, subject, bodyWrap, filepath, SERVER_LLAVA_HOST,
270
+ SERVER_LLAVA_PORT, AdditionalQuery) # Send the email to the LLaVa server for classification
271
+ ResponsesDict = {}
272
+ for i in range(len(AdditionalQuery) + 1):
273
+ start_index = response.find(f'Response {i}:')
274
+ end_index = response.find(f'Response {i + 1}:')
275
+ if end_index == -1:
276
+ end_index = len(response)
277
+ ResponsesDict[i] = response[start_index:end_index]
278
+
279
+ ClassificaionRespond1 = ResponsesDict[1]
280
+ data = ResponsesDict[4]
281
+
282
+ ClassificationAll = data.lower()
283
+
284
+ start_index = ClassificationAll.find('forward')
285
+ if start_index != -1:
286
+ Classification = 'forward'
287
+ else:
288
+ start_index = ClassificationAll.find('reply')
289
+ if start_index != -1:
290
+ Classification = 'reply'
291
+ else:
292
+ start_index = ClassificationAll.find('spam')
293
+ if start_index != -1:
294
+ Classification = 'spam'
295
+ else:
296
+ Classification = 'Error'
297
+
298
+ ClassificaionRespond1 = ClassificaionRespond1.lower() # if something goes wrong in the last JSON, the classification will be error and we will use the first JSON to classify the email
299
+ if Classification == 'Error':
300
+ start_index = ClassificaionRespond1.find('forward')
301
+ if start_index != -1:
302
+ Classification = 'forward'
303
+ else:
304
+ start_index = ClassificaionRespond1.find('reply')
305
+ if start_index != -1:
306
+ Classification = 'reply'
307
+ else:
308
+ start_index = ClassificaionRespond1.find('spam')
309
+ if start_index != -1:
310
+ Classification = 'spam'
311
+ else:
312
+ Classification = 'Error2'
313
+
314
+ print('Classification from LLaVa is:', Classification)
315
+
316
+ if Classification == 'reply': # if the LLaVa server chose to reply to the email, we will move the email to the Manual Folder
317
+ print('Manual action is required for replying to this email, so it will be transferred to the Manual Folder.')
318
+ pass
319
+ elif Classification == 'forward':
320
+ print('Starting to forward the emails to the correspondents')
321
+ EmailAddresses = re.findall(r'[\w\.-]+@[\w\.-]+', data)
322
+ Command = "SEND_EMAIL"
323
+ EmailAddresses = list(set(EmailAddresses))
324
+ for Email in EmailAddresses:
325
+ recipient = Email
326
+ response = send_Email(Command, MYEMAIL, recipient, subject, body, filepath, SERVER_EMAIL_HOST,
327
+ SERVER_EMAIL_PORT)
328
+ print(f'{response} to: {recipient}')
329
+
330
+ elif Classification == 'spam':# if the LLaVa server chose to move the email to the spam folder, we will move the email to the Spam Folder
331
+ print('Moving the email to the Spam Folder')
332
+ pass
333
+ else:
334
+ print('Error in classification')
335
+ pass
336
+
337
+ # remove the first email from the list with pop and append the new email to the end of the list
338
+ if CycleNewEmails: #this allows us to decide if we want to cycle the new emails or use the same base emails (in our experiment, we cycled the emails)
339
+ MyEmails.pop(0)
340
+ NewEmailString = f'sender email address:{sender} \n email content: {body}'
341
+ MyEmails.append(NewEmailString)
342
+ else:
343
+ pass
344
+
345
+
346
+
347
+ def main():
348
+ global MAILSERVER, SERVER_EMAIL_HOST, SERVER_EMAIL_PORT, SERVER_LLAVA_HOST, SERVER_LLAVA_PORT, MYEMAIL, BaseEmails_directory, saveMail_directory, MyEmails, CycleNewEmails, default_image
349
+
350
+ MAILSERVER = 'MailServer@example.com'
351
+ parser = argparse.ArgumentParser(description='Description of your program')
352
+ parser.add_argument('--SERVER_EMAIL_HOST', type=str, help='Server Email IP')
353
+ parser.add_argument('--SERVER_EMAIL_PORT', type=int, help='Server Email Port')
354
+ parser.add_argument('--SERVER_LLAVA_HOST', type=str, help='Server LLaVa IP')
355
+ parser.add_argument('--SERVER_LLAVA_PORT', type=int, help='Server LLaVa Port')
356
+ parser.add_argument('--MYEMAIL', type=str, help='PersonX@example.com Email')
357
+ parser.add_argument('--saveMail_directory', type=str, help='Directory to save the emails')
358
+ parser.add_argument('--BaseEmails_directory', type=str, help='Directory to save the base emails')
359
+ parser.add_argument('--CycleNewEmails', type=bool,
360
+ help='True if you want to cycle the new emails, False if you want to use the same base emails')
361
+ parser.add_argument('--default_image', type=str, help='Path to the default image, if you do not want to use the default image, leave it empty')
362
+
363
+ args = parser.parse_args()
364
+ SERVER_EMAIL_HOST = args.SERVER_EMAIL_HOST
365
+ SERVER_EMAIL_PORT = args.SERVER_EMAIL_PORT
366
+ SERVER_LLAVA_HOST = args.SERVER_LLAVA_HOST
367
+ SERVER_LLAVA_PORT = args.SERVER_LLAVA_PORT
368
+ MYEMAIL = args.MYEMAIL
369
+ saveMail_directory = args.saveMail_directory
370
+ BaseEmails_directory = args.BaseEmails_directory
371
+ CycleNewEmails = args.CycleNewEmails
372
+ default_image = args.default_image
373
+ MyEmails = read_emails_from_file()
374
+
375
+ print(f'Starting the Client for Email {MYEMAIL}')
376
+
377
+ while True:
378
+ print('-' * 50)
379
+ time.sleep(random.randint(10, 20))
380
+ print('Checking the inbox for new emails')
381
+ check_email_inbox()
382
+ print('-' * 50)
383
+
384
+
385
+ if __name__ == '__main__':
386
+ main()
core/integrations/email_handler.py ADDED
@@ -0,0 +1,81 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # app/core/integrations/email_handler.py
2
+
3
+ from typing import Optional, List, Dict
4
+ import imaplib
5
+ import email
6
+ import logging
7
+ import smtplib
8
+ from email.mime.text import MIMEText
9
+ from email.mime.multipart import MIMEMultipart
10
+
11
+ class EmailHandler:
12
+ def __init__(self, config):
13
+ self.logger = logging.getLogger(__name__)
14
+ self.config = config
15
+ self.imap = None
16
+ self.smtp = None
17
+
18
+ async def connect(self):
19
+ """Connect to email servers"""
20
+ try:
21
+ # IMAP connection
22
+ self.imap = imaplib.IMAP4_SSL(self.config.IMAP_HOST)
23
+ self.imap.login(self.config.EMAIL, self.config.PASSWORD)
24
+
25
+ # SMTP connection
26
+ self.smtp = smtplib.SMTP(self.config.SMTP_HOST, self.config.SMTP_PORT)
27
+ self.smtp.starttls()
28
+ self.smtp.login(self.config.EMAIL, self.config.PASSWORD)
29
+
30
+ return True
31
+ except Exception as e:
32
+ self.logger.error(f"Connection error: {e}")
33
+ return False
34
+
35
+ async def disconnect(self):
36
+ """Close email connections"""
37
+ try:
38
+ if self.imap:
39
+ self.imap.logout()
40
+ if self.smtp:
41
+ self.smtp.quit()
42
+ except Exception as e:
43
+ self.logger.error(f"Disconnect error: {e}")
44
+
45
+ async def fetch_recent_emails(self, folder="INBOX", limit=10) -> List[Dict]:
46
+ """Fetch recent emails from specified folder"""
47
+ try:
48
+ self.imap.select(folder)
49
+ _, messages = self.imap.search(None, "ALL")
50
+ email_list = []
51
+
52
+ for num in messages[0].split()[-limit:]:
53
+ _, msg = self.imap.fetch(num, "(RFC822)")
54
+ email_message = email.message_from_bytes(msg[0][1])
55
+
56
+ email_data = {
57
+ 'subject': email_message['subject'],
58
+ 'from': email_message['from'],
59
+ 'date': email_message['date'],
60
+ 'body': self.get_email_body(email_message)
61
+ }
62
+ email_list.append(email_data)
63
+
64
+ return email_list
65
+ except Exception as e:
66
+ self.logger.error(f"Error fetching emails: {e}")
67
+ return []
68
+
69
+ def get_email_body(self, email_message) -> str:
70
+ """Extract email body from message"""
71
+ try:
72
+ if email_message.is_multipart():
73
+ body = ""
74
+ for part in email_message.walk():
75
+ if part.get_content_type() == "text/plain":
76
+ body += part.get_payload(decode=True).decode()
77
+ return body
78
+ return email_message.get_payload(decode=True).decode()
79
+ except Exception as e:
80
+ self.logger.error(f"Error extracting email body: {e}")
81
+ return ""
core/plugin_manager.py ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import importlib
3
+ import logging
4
+ import inspect
5
+ import json
6
+ from jsonschema import validate, exceptions, Draft7Validator
7
+
8
+ PLUGIN_DIR = "scan_modules"
9
+
10
+ class PluginManager:
11
+ """Manages the loading and execution of scan modules (plugins)."""
12
+
13
+ def __init__(self):
14
+ self.plugins = {}
15
+ self.load_plugins()
16
+
17
+ def load_plugins(self):
18
+ """Loads all available plugins from the plugin directory."""
19
+ for filename in os.listdir(PLUGIN_DIR):
20
+ if filename.endswith("_module.py"):
21
+ module_name = filename[:-3]
22
+ module_path = os.path.join(PLUGIN_DIR, filename)
23
+ try:
24
+ spec = importlib.util.spec_from_file_location(module_name, module_path)
25
+ module = importlib.util.module_from_spec(spec)
26
+ spec.loader.exec_module(module)
27
+ if hasattr(module, 'Plugin'):
28
+ plugin_class = module.Plugin
29
+ dependencies = getattr(plugin_class, 'dependencies', [])
30
+ config_schema = getattr(plugin_class, 'config_schema', None)
31
+ if self.check_dependencies(dependencies):
32
+ plugin_instance = plugin_class()
33
+ if config_schema:
34
+ try:
35
+ Draft7Validator.check_schema(config_schema)
36
+ plugin_instance.config_schema = config_schema
37
+ except exceptions.SchemaError as e:
38
+ logging.error(f"Invalid config schema for plugin '{plugin_instance.name}': {e}")
39
+ continue
40
+ self.plugins[plugin_instance.name] = plugin_instance
41
+ logging.info(f"Plugin '{plugin_instance.name}' loaded successfully.")
42
+ else:
43
+ logging.warning(f"Plugin '{module_name}' dependencies not met, skipping.")
44
+ else:
45
+ logging.warning(f"Module '{module_name}' does not have a 'Plugin' class.")
46
+ except Exception as e:
47
+ logging.error(f"Error loading plugin '{module_name}': {e}")
48
+
49
+ def check_dependencies(self, dependencies):
50
+ """Checks if all plugin dependencies are met."""
51
+ for dep in dependencies:
52
+ if dep not in self.plugins:
53
+ logging.warning(f"Dependency '{dep}' not found.")
54
+ return False
55
+ return True
56
+
57
+ def get_plugin(self, plugin_name):
58
+ """Returns a plugin instance by name."""
59
+ return self.plugins.get(plugin_name)
60
+
61
+ def get_all_plugins(self):
62
+ """Returns a dictionary of all loaded plugins."""
63
+ return self.plugins
64
+
65
+ def run_plugin(self, plugin_name, target, config, output_dir):
66
+ """Runs a specific plugin."""
67
+ plugin = self.get_plugin(plugin_name)
68
+ if plugin:
69
+ try:
70
+ if hasattr(plugin, 'config_schema'):
71
+ self.validate_plugin_config(config['parameters'], plugin.config_schema)
72
+ plugin.run_scan(target, config, output_dir)
73
+ except Exception as e:
74
+ logging.error(f"Error running plugin '{plugin_name}': {e}")
75
+ else:
76
+ logging.warning(f"Plugin '{plugin_name}' not found.")
77
+
78
+ def validate_plugin_config(self, config_params, config_schema):
79
+ """Validates plugin configuration against its schema."""
80
+ try:
81
+ validate(instance=config_params, schema=config_schema)
82
+ except exceptions.ValidationError as e:
83
+ logging.error(f"Plugin configuration validation error: {e}")
84
+ raise ValueError(f"Plugin configuration validation error: {e}")
dashboard/dashboard.py ADDED
@@ -0,0 +1,440 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, render_template, request, redirect, url_for, session
2
+ from functools import wraps
3
+ from modules.advanced_malware_analysis import AdvancedMalwareAnalysis
4
+ from modules.advanced_social_engineering import AdvancedSocialEngineering
5
+ from modules.real_time_threat_intelligence import RealTimeThreatIntelligence
6
+ from modules.real_time_monitoring import RealTimeMonitoring
7
+ from modules.threat_intelligence import ThreatIntelligence
8
+ from modules.predictive_analytics import PredictiveAnalytics
9
+ from modules.automated_incident_response import AutomatedIncidentResponse
10
+ from modules.ai_red_teaming import AIRedTeaming
11
+ from modules.apt_simulation import APTSimulation
12
+ from modules.machine_learning_ai import MachineLearningAI
13
+ from modules.data_visualization import DataVisualization
14
+ from modules.blockchain_logger import BlockchainLogger
15
+ from modules.cloud_exploitation import CloudExploitation
16
+ from modules.iot_exploitation import IoTExploitation
17
+ from modules.quantum_computing import QuantumComputing
18
+ from modules.edge_computing import EdgeComputing
19
+ from modules.serverless_computing import ServerlessComputing
20
+ from modules.microservices_architecture import MicroservicesArchitecture
21
+ from modules.cloud_native_applications import CloudNativeApplications
22
+ from modules.advanced_decryption import AdvancedDecryption
23
+ from modules.advanced_malware_analysis import AdvancedMalwareAnalysis
24
+ from modules.advanced_social_engineering import AdvancedSocialEngineering
25
+ from modules.alerts_notifications import AlertsNotifications
26
+ from modules.device_fingerprinting import DeviceFingerprinting
27
+ from modules.exploit_payloads import ExploitPayloads
28
+ from modules.fuzzing_engine import FuzzingEngine
29
+ from modules.mitm_stingray import MITMStingray
30
+ from modules.network_exploitation import NetworkExploitation
31
+ from modules.vulnerability_scanner import VulnerabilityScanner
32
+ from modules.wireless_exploitation import WirelessExploitation
33
+ from modules.zero_day_exploits import ZeroDayExploits
34
+ from modules.device_control import DeviceControl
35
+ from modules.windows_control import WindowsControl
36
+ from modules.macos_control import MacOSControl
37
+ from modules.linux_control import LinuxControl
38
+ from modules.android_control import AndroidControl
39
+ from modules.ios_control import iOSControl
40
+ from modules.advanced_device_control import AdvancedDeviceControl
41
+ from modules.otp_interceptor import OTPInterceptor
42
+ from database.models import DocumentAnalysis
43
+ from sqlalchemy import create_engine
44
+ from sqlalchemy.orm import sessionmaker
45
+ import logging
46
+ import pika
47
+ from kafka import KafkaProducer, KafkaConsumer
48
+
49
+ app = Flask(__name__)
50
+ app.secret_key = 'your_secret_key'
51
+
52
+ DATABASE_URL = "sqlite:///document_analysis.db"
53
+ engine = create_engine(DATABASE_URL)
54
+ SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
55
+
56
+ # Dummy user data for RBAC
57
+ users = {
58
+ "admin": {"password": "admin123", "role": "admin"},
59
+ "user": {"password": "user123", "role": "user"},
60
+ "sponsored_user": {"password": "sponsored123", "role": "sponsored_user"}
61
+ }
62
+
63
+ # Role-Based Access Control (RBAC) decorator
64
+ def rbac_required(role):
65
+ def decorator(f):
66
+ @wraps(f)
67
+ def decorated_function(*args, **kwargs):
68
+ if 'username' not in session or users[session['username']]['role'] != role:
69
+ return redirect(url_for('login'))
70
+ return f(*args, **kwargs)
71
+ return decorated_function
72
+ return decorator
73
+
74
+ @app.route("/login", methods=["GET", "POST"])
75
+ def login():
76
+ if request.method == "POST":
77
+ username = request.form["username"]
78
+ password = request.form["password"]
79
+ if username in users and users[username]["password"] == password:
80
+ session["username"] = username
81
+ return redirect(url_for("dashboard"))
82
+ return "Invalid credentials"
83
+ return render_template("login.html")
84
+
85
+ @app.route("/logout")
86
+ def logout():
87
+ session.pop("username", None)
88
+ return redirect(url_for("login"))
89
+
90
+ @app.route("/")
91
+ @rbac_required("user")
92
+ def dashboard():
93
+ try:
94
+ malware_analysis = AdvancedMalwareAnalysis()
95
+ social_engineering = AdvancedSocialEngineering()
96
+ threat_intelligence = RealTimeThreatIntelligence(api_key="YOUR_API_KEY")
97
+ monitoring = RealTimeMonitoring(threat_intelligence_module=threat_intelligence)
98
+ advanced_threat_intelligence = ThreatIntelligence()
99
+ predictive_analytics = PredictiveAnalytics()
100
+ automated_incident_response = AutomatedIncidentResponse()
101
+ ai_red_teaming = AIRedTeaming()
102
+ apt_simulation = APTSimulation()
103
+ machine_learning_ai = MachineLearningAI()
104
+ data_visualization = DataVisualization()
105
+ blockchain_logger = BlockchainLogger()
106
+ cloud_exploitation = CloudExploitation()
107
+ iot_exploitation = IoTExploitation()
108
+ quantum_computing = QuantumComputing()
109
+ edge_computing = EdgeComputing()
110
+ serverless_computing = ServerlessComputing()
111
+ microservices_architecture = MicroservicesArchitecture()
112
+ cloud_native_applications = CloudNativeApplications()
113
+ advanced_decryption = AdvancedDecryption()
114
+ advanced_malware_analysis = AdvancedMalwareAnalysis()
115
+ advanced_social_engineering = AdvancedSocialEngineering()
116
+ alerts_notifications = AlertsNotifications(smtp_server="smtp.example.com", smtp_port=587, smtp_user="user@example.com", smtp_password="password")
117
+ device_fingerprinting = DeviceFingerprinting()
118
+ exploit_payloads = ExploitPayloads()
119
+ fuzzing_engine = FuzzingEngine()
120
+ mitm_stingray = MITMStingray(interface="wlan0")
121
+ network_exploitation = NetworkExploitation()
122
+ vulnerability_scanner = VulnerabilityScanner()
123
+ wireless_exploitation = WirelessExploitation()
124
+ zero_day_exploits = ZeroDayExploits()
125
+ device_control = DeviceControl()
126
+ windows_control = WindowsControl()
127
+ macos_control = MacOSControl()
128
+ linux_control = LinuxControl()
129
+ android_control = AndroidControl()
130
+ ios_control = iOSControl()
131
+ advanced_device_control = AdvancedDeviceControl()
132
+ otp_interceptor = OTPInterceptor(
133
+ email_config={
134
+ 'host': 'your_email_host',
135
+ 'username': 'your_email_username',
136
+ 'password': 'your_email_password'
137
+ },
138
+ twilio_config={
139
+ 'account_sid': 'your_twilio_account_sid',
140
+ 'auth_token': 'your_twilio_auth_token'
141
+ }
142
+ )
143
+
144
+ # Integration checks
145
+ if not all([malware_analysis, social_engineering, threat_intelligence, monitoring, advanced_threat_intelligence, predictive_analytics, automated_incident_response, ai_red_teaming, apt_simulation, machine_learning_ai, data_visualization, blockchain_logger, cloud_exploitation, iot_exploitation, quantum_computing, edge_computing, serverless_computing, microservices_architecture, cloud_native_applications, advanced_decryption, advanced_malware_analysis, advanced_social_engineering, alerts_notifications, device_fingerprinting, exploit_payloads, fuzzing_engine, mitm_stingray, network_exploitation, vulnerability_scanner, wireless_exploitation, zero_day_exploits, device_control, windows_control, macos_control, linux_control, android_control, ios_control, advanced_device_control, otp_interceptor]):
146
+ raise ValueError("Module integration check failed")
147
+
148
+ monitoring.threat_intelligence_module = advanced_threat_intelligence
149
+ monitoring.automated_incident_response = automated_incident_response
150
+ monitoring.ai_red_teaming = ai_red_teaming
151
+ monitoring.apt_simulation = apt_simulation
152
+ monitoring.predictive_analytics = predictive_analytics
153
+ monitoring.machine_learning_ai = machine_learning_ai
154
+ monitoring.data_visualization = data_visualization
155
+ monitoring.cloud_exploitation = cloud_exploitation
156
+ monitoring.iot_exploitation = iot_exploitation
157
+ monitoring.quantum_computing = quantum_computing
158
+ monitoring.edge_computing = edge_computing
159
+ monitoring.serverless_computing = serverless_computing
160
+ monitoring.microservices_architecture = microservices_architecture
161
+ monitoring.cloud_native_applications = cloud_native_applications
162
+ monitoring.device_control = device_control
163
+ monitoring.windows_control = windows_control
164
+ monitoring.macos_control = macos_control
165
+ monitoring.linux_control = linux_control
166
+ monitoring.android_control = android_control
167
+ monitoring.ios_control = ios_control
168
+ monitoring.advanced_device_control = advanced_device_control
169
+
170
+ # Add tool tips and advanced help options for all functions
171
+ def add_tool_tips():
172
+ tool_tips = {
173
+ "advanced_threat_intelligence": "Provides advanced threat intelligence capabilities.",
174
+ "predictive_analytics": "Utilizes predictive analytics for threat detection.",
175
+ "automated_incident_response": "Automates incident response processes.",
176
+ "ai_red_teaming": "AI-driven red teaming for security testing.",
177
+ "apt_simulation": "Simulates advanced persistent threats.",
178
+ "machine_learning_ai": "Machine learning-based AI for threat detection.",
179
+ "data_visualization": "Visualizes data for better insights.",
180
+ "blockchain_logger": "Logs data using blockchain technology.",
181
+ "cloud_exploitation": "Exploits vulnerabilities in cloud environments.",
182
+ "iot_exploitation": "Exploits vulnerabilities in IoT devices.",
183
+ "quantum_computing": "Utilizes quantum computing for security.",
184
+ "edge_computing": "Secures edge computing environments.",
185
+ "serverless_computing": "Secures serverless computing environments.",
186
+ "microservices_architecture": "Secures microservices architectures.",
187
+ "cloud_native_applications": "Secures cloud-native applications.",
188
+ "advanced_decryption": "Advanced decryption capabilities.",
189
+ "advanced_malware_analysis": "Analyzes and detects advanced malware.",
190
+ "advanced_social_engineering": "Detects and prevents social engineering attacks.",
191
+ "alerts_notifications": "Sends alerts and notifications.",
192
+ "device_fingerprinting": "Identifies devices using fingerprinting.",
193
+ "exploit_payloads": "Manages exploit payloads.",
194
+ "fuzzing_engine": "Fuzzing engine for vulnerability detection.",
195
+ "mitm_stingray": "Manages MITM Stingray attacks.",
196
+ "network_exploitation": "Exploits network vulnerabilities.",
197
+ "vulnerability_scanner": "Scans for vulnerabilities.",
198
+ "wireless_exploitation": "Exploits wireless vulnerabilities.",
199
+ "zero_day_exploits": "Manages zero-day exploits.",
200
+ "device_control": "Controls various device functions.",
201
+ "windows_control": "Controls Windows devices.",
202
+ "macos_control": "Controls macOS devices.",
203
+ "linux_control": "Controls Linux devices.",
204
+ "android_control": "Controls Android devices.",
205
+ "ios_control": "Controls iOS devices.",
206
+ "advanced_device_control": "Provides advanced device control features."
207
+ }
208
+ return tool_tips
209
+
210
+ tool_tips = add_tool_tips()
211
+
212
+ # Add a continue button for the AI chatbot to continue incomplete responses
213
+ continue_button = pn.widgets.Button(name="Continue", button_type="primary")
214
+
215
+ # Add a download icon button for downloading zip files of projects
216
+ download_button = pn.widgets.Button(name="Download .zip", button_type="primary", icon="download")
217
+
218
+ # Save dashboard data to the database
219
+ session = SessionLocal()
220
+ try:
221
+ dashboard_data = DocumentAnalysis(
222
+ source="dashboard",
223
+ title="Dashboard Data",
224
+ links=str({
225
+ "threats_detected": 5,
226
+ "exploits_deployed": 3,
227
+ "malware_analysis": malware_analysis.render(),
228
+ "social_engineering": social_engineering.render(),
229
+ "threat_intelligence": threat_intelligence.render(),
230
+ "monitoring": monitoring.render(),
231
+ "advanced_threat_intelligence": advanced_threat_intelligence.render(),
232
+ "predictive_analytics": predictive_analytics.render(),
233
+ "automated_incident_response": automated_incident_response.render(),
234
+ "ai_red_teaming": ai_red_teaming.render(),
235
+ "apt_simulation": apt_simulation.render(),
236
+ "machine_learning_ai": machine_learning_ai.render(),
237
+ "data_visualization": data_visualization.render(),
238
+ "blockchain_logger": blockchain_logger.render(),
239
+ "cloud_exploitation": cloud_exploitation.render(),
240
+ "iot_exploitation": iot_exploitation.render(),
241
+ "quantum_computing": quantum_computing.render(),
242
+ "edge_computing": edge_computing.render(),
243
+ "serverless_computing": serverless_computing.render(),
244
+ "microservices_architecture": microservices_architecture.render(),
245
+ "cloud_native_applications": cloud_native_applications.render(),
246
+ "advanced_decryption": advanced_decryption.render(),
247
+ "advanced_malware_analysis": advanced_malware_analysis.render(),
248
+ "advanced_social_engineering": advanced_social_engineering.render(),
249
+ "alerts_notifications": alerts_notifications.render(),
250
+ "device_fingerprinting": device_fingerprinting.render(),
251
+ "exploit_payloads": exploit_payloads.render(),
252
+ "fuzzing_engine": fuzzing_engine.render(),
253
+ "mitm_stingray": mitm_stingray.render(),
254
+ "network_exploitation": network_exploitation.render(),
255
+ "vulnerability_scanner": vulnerability_scanner.render(),
256
+ "wireless_exploitation": wireless_exploitation.render(),
257
+ "zero_day_exploits": zero_day_exploits.render(),
258
+ "device_control": device_control.render(),
259
+ "windows_control": windows_control.render(),
260
+ "macos_control": macos_control.render(),
261
+ "linux_control": linux_control.render(),
262
+ "android_control": android_control.render(),
263
+ "ios_control": ios_control.render(),
264
+ "advanced_device_control": advanced_device_control.render(),
265
+ "otp_interceptor": otp_interceptor.intercept_email_otp(),
266
+ "otp_interceptor": otp_interceptor.intercept_sms_otp()
267
+ }),
268
+ error=None
269
+ )
270
+ session.add(dashboard_data)
271
+ session.commit()
272
+ except Exception as e:
273
+ logging.error(f"Error saving dashboard data to database: {e}")
274
+ session.rollback()
275
+ finally:
276
+ session.close()
277
+
278
+ return render_template("dashboard.html", data={
279
+ "threats_detected": 5,
280
+ "exploits_deployed": 3,
281
+ "malware_analysis": malware_analysis.render(),
282
+ "social_engineering": social_engineering.render(),
283
+ "threat_intelligence": threat_intelligence.render(),
284
+ "monitoring": monitoring.render(),
285
+ "advanced_threat_intelligence": advanced_threat_intelligence.render(),
286
+ "predictive_analytics": predictive_analytics.render(),
287
+ "automated_incident_response": automated_incident_response.render(),
288
+ "ai_red_teaming": ai_red_teaming.render(),
289
+ "apt_simulation": apt_simulation.render(),
290
+ "machine_learning_ai": machine_learning_ai.render(),
291
+ "data_visualization": data_visualization.render(),
292
+ "blockchain_logger": blockchain_logger.render(),
293
+ "cloud_exploitation": cloud_exploitation.render(),
294
+ "iot_exploitation": iot_exploitation.render(),
295
+ "quantum_computing": quantum_computing.render(),
296
+ "edge_computing": edge_computing.render(),
297
+ "serverless_computing": serverless_computing.render(),
298
+ "microservices_architecture": microservices_architecture.render(),
299
+ "cloud_native_applications": cloud_native_applications.render(),
300
+ "advanced_decryption": advanced_decryption.render(),
301
+ "advanced_malware_analysis": advanced_malware_analysis.render(),
302
+ "advanced_social_engineering": advanced_social_engineering.render(),
303
+ "alerts_notifications": alerts_notifications.render(),
304
+ "device_fingerprinting": device_fingerprinting.render(),
305
+ "exploit_payloads": exploit_payloads.render(),
306
+ "fuzzing_engine": fuzzing_engine.render(),
307
+ "mitm_stingray": mitm_stingray.render(),
308
+ "network_exploitation": network_exploitation.render(),
309
+ "vulnerability_scanner": vulnerability_scanner.render(),
310
+ "wireless_exploitation": wireless_exploitation.render(),
311
+ "zero_day_exploits": zero_day_exploits.render(),
312
+ "device_control": device_control.render(),
313
+ "windows_control": windows_control.render(),
314
+ "macos_control": macos_control.render(),
315
+ "linux_control": linux_control.render(),
316
+ "android_control": android_control.render(),
317
+ "ios_control": ios_control.render(),
318
+ "advanced_device_control": advanced_device_control.render(),
319
+ "otp_interceptor": otp_interceptor.intercept_email_otp(),
320
+ "otp_interceptor": otp_interceptor.intercept_sms_otp(),
321
+ "continue_button": continue_button,
322
+ "download_button": download_button
323
+ })
324
+ except Exception as e:
325
+ logging.error(f"Error initializing dashboard: {e}")
326
+ return "Error initializing dashboard"
327
+
328
+ @app.route("/admin")
329
+ @rbac_required("admin")
330
+ def admin_dashboard():
331
+ try:
332
+ compliance_status = "Compliant"
333
+ training_status = "Completed"
334
+ return render_template("admin_dashboard.html", data={"compliance_status": compliance_status, "training_status": training_status})
335
+ except Exception as e:
336
+ logging.error(f"Error initializing admin dashboard: {e}")
337
+ return "Error initializing admin dashboard"
338
+
339
+ @app.route("/compliance")
340
+ @rbac_required("admin")
341
+ def compliance_dashboard():
342
+ return render_template("compliance_dashboard.html", data={"compliance_status": "Compliant"})
343
+
344
+ @app.route("/training")
345
+ @rbac_required("user")
346
+ def training_dashboard():
347
+ return render_template("training_dashboard.html", data={"training_status": "Completed"})
348
+
349
+ @app.route("/sponsored_user_dashboard")
350
+ @rbac_required("sponsored_user")
351
+ def sponsored_user_dashboard():
352
+ try:
353
+ # Logic to handle sponsored user-specific settings and profile modifications
354
+ sponsored_user_settings = {
355
+ "profile_setting_1": "Value 1",
356
+ "profile_setting_2": "Value 2"
357
+ }
358
+
359
+ # Logic to handle user access settings and settings of the entity
360
+ user_access_settings = {
361
+ "access_setting_1": "Value 1",
362
+ "access_setting_2": "Value 2"
363
+ }
364
+
365
+ # Logic to handle sponsored employee accounts or user profiles of employees of government agencies, government contractors, and other approved entities
366
+ sponsored_employee_accounts = {
367
+ "employee_account_1": "Value 1",
368
+ "employee_account_2": "Value 2"
369
+ }
370
+
371
+ return render_template("sponsored_user_dashboard.html", data={
372
+ "sponsored_user_settings": sponsored_user_settings,
373
+ "user_access_settings": user_access_settings,
374
+ "sponsored_employee_accounts": sponsored_employee_accounts
375
+ })
376
+ except Exception as e:
377
+ logging.error(f"Error initializing sponsored user dashboard: {e}")
378
+ return "Error initializing sponsored user dashboard"
379
+
380
+ # Implement best practices for integrating message queues
381
+ def setup_message_queue():
382
+ try:
383
+ connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
384
+ channel = connection.channel()
385
+ channel.queue_declare(queue='task_queue', durable=True)
386
+ return channel
387
+ except Exception as e:
388
+ logging.error(f"Error setting up message queue: {e}")
389
+ return None
390
+
391
+ message_queue_channel = setup_message_queue()
392
+
393
+ def send_message_to_queue(message):
394
+ try:
395
+ if message_queue_channel:
396
+ message_queue_channel.basic_publish(
397
+ exchange='',
398
+ routing_key='task_queue',
399
+ body=message,
400
+ properties=pika.BasicProperties(
401
+ delivery_mode=2, # make message persistent
402
+ ))
403
+ logging.info(f"Sent message to queue: {message}")
404
+ else:
405
+ logging.error("Message queue channel is not available.")
406
+ except Exception as e:
407
+ logging.error(f"Error sending message to queue: {e}")
408
+
409
+ # Example usage of sending a message to the queue
410
+ send_message_to_queue("Test message")
411
+
412
+ def setup_kafka():
413
+ try:
414
+ producer = KafkaProducer(bootstrap_servers='localhost:9092')
415
+ consumer = KafkaConsumer('my_topic', bootstrap_servers='localhost:9092', auto_offset_reset='earliest', enable_auto_commit=True, group_id='my-group')
416
+ return producer, consumer
417
+ except Exception as e:
418
+ logging.error(f"Error setting up Kafka: {e}")
419
+ return None, None
420
+
421
+ def send_message_to_kafka(producer, topic, message):
422
+ try:
423
+ producer.send(topic, message.encode('utf-8'))
424
+ producer.flush()
425
+ logging.info(f"Sent message to Kafka topic {topic}: {message}")
426
+ except Exception as e:
427
+ logging.error(f"Error sending message to Kafka: {e}")
428
+
429
+ def receive_message_from_kafka(consumer):
430
+ try:
431
+ for message in consumer:
432
+ logging.info(f"Received message from Kafka: {message.value.decode('utf-8')}")
433
+ except Exception as e:
434
+ logging.error(f"Error receiving message from Kafka: {e}")
435
+
436
+ if __name__ == "__main__":
437
+ producer, consumer = setup_kafka()
438
+ if producer and consumer:
439
+ send_message_to_kafka(producer, 'my_topic', 'Test Kafka message')
440
+ receive_message_from_kafka(consumer)
database/__pycache__/models.cpython-311.pyc ADDED
Binary file (1.41 kB). View file
 
database/models.py ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from sqlalchemy import create_engine, Column, String, Integer, Text
2
+ from sqlalchemy.ext.declarative import declarative_base
3
+ from sqlalchemy.orm import sessionmaker
4
+ import logging
5
+ import os
6
+
7
+ Base = declarative_base()
8
+
9
+ class DocumentAnalysis(Base):
10
+ __tablename__ = "document_analysis"
11
+ id = Column(Integer, primary_key=True, autoincrement=True)
12
+ source = Column(String, nullable=False)
13
+ title = Column(String, nullable=True)
14
+ links = Column(Text, nullable=True)
15
+ error = Column(Text, nullable=True)
16
+
17
+ DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///document_analysis.db")
18
+ if not DATABASE_URL:
19
+ raise ValueError("DATABASE_URL environment variable is not set.")
20
+ engine = create_engine(DATABASE_URL)
21
+ SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
22
+ Base.metadata.create_all(bind=engine)
23
+
24
+ # Connect to the apps, dashboards, modules, tools, payloads, and exploits
25
+ from app_security.app_vulnerability_scanner import scan_application
26
+ from app import monitoring, threat_intelligence, advanced_threat_intelligence, predictive_analytics, automated_incident_response, ai_red_teaming, apt_simulation, machine_learning_ai, data_visualization, blockchain_logger, cloud_exploitation, iot_exploitation, quantum_computing, edge_computing, serverless_computing, microservices_architecture, cloud_native_applications
27
+ from backend.code_parser import CodeParser
28
+ from backend.pipeline_manager import PipelineManager
29
+ from c2_dashboard import C2Dashboard
30
+ from chatbot.app import scan_network, deploy_exploit
31
+ from chatbot.chatbot import handle_vulnerability_scanning, handle_exploit_deployment
32
+ from dashboard.dashboard import malware_analysis, social_engineering
33
+ from exploits.exploits2 import deploy_exploit as deploy_exploit2
34
+ from exploits.ios_framework_extracted.iOS_Zero_Click_Framework_Updated.exploits import deploy_exploit as deploy_exploit_ios
35
+ from modules.alerts_notifications import AlertsNotifications
36
+ from modules.apt_simulation import APTSimulation
37
+
38
+ # Configure logging
39
+ logging.basicConfig(level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
40
+
41
+ # Verification of component connections
42
+ def verify_component_connections():
43
+ try:
44
+ # Check database connection
45
+ session = SessionLocal()
46
+ session.execute('SELECT 1')
47
+ session.close()
48
+ logging.info("Database connection verified.")
49
+
50
+ # Check app components
51
+ if not all([monitoring, threat_intelligence, advanced_threat_intelligence, predictive_analytics, automated_incident_response, ai_red_teaming, apt_simulation, machine_learning_ai, data_visualization, blockchain_logger, cloud_exploitation, iot_exploitation, quantum_computing, edge_computing, serverless_computing, microservices_architecture, cloud_native_applications]):
52
+ raise ValueError("App component connection check failed")
53
+ logging.info("App components connection verified.")
54
+
55
+ # Check backend components
56
+ if not all([CodeParser, PipelineManager]):
57
+ raise ValueError("Backend component connection check failed")
58
+ logging.info("Backend components connection verified.")
59
+
60
+ # Check chatbot components
61
+ if not all([scan_network, deploy_exploit, handle_vulnerability_scanning, handle_exploit_deployment]):
62
+ raise ValueError("Chatbot component connection check failed")
63
+ logging.info("Chatbot components connection verified.")
64
+
65
+ # Check dashboard components
66
+ if not all([malware_analysis, social_engineering]):
67
+ raise ValueError("Dashboard component connection check failed")
68
+ logging.info("Dashboard components connection verified.")
69
+
70
+ # Check exploits components
71
+ if not all([deploy_exploit2, deploy_exploit_ios]):
72
+ raise ValueError("Exploits component connection check failed")
73
+ logging.info("Exploits components connection verified.")
74
+
75
+ # Check modules components
76
+ if not all([AlertsNotifications, APTSimulation]):
77
+ raise ValueError("Modules components connection check failed")
78
+ logging.info("Modules components connection verified.")
79
+
80
+ except Exception as e:
81
+ logging.error(f"Component connection verification failed: {e}")
82
+
83
+ # Run verification
84
+ verify_component_connections()
defense/__pycache__/active_defense.cpython-311.pyc ADDED
Binary file (812 Bytes). View file
 
defense/__pycache__/memory_forensics.cpython-311.pyc ADDED
Binary file (824 Bytes). View file
 
defense/active_defense.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import subprocess
3
+
4
+ def active_defense(malicious_domain):
5
+ print(f"Blocking malicious domain: {malicious_domain}")
6
+ try:
7
+ command = f"iptables -A OUTPUT -d {malicious_domain} -j REJECT"
8
+ subprocess.run(command, shell=True)
9
+ except Exception as e:
10
+ print(f"Failed to block domain: {e}")
defense/memory_forensics.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import subprocess
3
+
4
+ def analyze_memory(memory_dump, profile):
5
+ print("Analyzing memory for malware...")
6
+ try:
7
+ command = f"volatility -f {memory_dump} --profile={profile} malfind"
8
+ subprocess.run(command, shell=True)
9
+ except Exception as e:
10
+ print(f"Memory analysis failed: {e}")