2 * Copyright (c) 2014 Contextream, Inc. and others. All rights reserved.
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
8 package org.opendaylight.lispflowmapping.lisp.serializer;
10 import java.nio.ByteBuffer;
11 import java.util.ArrayList;
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;
24 * This class deals with serializing map reply from the java object to udp.
26 public final class MapReplySerializer {
28 private static final MapReplySerializer INSTANCE = new MapReplySerializer();
30 // Private constructor prevents instantiation from other classes
31 private MapReplySerializer() {
34 public static MapReplySerializer getInstance() {
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());
44 ByteBuffer replyBuffer = ByteBuffer.allocate(size);
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)));
50 replyBuffer.position(replyBuffer.position() + Length.RES);
51 if (mapReply.getMappingRecordItem() != null) {
52 replyBuffer.put((byte) mapReply.getMappingRecordItem().size());
54 replyBuffer.put((byte) 0);
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());
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);
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());
86 return builder.build();
89 private interface Length {
94 private interface Flags {
96 int ECHO_NONCE_ENABLED = 0x04;
97 int SECURITY_ENABLED = 0x02;