FilesExpand file tree

main

/

parse_args.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

176 lines (144 loc) · 5.91 KB

main

/

parse_args.py

Copy path

Top

File metadata and controls

• Code
• Blame

176 lines (144 loc) · 5.91 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

import os

import argparse

import sys

# CLI Arguments

def parse_args():

# Define CLI Arguments

parser = argparse.ArgumentParser()

server_group = parser.add_mutually_exclusive_group()

query_group = parser.add_mutually_exclusive_group()

verbosity_group = parser.add_mutually_exclusive_group()

server_group.add_argument('-s',

'--server',

type=str,

help='DorXNG Server Instance - '

'Example: \'https://172.17.0.2/search\'')

server_group.add_argument('-S',

'--serverlist',

type=str,

help='Issue Search Queries Across a List of Servers - '

'Format: Newline Delimited')

query_group.add_argument('-q',

'--query',

type=str,

help='Issue a Search Query - Examples: '

'\'search query\' | '

'\'!tch search query\' | '

'\'site:example.com intext:example\'')

query_group.add_argument('-Q',

'--querylist',

type=str,

help='Iterate Through a Search Query List - '

'Format: Newline Delimited')

parser.add_argument('-n',

'--number',

type=int,

default=1,

help='Define the Number of Page Result Iterations')

parser.add_argument('-c',

'--concurrent',

type=int,

help='Define the Number of Concurrent Page Requests')

parser.add_argument('-l',

'--limitdatabase',

type=int,

help='Set Maximum Database Size Limit - '

'Starts New Database After Exceeded - '

'Example: --limitdatabase 10 (10k Database Entries) - '

'Suggested Maximum Database Size is 50k when doing Deep Recursion')

parser.add_argument('-L',

'--loop',

type=int,

help='Define the Number of Main Function Loop Iterations - '

'Infinite Loop with 0')

parser.add_argument('-d',

'--database',

type=str,

default='dorxng.db',

help='Specify SQL Database File - '

'Default: \'dorxng.db\'')

parser.add_argument('-D',

'--databasequery',

type=str,

help='Issue Database Query - '

'Format: Regex')

parser.add_argument('-m',

'--mergedatabase',

type=str,

help='Merge SQL Database File')

parser.add_argument('-t',

'--timeout',

type=int,

default=4,

help='Specify Timeout Interval Between Requests - '

'Default: 4 Seconds - Disable with 0')

parser.add_argument('-r',

'--nonewresults',

type=int,

default='4',

help='Specify Number of Iterations with No New Results - '

'Default: 4 (3 Attempts) - Disable with 0')

verbosity_group.add_argument('-v',

'--verbose',

help='Enable Verbose Output',

action='store_true')

verbosity_group.add_argument('-vv',

'--veryverbose',

help='Enable Very Verbose Output - '

'Displays Raw JSON Output',

action='store_true')

# Parse CLI Arguments

args = parser.parse_args()

# If no CLI Arguments Were Given Print Help

if len(sys.argv) == 1:

parser.print_help()

exit(0)

# HARD CODE ME OR USE -s/--server OR -S/--serverlist

server = 'https://172.17.0.2/search'

# Set Server via -s/--server CLI Switch

if args.server is not None:

server = args.server

# Set Server to an Empty String if Not Set with -s or Hardcoded

if args.server is None and server is None:

server = str()

# Handle Page Number Iteration Option

if args.number != 1:

page_iteration_mode = True

else:

page_iteration_mode = False

# Handle Request Timer Option

if args.timeout is not None:

timer = args.timeout

# Handle Database File Option

if args.database is not str('dorxng.db'):

database_name = args.database

else:

database_name = 'dorxng.db'

# Handle Server List Option

if args.serverlist is not None:

if not os.path.exists('./' + args.serverlist):

print('No Server List "' + str(args.serverlist) + '" Detected..')

exit(1)

else:

with open(args.serverlist) as file:

server_list = [line.rstrip() for line in file]

else:

server_list = []

# Handle Query List Option

if args.querylist is not None:

if not os.path.exists('./' + args.querylist):

print('No Query List "' + str(args.querylist) + '" Detected..')

exit(1)

else:

with open(args.querylist) as file:

query_list = [line.rstrip() for line in file]

args.query = query_list[0]

else:

query_list = []

# Handle Concurrency Option

if args.concurrent is not None:

concurrent_connections = args.concurrent

else:

concurrent_connections = int()

# Return CLI Arguments to Main

return args, server, page_iteration_mode, timer, database_name, server_list, query_list, concurrent_connections