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 import org.apache.commons.lang3.BooleanUtils;
  13 import org.opendaylight.lispflowmapping.lisp.serializer.exception.LispSerializationException;
  14 import org.opendaylight.lispflowmapping.lisp.util.ByteUtil;
  15 import org.opendaylight.lispflowmapping.lisp.util.NumberUtil;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapReply;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MessageType;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItem;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemBuilder;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemKey;
  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.nonnullMappingRecordItem()) {
  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.getProbe()) ? Flags.PROBE : 0x00)
  48                 | (BooleanUtils.isTrue(mapReply.getEchoNonceEnabled()) ? 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.nonnullMappingRecordItem()) {
  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()
  83                     .withKey(new MappingRecordItemKey(Integer.toString(i)))
  84                     .setMappingRecord(MappingRecordSerializer.getInstance().deserialize(replyBuffer))
  85                     .build());
  86         }
  87
  88         return builder.build();
  89     }
  90
  91     private interface Length {
  92         int RES = 2;
  93         int HEADER_SIZE = 12;
  94     }
  95
  96     private interface Flags {
  97         int PROBE = 0x08;
  98         int ECHO_NONCE_ENABLED = 0x04;
  99         int SECURITY_ENABLED = 0x02;
 100     }
 101
 102 }