53fae7bf8889cdcc52553fd2a0574b16dce9653a
[openflowjava.git] / openflow-protocol-impl / src / main / java / org / opendaylight / openflowjava / protocol / impl / serialization / factories / MeterModInputMessageFactory.java
1 /*
2  * Copyright (c) 2013 Pantheon Technologies s.r.o. 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
9 package org.opendaylight.openflowjava.protocol.impl.serialization.factories;
10
11 import io.netty.buffer.ByteBuf;
12
13 import java.util.HashMap;
14 import java.util.List;
15 import java.util.Map;
16
17 import org.opendaylight.openflowjava.protocol.api.extensibility.MessageTypeKey;
18 import org.opendaylight.openflowjava.protocol.api.extensibility.OFSerializer;
19 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
20 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistryInjector;
21 import org.opendaylight.openflowjava.protocol.impl.util.ByteBufUtils;
22 import org.opendaylight.openflowjava.protocol.impl.util.EncodeConstants;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterFlags;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MeterBandCommons;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MeterModInput;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.MeterBand;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.meter.band.MeterBandDropCase;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.meter.band.MeterBandDscpRemarkCase;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.meter.band.MeterBandExperimenterCase;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.meter.band.meter.band.drop._case.MeterBandDrop;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.meter.band.meter.band.dscp.remark._case.MeterBandDscpRemark;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.meter.band.meter.band.experimenter._case.MeterBandExperimenter;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.mod.Bands;
34
35 /**
36  * Translates MeterMod messages
37  * @author timotej.kubas
38  * @author michal.polkorab
39  */
40 public class MeterModInputMessageFactory implements OFSerializer<MeterModInput>,
41                 SerializerRegistryInjector {
42         
43     private static final byte MESSAGE_TYPE = 29;
44     private static final short LENGTH_OF_METER_BANDS = 16;
45     private static final short PADDING_IN_METER_BAND_DROP = 4;
46     private static final short PADDING_IN_METER_BAND_DSCP_REMARK = 3;
47         private SerializerRegistry registry;
48
49     @Override
50     public void serialize(MeterModInput message, ByteBuf outBuffer) {
51         ByteBufUtils.writeOFHeader(MESSAGE_TYPE, message, outBuffer, EncodeConstants.EMPTY_LENGTH);
52         outBuffer.writeShort(message.getCommand().getIntValue());
53         outBuffer.writeShort(createMeterFlagsBitmask(message.getFlags()));
54         outBuffer.writeInt(message.getMeterId().getValue().intValue());
55         serializeBands(message.getBands(), outBuffer);
56         ByteBufUtils.updateOFHeaderLength(outBuffer);
57     }
58
59     private static int createMeterFlagsBitmask(MeterFlags flags) {
60         int meterFlagBitmask = 0;
61         Map<Integer, Boolean> meterModFlagsMap = new HashMap<>();
62         meterModFlagsMap.put(0, flags.isOFPMFKBPS());
63         meterModFlagsMap.put(1, flags.isOFPMFPKTPS());
64         meterModFlagsMap.put(2, flags.isOFPMFBURST());
65         meterModFlagsMap.put(3, flags.isOFPMFSTATS());
66         
67         meterFlagBitmask = ByteBufUtils.fillBitMaskFromMap(meterModFlagsMap);
68         return meterFlagBitmask;
69     }
70     
71     private void serializeBands(List<Bands> bands, ByteBuf outBuffer) {
72         if (bands != null) {
73             for (Bands currentBand : bands) {
74                 MeterBand meterBand = currentBand.getMeterBand();
75                 if (meterBand instanceof MeterBandDropCase) {
76                     MeterBandDropCase dropBandCase = (MeterBandDropCase) meterBand;
77                     MeterBandDrop dropBand = dropBandCase.getMeterBandDrop();
78                     writeBandCommonFields(dropBand, outBuffer);
79                     ByteBufUtils.padBuffer(PADDING_IN_METER_BAND_DROP, outBuffer);
80                 } else if (meterBand instanceof MeterBandDscpRemarkCase) {
81                     MeterBandDscpRemarkCase dscpRemarkBandCase = (MeterBandDscpRemarkCase) meterBand;
82                     MeterBandDscpRemark dscpRemarkBand = dscpRemarkBandCase.getMeterBandDscpRemark();
83                     writeBandCommonFields(dscpRemarkBand, outBuffer);
84                     outBuffer.writeByte(dscpRemarkBand.getPrecLevel());
85                     ByteBufUtils.padBuffer(PADDING_IN_METER_BAND_DSCP_REMARK, outBuffer);
86                 } else if (meterBand instanceof MeterBandExperimenterCase) {
87                         OFSerializer<MeterBandExperimenter> serializer = registry.getSerializer(
88                                         new MessageTypeKey<>(EncodeConstants.OF13_VERSION_ID, MeterBandExperimenter.class));
89                     MeterBandExperimenterCase experimenterBandCase = (MeterBandExperimenterCase) meterBand;
90                     MeterBandExperimenter experimenterBand = experimenterBandCase.getMeterBandExperimenter();
91                     serializer.serialize(experimenterBand, outBuffer);
92                 }
93             }
94         }
95     }
96     
97     private static void writeBandCommonFields(MeterBandCommons meterBand, ByteBuf outBuffer) {
98         outBuffer.writeShort(meterBand.getType().getIntValue());
99         outBuffer.writeShort(LENGTH_OF_METER_BANDS);
100         outBuffer.writeInt(meterBand.getRate().intValue());
101         outBuffer.writeInt(meterBand.getBurstSize().intValue());
102     }
103
104         @Override
105         public void injectSerializerRegistry(SerializerRegistry serializerRegistry) {
106                 registry = serializerRegistry;
107         }
108
109 }