io.vertx.mssqlclient.impl.codec.TdsPacketDecoder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertx-mssql-client Show documentation

The Reactive MSSQL Client

There is a newer version: 5.0.0.CR3

/*
 * Copyright (c) 2011-2021 Contributors to the Eclipse Foundation
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
 * which is available at https://www.apache.org/licenses/LICENSE-2.0.
 *
 * SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
 */

package io.vertx.mssqlclient.impl.codec;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.LengthFieldBasedFrameDecoder;

import static io.vertx.mssqlclient.MSSQLConnectOptions.MAX_PACKET_SIZE;
import static io.vertx.mssqlclient.impl.codec.TdsPacket.PACKET_HEADER_SIZE;

public class TdsPacketDecoder extends LengthFieldBasedFrameDecoder {

  public TdsPacketDecoder() {
    super(MAX_PACKET_SIZE, 2, 2, -4, 0);
  }

  @Override
  protected TdsPacket decode(ChannelHandlerContext ctx, ByteBuf in) throws Exception {
    ByteBuf byteBuf = (ByteBuf) super.decode(ctx, in);
    if (byteBuf == null) {
      return null;
    }

    short type = byteBuf.getUnsignedByte(0);
    short status = byteBuf.getUnsignedByte(1);
    int length = byteBuf.getUnsignedShort(2);

    byteBuf.skipBytes(PACKET_HEADER_SIZE);

    return new TdsPacket(type, status, length, byteBuf);
  }
}