1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21 package org.apache.mina.filter.errorgenerating;
22
23 import java.util.Random;
24
25 import org.apache.mina.core.buffer.IoBuffer;
26 import org.apache.mina.core.filterchain.IoFilter;
27 import org.apache.mina.core.filterchain.IoFilterAdapter;
28 import org.apache.mina.core.session.IoSession;
29 import org.apache.mina.core.write.DefaultWriteRequest;
30 import org.apache.mina.core.write.WriteRequest;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55 public class ErrorGeneratingFilter extends IoFilterAdapter {
56 private int removeByteProbability = 0;
57
58 private int insertByteProbability = 0;
59
60 private int changeByteProbability = 0;
61
62 private int removePduProbability = 0;
63
64 private int duplicatePduProbability = 0;
65
66 private int resendPduLasterProbability = 0;
67
68 private int maxInsertByte = 10;
69
70 private boolean manipulateWrites = false;
71
72 private boolean manipulateReads = false;
73
74 private Random rng = new Random();
75
76 private final Logger logger = LoggerFactory.getLogger(ErrorGeneratingFilter.class);
77
78 @Override
79 public void filterWrite(NextFilter nextFilter, IoSession session, WriteRequest writeRequest) throws Exception {
80 if (manipulateWrites) {
81
82 if (writeRequest.getMessage() instanceof IoBuffer) {
83 manipulateIoBuffer(session, (IoBuffer) writeRequest.getMessage());
84 IoBufferbuffer/IoBuffer.html#IoBuffer">IoBuffer buffer = insertBytesToNewIoBuffer(session, (IoBuffer) writeRequest.getMessage());
85
86 if (buffer != null) {
87 writeRequest = new DefaultWriteRequest(buffer, writeRequest.getFuture(),
88 writeRequest.getDestination());
89 }
90
91 } else {
92 if (duplicatePduProbability > rng.nextInt()) {
93 nextFilter.filterWrite(session, writeRequest);
94 }
95
96 if (resendPduLasterProbability > rng.nextInt()) {
97
98
99
100 }
101
102 if (removePduProbability > rng.nextInt()) {
103 return;
104 }
105 }
106 }
107
108 nextFilter.filterWrite(session, writeRequest);
109 }
110
111 @Override
112 public void messageReceived(NextFilter nextFilter, IoSession session, Object message) throws Exception {
113 if (manipulateReads && (message instanceof IoBuffer)) {
114
115 manipulateIoBuffer(session, (IoBuffer) message);
116 IoBufferbuffer/IoBuffer.html#IoBuffer">IoBuffer buffer = insertBytesToNewIoBuffer(session, (IoBuffer) message);
117
118 if (buffer != null) {
119 message = buffer;
120 }
121 }
122
123 nextFilter.messageReceived(session, message);
124 }
125
126 private IoBufferbuffer/IoBuffer.html#IoBuffer">IoBuffer insertBytesToNewIoBuffer(IoSession session, IoBuffer buffer) {
127 if (insertByteProbability > rng.nextInt(1000)) {
128 logger.info(buffer.getHexDump());
129
130 int pos = rng.nextInt(buffer.remaining()) - 1;
131
132
133 int count = rng.nextInt(maxInsertByte - 1) + 1;
134
135 IoBuffer newBuff = IoBuffer.allocate(buffer.remaining() + count);
136 for (int i = 0; i < pos; i++)
137 newBuff.put(buffer.get());
138 for (int i = 0; i < count; i++) {
139 newBuff.put((byte) (rng.nextInt(256)));
140 }
141 while (buffer.remaining() > 0) {
142 newBuff.put(buffer.get());
143 }
144 newBuff.flip();
145
146 logger.info("Inserted " + count + " bytes.");
147 logger.info(newBuff.getHexDump());
148 return newBuff;
149 }
150 return null;
151 }
152
153 private void manipulateIoBuffer(IoSession session, IoBuffer buffer) {
154 if ((buffer.remaining() > 0) && (removeByteProbability > rng.nextInt(1000))) {
155 logger.info(buffer.getHexDump());
156
157 int pos = rng.nextInt(buffer.remaining());
158
159 int count = rng.nextInt(buffer.remaining() - pos) + 1;
160 if (count == buffer.remaining())
161 count = buffer.remaining() - 1;
162
163 IoBuffer newBuff = IoBuffer.allocate(buffer.remaining() - count);
164 for (int i = 0; i < pos; i++)
165 newBuff.put(buffer.get());
166
167 buffer.skip(count);
168 while (newBuff.remaining() > 0)
169 newBuff.put(buffer.get());
170 newBuff.flip();
171
172 buffer.rewind();
173 buffer.put(newBuff);
174 buffer.flip();
175 logger.info("Removed " + count + " bytes at position " + pos + ".");
176 logger.info(buffer.getHexDump());
177 }
178 if ((buffer.remaining() > 0) && (changeByteProbability > rng.nextInt(1000))) {
179 logger.info(buffer.getHexDump());
180
181 int count = rng.nextInt(buffer.remaining() - 1) + 1;
182
183 byte[] values = new byte[count];
184 rng.nextBytes(values);
185 for (int i = 0; i < values.length; i++) {
186 int pos = rng.nextInt(buffer.remaining());
187 buffer.put(pos, values[i]);
188 }
189 logger.info("Modified " + count + " bytes.");
190 logger.info(buffer.getHexDump());
191 }
192 }
193
194
195
196
197 public int getChangeByteProbability() {
198 return changeByteProbability;
199 }
200
201
202
203
204
205
206
207 public void setChangeByteProbability(int changeByteProbability) {
208 this.changeByteProbability = changeByteProbability;
209 }
210
211
212
213
214 public int getDuplicatePduProbability() {
215 return duplicatePduProbability;
216 }
217
218
219
220
221
222 public void setDuplicatePduProbability(int duplicatePduProbability) {
223 this.duplicatePduProbability = duplicatePduProbability;
224 }
225
226
227
228
229 public int getInsertByteProbability() {
230 return insertByteProbability;
231 }
232
233
234
235
236
237
238
239 public void setInsertByteProbability(int insertByteProbability) {
240 this.insertByteProbability = insertByteProbability;
241 }
242
243
244
245
246 public boolean isManipulateReads() {
247 return manipulateReads;
248 }
249
250
251
252
253
254
255 public void setManipulateReads(boolean manipulateReads) {
256 this.manipulateReads = manipulateReads;
257 }
258
259
260
261
262 public boolean isManipulateWrites() {
263 return manipulateWrites;
264 }
265
266
267
268
269
270
271 public void setManipulateWrites(boolean manipulateWrites) {
272 this.manipulateWrites = manipulateWrites;
273 }
274
275
276
277
278 public int getRemoveByteProbability() {
279 return removeByteProbability;
280 }
281
282
283
284
285
286
287
288
289 public void setRemoveByteProbability(int removeByteProbability) {
290 this.removeByteProbability = removeByteProbability;
291 }
292
293
294
295
296 public int getRemovePduProbability() {
297 return removePduProbability;
298 }
299
300
301
302
303
304 public void setRemovePduProbability(int removePduProbability) {
305 this.removePduProbability = removePduProbability;
306 }
307
308
309
310
311 public int getResendPduLasterProbability() {
312 return resendPduLasterProbability;
313 }
314
315
316
317
318
319 public void setResendPduLasterProbability(int resendPduLasterProbability) {
320 this.resendPduLasterProbability = resendPduLasterProbability;
321 }
322
323
324
325
326 public int getMaxInsertByte() {
327 return maxInsertByte;
328 }
329
330
331
332
333
334
335 public void setMaxInsertByte(int maxInsertByte) {
336 this.maxInsertByte = maxInsertByte;
337 }
338 }