/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package dexfuzz.rawdex.formats;

import dexfuzz.rawdex.DexRandomAccessFile;
import dexfuzz.rawdex.Instruction;
import dexfuzz.rawdex.Opcode;
import dexfuzz.rawdex.OpcodeInfo;

import java.io.IOException;

public class Format21c extends Format2 implements ContainsVRegs, ContainsPoolIndex {
  @Override
  public void writeToFile(DexRandomAccessFile file, Instruction insn) throws IOException {
    file.writeByte((byte) insn.info.value);
    file.writeByte((byte) insn.vregA);
    file.writeUShort((short) insn.vregB);
    return;
  }

  @Override
  public long getA(byte[] raw) throws IOException {
    return RawInsnHelper.getUnsignedByteFromByte(raw, 1);
  }

  @Override
  public long getB(byte[] raw) throws IOException {
    return RawInsnHelper.getUnsignedShortFromTwoBytes(raw, 2);
  }

  @Override
  public long getC(byte[] raw) throws IOException {
    return (long) 0;
  }

  @Override
  public int getVRegCount() {
    return 1;
  }

  @Override
  public int getPoolIndex(Instruction insn) {
    return (int) insn.vregB;
  }

  @Override
  public void setPoolIndex(Instruction insn, int poolIndex) {
    insn.vregB = poolIndex;
  }

  @Override
  public PoolIndexKind getPoolIndexKind(OpcodeInfo info) {
    if (info.opcode == Opcode.CONST_STRING) {
      return PoolIndexKind.String;
    }
    if (info.opcode == Opcode.CONST_CLASS
        || info.opcode == Opcode.CHECK_CAST
        || info.opcode == Opcode.NEW_INSTANCE) {
      return PoolIndexKind.Type;
    }
    // everything else is static field accesses
    return PoolIndexKind.Field;
  }
}