pcep/impl/src/main/java/org/opendaylight/protocol/pcep/impl/message/AbstractMessageParser.java

   1 package org.opendaylight.protocol.pcep.impl.message;
   2
   3 import java.util.Arrays;
   4 import java.util.BitSet;
   5 import java.util.List;
   6
   7 import org.opendaylight.protocol.pcep.spi.MessageParser;
   8 import org.opendaylight.protocol.pcep.spi.MessageSerializer;
   9 import org.opendaylight.protocol.pcep.spi.ObjectHandlerRegistry;
  10 import org.opendaylight.protocol.pcep.spi.ObjectParser;
  11 import org.opendaylight.protocol.pcep.spi.ObjectSerializer;
  12 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
  13 import org.opendaylight.protocol.pcep.spi.PCEPErrorMapping;
  14 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
  15 import org.opendaylight.protocol.util.ByteArray;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcerrBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ObjectHeader;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcep.error.object.ErrorObjectBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessageBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.ErrorsBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.RequestCaseBuilder;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request._case.RequestBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request._case.request.RpsBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
  27
  28 import com.google.common.base.Preconditions;
  29 import com.google.common.collect.Lists;
  30 import com.google.common.primitives.UnsignedBytes;
  31
  32 abstract class AbstractMessageParser implements MessageParser, MessageSerializer {
  33
  34         private static final int COMMON_OBJECT_HEADER_LENGTH = 4;
  35
  36         private static final int OC_F_LENGTH = 1;
  37         private static final int OT_FLAGS_MF_LENGTH = 1;
  38         private static final int OBJ_LENGTH_F_LENGTH = 2;
  39
  40         private static final int OC_F_OFFSET = 0;
  41         private static final int OT_FLAGS_MF_OFFSET = OC_F_OFFSET + OC_F_LENGTH;
  42         private static final int OBJ_LENGTH_F_OFFSET = OT_FLAGS_MF_OFFSET + OT_FLAGS_MF_LENGTH;
  43
  44         private static final int OT_SF_LENGTH = 4;
  45         private static final int FLAGS_SF_LENGTH = 4;
  46
  47         /*
  48          * offsets of fields inside of multi-field in bits
  49          */
  50         private static final int OT_SF_OFFSET = 0;
  51         private static final int FLAGS_SF_OFFSET = OT_SF_OFFSET + OT_SF_LENGTH;
  52
  53         /*
  54          * flags offsets inside multi-filed
  55          */
  56         private static final int P_FLAG_OFFSET = 6;
  57         private static final int I_FLAG_OFFSET = 7;
  58
  59         private final ObjectHandlerRegistry registry;
  60
  61         protected AbstractMessageParser(final ObjectHandlerRegistry registry) {
  62                 this.registry = Preconditions.checkNotNull(registry);
  63         }
  64
  65         protected byte[] serializeObject(final Object object) {
  66                 if (object == null) {
  67                         throw new IllegalArgumentException("Null object passed.");
  68                 }
  69
  70                 final ObjectSerializer serializer = this.registry.getObjectSerializer(object);
  71
  72                 final byte[] valueBytes = serializer.serializeObject(object);
  73
  74                 final byte[] retBytes = new byte[COMMON_OBJECT_HEADER_LENGTH + valueBytes.length];
  75
  76                 // objClass
  77                 retBytes[OC_F_OFFSET] = UnsignedBytes.checkedCast(serializer.getObjectClass());
  78
  79                 // objType_flags multi-field
  80                 retBytes[OT_FLAGS_MF_OFFSET] = UnsignedBytes.checkedCast(serializer.getObjectType() << (Byte.SIZE - OT_SF_LENGTH));
  81                 if (object.isProcessingRule() != null && object.isProcessingRule()) {
  82                         retBytes[OT_FLAGS_MF_OFFSET] |= 1 << Byte.SIZE - (P_FLAG_OFFSET) - 1;
  83                 }
  84                 if (object.isIgnore() != null && object.isIgnore()) {
  85                         retBytes[OT_FLAGS_MF_OFFSET] |= 1 << Byte.SIZE - (I_FLAG_OFFSET) - 1;
  86                 }
  87
  88                 // objLength
  89                 System.arraycopy(ByteArray.intToBytes(valueBytes.length + COMMON_OBJECT_HEADER_LENGTH, OBJ_LENGTH_F_LENGTH), 0, retBytes,
  90                                 OBJ_LENGTH_F_OFFSET, OBJ_LENGTH_F_LENGTH);
  91
  92                 ByteArray.copyWhole(valueBytes, retBytes, COMMON_OBJECT_HEADER_LENGTH);
  93                 return retBytes;
  94         }
  95
  96         private final List<Object> parseObjects(final byte[] bytes) throws PCEPDeserializerException {
  97                 int offset = 0;
  98                 final List<Object> objs = Lists.newArrayList();
  99                 while (bytes.length - offset > 0) {
 100                         if (bytes.length - offset < COMMON_OBJECT_HEADER_LENGTH) {
 101                                 throw new PCEPDeserializerException("Too few bytes in passed array. Passed: " + (bytes.length - offset) + " Expected: >= "
 102                                                 + COMMON_OBJECT_HEADER_LENGTH + ".");
 103                         }
 104
 105                         final int objClass = UnsignedBytes.toInt(bytes[offset]);
 106
 107                         offset += OC_F_LENGTH;
 108
 109                         final int objType = UnsignedBytes.toInt(ByteArray.copyBitsRange(bytes[offset], OT_SF_OFFSET, OT_SF_LENGTH));
 110
 111                         final byte[] flagsBytes = { ByteArray.copyBitsRange(bytes[offset], FLAGS_SF_OFFSET, FLAGS_SF_LENGTH) };
 112
 113                         final BitSet flags = ByteArray.bytesToBitSet(flagsBytes);
 114
 115                         offset += OT_FLAGS_MF_LENGTH;
 116
 117                         final int objLength = ByteArray.bytesToInt(ByteArray.subByte(bytes, offset, OBJ_LENGTH_F_LENGTH));
 118
 119                         if (bytes.length - offset < objLength - COMMON_OBJECT_HEADER_LENGTH) {
 120                                 throw new PCEPDeserializerException("Too few bytes in passed array. Passed: " + (bytes.length - offset) + " Expected: >= "
 121                                                 + objLength + ".");
 122                         }
 123
 124                         offset += OBJ_LENGTH_F_LENGTH;
 125
 126                         // copy bytes for deeper parsing
 127                         final byte[] bytesToPass = ByteArray.subByte(bytes, offset, objLength - COMMON_OBJECT_HEADER_LENGTH);
 128
 129                         offset += objLength - COMMON_OBJECT_HEADER_LENGTH;
 130
 131                         final ObjectParser parser = Preconditions.checkNotNull(this.registry.getObjectParser(objClass, objType));
 132                         final ObjectHeader header = new ObjectHeaderImpl(flags.get(P_FLAG_OFFSET), flags.get(I_FLAG_OFFSET));
 133
 134                         // parseObject is required to return null for P=0 errored objects
 135                         final Object o = parser.parseObject(header, bytesToPass);
 136                         if (o != null) {
 137                                 objs.add(o);
 138                         }
 139                 }
 140
 141                 return objs;
 142         }
 143
 144         public static Message createErrorMsg(final PCEPErrors e) {
 145                 final PCEPErrorMapping maping = PCEPErrorMapping.getInstance();
 146                 return new PcerrBuilder().setPcerrMessage(
 147                                 new PcerrMessageBuilder().setErrors(
 148                                                 Arrays.asList(new ErrorsBuilder().setErrorObject(
 149                                                                 new ErrorObjectBuilder().setType(maping.getFromErrorsEnum(e).type).setValue(
 150                                                                                 maping.getFromErrorsEnum(e).value).build()).build())).build()).build();
 151         }
 152
 153         public static Message createErrorMsg(final PCEPErrors e, final Rp rp) {
 154                 final PCEPErrorMapping maping = PCEPErrorMapping.getInstance();
 155                 return new PcerrBuilder().setPcerrMessage(
 156                                 new PcerrMessageBuilder().setErrorType(
 157                                                 new RequestCaseBuilder().setRequest(
 158                                                                 new RequestBuilder().setRps(Lists.newArrayList(new RpsBuilder().setRp(rp).build())).build()).build()).setErrors(
 159                                                 Arrays.asList(new ErrorsBuilder().setErrorObject(
 160                                                                 new ErrorObjectBuilder().setType(maping.getFromErrorsEnum(e).type).setValue(
 161                                                                                 maping.getFromErrorsEnum(e).value).build()).build())).build()).build();
 162         }
 163
 164         abstract protected Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException;
 165
 166         @Override
 167         public final Message parseMessage(final byte[] buffer, final List<Message> errors) throws PCEPDeserializerException {
 168                 Preconditions.checkNotNull(buffer, "Buffer may not be null");
 169
 170                 // Parse objects first
 171                 final List<Object> objs = parseObjects(buffer);
 172
 173                 // Run validation
 174                 return validate(objs, errors);
 175         }
 176 }