X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=bgp%2Fparser-impl%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fprotocol%2Fbgp%2Fparser%2Fimpl%2Fmessage%2FBGPUpdateMessageParser.java;h=8f76c4e01320b93619c01f047f23720722f0975f;hb=3c135df98a164d4207775de96a5217a392d1949e;hp=bc2f5f324f6ce370fa5014e57546263cb650052f;hpb=fef52c72498ea2e4d84ea6686f708fd9898ff0ac;p=bgpcep.git diff --git a/bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/BGPUpdateMessageParser.java b/bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/BGPUpdateMessageParser.java index bc2f5f324f..8f76c4e013 100644 --- a/bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/BGPUpdateMessageParser.java +++ b/bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/BGPUpdateMessageParser.java @@ -5,107 +5,132 @@ * terms of the Eclipse Public License v1.0 which accompanies this distribution, * and is available at http://www.eclipse.org/legal/epl-v10.html */ - package org.opendaylight.protocol.bgp.parser.impl.message; -import java.util.Arrays; -import java.util.List; +import com.google.common.base.Preconditions; +import io.netty.buffer.ByteBuf; +import io.netty.buffer.ByteBufUtil; +import io.netty.buffer.Unpooled; +import java.util.List; import org.opendaylight.protocol.bgp.parser.BGPDocumentedException; import org.opendaylight.protocol.bgp.parser.BGPError; import org.opendaylight.protocol.bgp.parser.BGPParsingException; import org.opendaylight.protocol.bgp.parser.spi.AttributeRegistry; import org.opendaylight.protocol.bgp.parser.spi.MessageParser; -import org.opendaylight.protocol.concepts.Ipv4Util; +import org.opendaylight.protocol.bgp.parser.spi.MessageSerializer; +import org.opendaylight.protocol.bgp.parser.spi.MessageUtil; import org.opendaylight.protocol.util.ByteArray; +import org.opendaylight.protocol.util.Ipv4Util; import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Prefix; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.Update; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.UpdateBuilder; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.update.NlriBuilder; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.update.PathAttributes; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.update.WithdrawnRoutesBuilder; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Update; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.UpdateBuilder; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.Attributes; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.update.Nlri; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.update.NlriBuilder; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.update.WithdrawnRoutes; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.update.WithdrawnRoutesBuilder; +import org.opendaylight.yangtools.yang.binding.Notification; import org.slf4j.Logger; import org.slf4j.LoggerFactory; -import com.google.common.base.Preconditions; - /** * LENGTH fields, that denote the length of the fields with variable length, have fixed SIZE. - * + * * @see BGP-4 Update Message Format - * */ -public class BGPUpdateMessageParser implements MessageParser { - public static final int TYPE = 2; - - private static Logger logger = LoggerFactory.getLogger(BGPUpdateMessageParser.class); - - /** - * Size of the withdrawn_routes_length field, in bytes. - */ - public static final int WITHDRAWN_ROUTES_LENGTH_SIZE = 2; - - /** - * Size of the total_path_attr_length field, in bytes. - */ - public static final int TOTAL_PATH_ATTR_LENGTH_SIZE = 2; - - private final AttributeRegistry reg; - - // Constructors ------------------------------------------------------- - public BGPUpdateMessageParser(final AttributeRegistry reg) { - this.reg = Preconditions.checkNotNull(reg); - } - - // Getters & setters -------------------------------------------------- - - @Override - public Update parseMessageBody(final byte[] body, final int messageLength) throws BGPDocumentedException { - if (body == null || body.length == 0) { - throw new IllegalArgumentException("Byte array cannot be null or empty."); - } - logger.trace("Started parsing of update message: {}", Arrays.toString(body)); - - int byteOffset = 0; - - final int withdrawnRoutesLength = ByteArray.bytesToInt(ByteArray.subByte(body, byteOffset, WITHDRAWN_ROUTES_LENGTH_SIZE)); - byteOffset += WITHDRAWN_ROUTES_LENGTH_SIZE; - - final UpdateBuilder eventBuilder = new UpdateBuilder(); - - if (withdrawnRoutesLength > 0) { - final List withdrawnRoutes = Ipv4Util.prefixListForBytes(ByteArray.subByte(body, byteOffset, withdrawnRoutesLength)); - byteOffset += withdrawnRoutesLength; - eventBuilder.setWithdrawnRoutes(new WithdrawnRoutesBuilder().setWithdrawnRoutes(withdrawnRoutes).build()); - } - - final int totalPathAttrLength = ByteArray.bytesToInt(ByteArray.subByte(body, byteOffset, TOTAL_PATH_ATTR_LENGTH_SIZE)); - byteOffset += TOTAL_PATH_ATTR_LENGTH_SIZE; - - if (withdrawnRoutesLength + totalPathAttrLength > body.length) { - throw new BGPDocumentedException("Message length inconsistent with withdrawn router length.", BGPError.MALFORMED_ATTR_LIST); - } - - if (withdrawnRoutesLength == 0 && totalPathAttrLength == 0) { - return eventBuilder.build(); - } - - try { - if (totalPathAttrLength > 0) { - final PathAttributes pathAttributes = reg.parseAttributes(ByteArray.subByte(body, byteOffset, - totalPathAttrLength)); - byteOffset += totalPathAttrLength; - eventBuilder.setPathAttributes(pathAttributes); - } - } catch (final BGPParsingException e) { - logger.warn("Could not parse BGP attributes: {}", e.getMessage(), e); - throw new BGPDocumentedException("Could not parse BGP attributes.", BGPError.MALFORMED_ATTR_LIST); - } - - final List nlri = Ipv4Util.prefixListForBytes(ByteArray.subByte(body, byteOffset, body.length - byteOffset)); - eventBuilder.setNlri(new NlriBuilder().setNlri(nlri).build()); - - logger.trace("Update message was parsed."); - return eventBuilder.build(); - } +public class BGPUpdateMessageParser implements MessageParser, MessageSerializer { + public static final int TYPE = 2; + + private static final Logger LOG = LoggerFactory.getLogger(BGPUpdateMessageParser.class); + + /** + * Size of the withdrawn_routes_length field, in bytes. + */ + public static final int WITHDRAWN_ROUTES_LENGTH_SIZE = 2; + /** + * Size of the total_path_attr_length field, in bytes. + */ + public static final int TOTAL_PATH_ATTR_LENGTH_SIZE = 2; + + private final AttributeRegistry reg; + + // Constructors ------------------------------------------------------- + public BGPUpdateMessageParser(final AttributeRegistry reg) { + this.reg = Preconditions.checkNotNull(reg); + } + + // Getters & setters -------------------------------------------------- + + @Override + public Update parseMessageBody(final ByteBuf buffer, final int messageLength) throws BGPDocumentedException { + Preconditions.checkArgument(buffer != null && buffer.readableBytes() != 0, "Byte array cannot be null or empty."); + LOG.trace("Started parsing of update message: {}", ByteBufUtil.hexDump(buffer)); + + final int withdrawnRoutesLength = buffer.readUnsignedShort(); + final UpdateBuilder eventBuilder = new UpdateBuilder(); + + if (withdrawnRoutesLength > 0) { + final List withdrawnRoutes = Ipv4Util.prefixListForBytes(ByteArray.readBytes(buffer, withdrawnRoutesLength)); + eventBuilder.setWithdrawnRoutes(new WithdrawnRoutesBuilder().setWithdrawnRoutes(withdrawnRoutes).build()); + } + final int totalPathAttrLength = buffer.readUnsignedShort(); + + if (withdrawnRoutesLength == 0 && totalPathAttrLength == 0) { + return eventBuilder.build(); + } + if (totalPathAttrLength > 0) { + try { + final Attributes pathAttributes = this.reg.parseAttributes(buffer.readSlice(totalPathAttrLength)); + eventBuilder.setAttributes(pathAttributes); + } catch (final BGPParsingException | RuntimeException e) { + // Catch everything else and turn it into a BGPDocumentedException + LOG.warn("Could not parse BGP attributes", e); + throw new BGPDocumentedException("Could not parse BGP attributes.", BGPError.MALFORMED_ATTR_LIST, e); + } + } + final List nlri = Ipv4Util.prefixListForBytes(ByteArray.readAllBytes(buffer)); + if (nlri != null && !nlri.isEmpty()) { + eventBuilder.setNlri(new NlriBuilder().setNlri(nlri).build()); + } + final Update msg = eventBuilder.build(); + LOG.debug("BGP Update message was parsed {}.", msg); + return msg; + } + + @Override + public void serializeMessage(final Notification message, final ByteBuf bytes) { + Preconditions.checkArgument(message instanceof Update, "BGPUpdate message cannot be null"); + LOG.trace("Started serializing update message: {}", message); + final Update update = (Update) message; + + final ByteBuf messageBody = Unpooled.buffer(); + final WithdrawnRoutes withdrawnRoutes = update.getWithdrawnRoutes(); + if (withdrawnRoutes != null) { + final ByteBuf withdrawnRoutesBuf = Unpooled.buffer(); + for (final Ipv4Prefix prefix : withdrawnRoutes.getWithdrawnRoutes()) { + withdrawnRoutesBuf.writeBytes(Ipv4Util.bytesForPrefixBegin(prefix)); + } + messageBody.writeShort(withdrawnRoutesBuf.writerIndex()); + messageBody.writeBytes(withdrawnRoutesBuf); + } else { + messageBody.writeZero(WITHDRAWN_ROUTES_LENGTH_SIZE); + } + if (update.getAttributes() != null) { + final ByteBuf pathAttributesBuf = Unpooled.buffer(); + this.reg.serializeAttribute(update.getAttributes(), pathAttributesBuf); + messageBody.writeShort(pathAttributesBuf.writerIndex()); + messageBody.writeBytes(pathAttributesBuf); + } else { + messageBody.writeZero(TOTAL_PATH_ATTR_LENGTH_SIZE); + } + final Nlri nlri = update.getNlri(); + if (nlri != null) { + for (final Ipv4Prefix prefix : nlri.getNlri()) { + messageBody.writeBytes(Ipv4Util.bytesForPrefixBegin(prefix)); + } + } + LOG.trace("Update message serialized to {}", ByteBufUtil.hexDump(messageBody)); + MessageUtil.formatMessage(TYPE, messageBody, bytes); + } }