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
|
package com.it_jaros.jscanner;
import java.util.BitSet;
import java.util.Iterator;
import java.util.NoSuchElementException;
public class PortRange implements Iterator<Integer> {
public static final int MIN_PORT = 1;
public static final int MAX_PORT = 65535;
private final BitSet specifiedPorts = new BitSet(MAX_PORT);
private final BitSet availablePorts = new BitSet(MAX_PORT);
private int currentPortCursor;
private int done = 0;
public PortRange(String ports) {
if (ports == null || "".equals(ports)) {
specifiedPorts.set(MIN_PORT, 1024);
} else if ("all".equalsIgnoreCase(ports)) {
specifiedPorts.set(MIN_PORT, MAX_PORT);
} else {
parsePortRange(ports);
}
availablePorts.or(specifiedPorts);
currentPortCursor = availablePorts.nextSetBit(MIN_PORT);
}
/**
* Possible values are 1,10 or 1-10 or a mix of 1,5-10
* So we treat , stronger and handle them first and then check for -
*
* @param ports
*/
private void parsePortRange(String ports) {
String[] splitComma = ports.split(",");
for (String commaValue : splitComma) {
String[] rangeValue = commaValue.split("-");
// more then two values are not possible
if (rangeValue.length > 2) {
throw new IllegalArgumentException(String.format("Argument contains too many '-'' %s", commaValue));
}
// no range given, only single port
int port = Integer.parseInt(rangeValue[0]);
if (rangeValue.length == 1) {
checkValue(port);
specifiedPorts.set(port);
continue;
}
// range given
int end = Integer.parseInt(rangeValue[1]);
checkValues(port, end);
specifiedPorts.set(port, end + 1);
}
}
private void checkValues(int start, int end) {
checkValue(start);
checkValue(end);
if (start >= end) {
throw new IllegalArgumentException(
String.format("Start value cannot be equal or bigger than end value '%s >= %s'", start, end));
}
}
private void checkValue(int port) {
if (port < MIN_PORT) {
throw new IllegalArgumentException(
String.format("Start value smaller than allowed range %s < %s", port, MIN_PORT));
}
if (port > MAX_PORT) {
throw new IllegalArgumentException(
String.format("End value bigger than allowed range %s > %s", port, MAX_PORT));
}
}
@Override
public Integer next() {
int p = availablePorts.nextSetBit(currentPortCursor);
if (p > 0) {
availablePorts.clear(p);
currentPortCursor = p + 1;
done++;
return p;
}
throw new NoSuchElementException("Reached end of port range");
}
@Override
public boolean hasNext() {
return availablePorts.nextSetBit(currentPortCursor) > 0;
}
public int getTotal() {
return specifiedPorts.cardinality();
}
public int getDone() {
return done;
}
}
|
