Bug 9127: Make IT more robust when receiving packets
[lispflowmapping.git] / mappingservice / lisp-proto / src / main / java / org / opendaylight / lispflowmapping / lisp / serializer / MapReplySerializer.java
1 /*
2  * Copyright (c) 2014 Contextream, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.lispflowmapping.lisp.serializer;
9
10 import java.nio.ByteBuffer;
11 import java.util.ArrayList;
12
13 import org.apache.commons.lang3.BooleanUtils;
14 import org.opendaylight.lispflowmapping.lisp.serializer.exception.LispSerializationException;
15 import org.opendaylight.lispflowmapping.lisp.util.ByteUtil;
16 import org.opendaylight.lispflowmapping.lisp.util.NumberUtil;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapReply;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MessageType;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItem;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapreplymessage.MapReplyBuilder;
22
23 /**
24  * This class deals with serializing map reply from the java object to udp.
25  */
26 public final class MapReplySerializer {
27
28     private static final MapReplySerializer INSTANCE = new MapReplySerializer();
29
30     // Private constructor prevents instantiation from other classes
31     private MapReplySerializer() {
32     }
33
34     public static MapReplySerializer getInstance() {
35         return INSTANCE;
36     }
37
38     public ByteBuffer serialize(MapReply mapReply) {
39         int size = Length.HEADER_SIZE;
40         for (MappingRecordItem eidToLocatorRecord : mapReply.getMappingRecordItem()) {
41             size += MappingRecordSerializer.getInstance().getSerializationSize(eidToLocatorRecord.getMappingRecord());
42         }
43
44         ByteBuffer replyBuffer = ByteBuffer.allocate(size);
45
46         replyBuffer.put((byte) ((MessageType.MapReply.getIntValue() << 4)
47                 | (BooleanUtils.isTrue(mapReply.isProbe()) ? Flags.PROBE : 0x00)
48                 | (BooleanUtils.isTrue(mapReply.isEchoNonceEnabled()) ? Flags.ECHO_NONCE_ENABLED : 0x00)));
49
50         replyBuffer.position(replyBuffer.position() + Length.RES);
51         if (mapReply.getMappingRecordItem() != null) {
52             replyBuffer.put((byte) mapReply.getMappingRecordItem().size());
53         } else {
54             replyBuffer.put((byte) 0);
55
56         }
57         replyBuffer.putLong(NumberUtil.asLong(mapReply.getNonce()));
58         if (mapReply.getMappingRecordItem() != null) {
59             for (MappingRecordItem eidToLocatorRecord : mapReply.getMappingRecordItem()) {
60                 MappingRecordSerializer.getInstance().serialize(replyBuffer, eidToLocatorRecord.getMappingRecord());
61             }
62         }
63         return replyBuffer;
64     }
65
66     public MapReply deserialize(ByteBuffer replyBuffer) {
67         final byte typeAndFlags = replyBuffer.get();
68         final int type = typeAndFlags >> 4;
69         if (MessageType.forValue(type) != MessageType.MapReply) {
70             throw new LispSerializationException("Expected Map-Reply packet (type 2), but was type " + type);
71         }
72
73         MapReplyBuilder builder = new MapReplyBuilder();
74         builder.setProbe(ByteUtil.extractBit(typeAndFlags, Flags.PROBE));
75         builder.setEchoNonceEnabled(ByteUtil.extractBit(typeAndFlags, Flags.ECHO_NONCE_ENABLED));
76         builder.setSecurityEnabled(ByteUtil.extractBit(typeAndFlags, Flags.SECURITY_ENABLED));
77         replyBuffer.getShort();
78         int recordCount = ByteUtil.getUnsignedByte(replyBuffer);
79         builder.setNonce(replyBuffer.getLong());
80         builder.setMappingRecordItem(new ArrayList<MappingRecordItem>());
81         for (int i = 0; i < recordCount; i++) {
82             builder.getMappingRecordItem().add(new MappingRecordItemBuilder().setMappingRecord(
83                     MappingRecordSerializer.getInstance().deserialize(replyBuffer)).build());
84         }
85
86         return builder.build();
87     }
88
89     private interface Length {
90         int RES = 2;
91         int HEADER_SIZE = 12;
92     }
93
94     private interface Flags {
95         int PROBE = 0x08;
96         int ECHO_NONCE_ENABLED = 0x04;
97         int SECURITY_ENABLED = 0x02;
98     }
99
100 }