Пример #1
0
  @Override
  public SmbData parseResponse(SmbHeader h, Buffer b, SmbSession session) {
    Transaction2Response data = new Transaction2Response();
    Trans2SubcommandMapper mapper = new Trans2SubcommandMapper();
    byte[] setup;
    byte[] pad1;
    byte[] pad2;
    byte[] parameters;
    byte[] buff;
    data.setWordCount(b.get());
    if (data.getWordCount() != 0x00) {
      data.setTotalParameterCount(ByteOrderConverter.swap(b.getShort()));
      data.setTotalDataCount(ByteOrderConverter.swap(b.getShort()));
      data.setReserved1(ByteOrderConverter.swap(b.getShort()));
      data.setParameterCount(ByteOrderConverter.swap(b.getShort()));
      data.setParameterOffset(ByteOrderConverter.swap(b.getShort()));
      data.setParameterDisplacement(ByteOrderConverter.swap(b.getShort()));
      data.setDataCount(ByteOrderConverter.swap(b.getShort()));
      data.setDataOffset(ByteOrderConverter.swap(b.getShort()));
      data.setDataDisplacement(ByteOrderConverter.swap(b.getShort()));
      data.setSetupCount(b.get());
      data.setResreved2(b.get());
      setup = new byte[data.getSetupCount() * 2];
      b.gets(setup);
      data.setSetup(setup);
    } // it final response
    data.setByteCount(ByteOrderConverter.swap(b.getShort()));
    if (b.readableBytes() != data.getByteCount()) {
      data.setMalformed(true);
      return data;
    }
    if (data.getByteCount() != 0x00) {
      if (data.getParameterCount() != 0) {
        pad1 = new byte[data.getParameterOffset() - b.position()];
        if (pad1.length != 0) {
          b.gets(pad1);
          data.setPad1(pad1);
        }
        parameters = new byte[data.getParameterCount()];
        b.gets(parameters);
        data.setTrans2Parameter(parameters);
      }
      if (data.getDataCount() != 0) {
        pad2 = new byte[data.getDataOffset() - b.position()];
        if (pad2.length != 0) {
          b.gets(pad2);
          data.setPad2(pad2);
        }
        buff = new byte[data.getDataCount()];
        b.gets(buff);
        data.setTrans2Data(buff);
      }
    }
    Buffer setupBuffer = new ChainBuffer();
    Buffer parameterBuffer = new ChainBuffer();
    Buffer dataBuffer = new ChainBuffer();
    setupBuffer.addLast(data.getSetup());
    parameterBuffer.addLast(data.getTrans2Parameter());
    dataBuffer.addLast(data.getTrans2Data());
    if (setupBuffer.readableBytes() + parameterBuffer.readableBytes() + dataBuffer.readableBytes()
        == 0) {
      return data;
    }
    // setupBuffer.mark();
    TransParser parser = mapper.getParser(session.getSessionTrans2Command());
    if (parser == null) {
      return data;
    }

    // add buffer;
    if (data.getTotalDataCount() != dataBuffer.readableBytes()) {
      if (session.getTransBuffer().readableBytes() == 0) {
        session.setParameterBuffer(parameterBuffer);
        session.setTransBuffer(dataBuffer);
      } else if (session.getTransBuffer().readableBytes() != 0) {
        session.addTransBuffer(dataBuffer);
        if (data.getTotalDataCount() == session.getTransBuffer().readableBytes()) {
          data.setTransaction2Data(
              parser.parseResponse(
                  setupBuffer, session.getParameterBuffer(), session.getTransBuffer(), session));
          session.getTransBuffer().skip(session.getTransBuffer().readableBytes());
          session.getParameterBuffer().discardReadBytes();
        }
      }
      return data;
    } else {
      data.setTransaction2Data(
          parser.parseResponse(setupBuffer, parameterBuffer, dataBuffer, session));
    }
    return data;
  }
Пример #2
0
  @Override
  public SmbData parseRequest(SmbHeader h, Buffer b, SmbSession session) {
    Transaction2Request data = new Transaction2Request();
    Trans2SubcommandMapper mapper = new Trans2SubcommandMapper();
    byte[] setup;
    byte[] pad1;
    byte[] pad2;
    byte[] parameters;
    byte[] buff;
    data.setWordCount(b.get());
    if (data.getWordCount() >= 0x0E) {
      data.setTotalParameterCount(ByteOrderConverter.swap(b.getShort()));
      data.setTotalDataCount(ByteOrderConverter.swap(b.getShort()));
      data.setMaxParameterCount(ByteOrderConverter.swap(b.getShort()));
      data.setMaxDataCount(ByteOrderConverter.swap(b.getShort()));
      data.setMaxSetupCount(b.get());
      data.setReserved1(b.get());
      data.setFlags(ByteOrderConverter.swap(b.getShort()));
      data.setTimeout(ByteOrderConverter.swap(b.getInt()));
      data.setReserved2(ByteOrderConverter.swap(b.getShort()));
      data.setParameterCount(ByteOrderConverter.swap(b.getShort()));
      data.setParameterOffset(ByteOrderConverter.swap(b.getShort()));
      data.setDataCount(ByteOrderConverter.swap(b.getShort()));
      data.setDataOffset(ByteOrderConverter.swap(b.getShort()));
      data.setSetupCount(b.get());
      data.setReserved3(b.get());
      setup = new byte[data.getSetupCount() * 2];
      b.gets(setup);
      data.setSetup(setup);
    } else {
      data.setMalformed(true);
      b.skip(data.getWordCount() * 2);
    }
    data.setByteCount(ByteOrderConverter.swap(b.getShort()));
    if (b.readableBytes() != data.getByteCount()) {
      data.setMalformed(true);
      return data;
    } else if (data.isMalformed()) {
      return data;
    }
    data.setName(NetBiosNameCodec.readOemName(b));

    if (data.getParameterCount() != 0) {
      pad1 = new byte[data.getParameterOffset() - b.position()];
      if (pad1.length != 0) {
        b.gets(pad1);
        data.setPad1(pad1);
      }
      parameters = new byte[data.getParameterCount()];
      b.gets(parameters);
      data.setTrans2Parameters(parameters);
    }
    if (data.getDataCount() != 0) {
      pad2 = new byte[data.getDataOffset() - b.position()];
      if (pad2.length != 0) {
        b.gets(pad2);
        data.setPad2(pad2);
      }
      buff = new byte[data.getDataCount()];
      b.gets(buff);
      data.setTrans2Data(buff);
    }

    Buffer setupBuffer = new ChainBuffer();
    Buffer parameterBuffer = new ChainBuffer();
    Buffer dataBuffer = new ChainBuffer();
    setupBuffer.addLast(data.getSetup());
    parameterBuffer.addLast(data.getTrans2Parameters());
    dataBuffer.addLast(data.getTrans2Data());
    setupBuffer.mark();
    TransParser parser =
        mapper.getParser(
            Transaction2Command.parse(ByteOrderConverter.swap(setupBuffer.getShort())));
    setupBuffer.reset();
    data.setTransaction2Data(parser.parseRequest(setupBuffer, parameterBuffer, dataBuffer));
    return data;
  }