Skip to content

Commit

Permalink
Removed V4 + V6 options
Browse files Browse the repository at this point in the history
  • Loading branch information
h345983745 committed Jun 29, 2019
1 parent 219469f commit bac2e8c
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 49 deletions.
37 changes: 13 additions & 24 deletions src/core/operations/DefangIP.mjs
Original file line number Diff line number Diff line change
Expand Up @@ -24,18 +24,8 @@ class DefangIP extends Operation {
this.infoURL = "https://isc.sans.edu/forums/diary/Defang+all+the+things/22744/";
this.inputType = "string";
this.outputType = "string";
this.args = [
{
name: "IPV4",
type: "boolean",
value: true
},
{
name: "IPV6",
type: "boolean",
value: true
}
];
this.args = [];

}

/**
Expand All @@ -44,18 +34,17 @@ class DefangIP extends Operation {
* @returns {string}
*/
run(input, args) {
const [IPV4, IPV6] = args;

if (IPV4) {
input = input.replace(IPV4_REGEX, x => {
return x.replace(/\./g, "[.]");
});
}
if (IPV6) {
input = input.replace(IPV6_REGEX, x => {
return x.replace(/:/g, "[:]");
});
}


input = input.replace(IPV4_REGEX, x => {
return x.replace(/\./g, "[.]");
});


input = input.replace(IPV6_REGEX, x => {
return x.replace(/:/g, "[:]");
});

return input;
}
}
Expand Down
29 changes: 4 additions & 25 deletions tests/operations/tests/DefangIP.mjs
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
*
* @author h345983745
*
* @copyright Crown Copyright 2017
* @copyright Crown Copyright 2019
* @license Apache-2.0
*/
import TestRegister from "../TestRegister";
Expand All @@ -16,7 +16,7 @@ TestRegister.addTests([
recipeConfig: [
{
op: "Defang IP",
args: [true, true],
args: [],
},
],
}, {
Expand All @@ -26,7 +26,7 @@ TestRegister.addTests([
recipeConfig: [
{
op: "Defang IP",
args: [true, true],
args: [],
},
],
}, {
Expand All @@ -36,29 +36,8 @@ TestRegister.addTests([
recipeConfig: [
{
op: "Defang IP",
args: [true, true],
args: [],
},
],
},
{
name: "Defang IP: IPV4 Only",
input: "192.168.1.1 2001:0db8:85a3:0000:0000:8a2e:0370:7343",
expectedOutput: "192[.]168[.]1[.]1 2001:0db8:85a3:0000:0000:8a2e:0370:7343",
recipeConfig: [
{
op: "Defang IP",
args: [true, false],
},
],
}, {
name: "Defang IP: IPV6 Only",
input: "192.168.1.1 2001:0db8:85a3:0000:0000:8a2e:0370:7343",
expectedOutput: "192.168.1.1 2001[:]0db8[:]85a3[:]0000[:]0000[:]8a2e[:]0370[:]7343",
recipeConfig: [
{
op: "Defang IP",
args: [false, true],
},
],
}
]);

0 comments on commit bac2e8c

Please sign in to comment.