instruction_test.cc

// Copyright 2011-2024 Google LLC
//
// 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
//
//     https://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.

#include "third_party/zynamics/bindiff/instruction.h"

#include "gtest/gtest.h"

namespace security::bindiff {
namespace {

TEST(InstructionTest, Instruction) {
  Instruction::Cache cache;
  Instruction instruction(&cache, 0xbaadf00dbaadf00d, "mov", 1234);
  EXPECT_STREQ("mov", instruction.GetMnemonic(&cache).c_str());
  EXPECT_EQ(1234, instruction.GetPrime());
  EXPECT_EQ(0xbaadf00dbaadf00d, instruction.GetAddress());
}

TEST(InstructionTest, Instructions) {
  Instruction::Cache cache;
  Instructions instructions;
  instructions.emplace_back(&cache, 0x1000000010000000, "one", 1234);
  instructions.emplace_back(&cache, 0x1000000010000001, "two", 1235);
  instructions.emplace_back(&cache, 0x1000000010000005, "three", 1236);

  Instructions instructions2;
  instructions2.emplace_back(&cache, 0x1000000010012000, "one", 1232);
  instructions2.emplace_back(&cache, 0x1000000010012302, "one", 1234);
  instructions2.emplace_back(&cache, 0x1000000010033300, "one", 1237);
  instructions2.emplace_back(&cache, 0x1000000010112334, "two", 1235);
  instructions2.emplace_back(&cache, 0x1000000010234205, "three", 1236);
  instructions2.emplace_back(&cache, 0x1000000010234206, "three", 1237);
  instructions2.emplace_back(&cache, 0x1000000010234207, "three", 1236);
  InstructionMatches matches;
  ComputeLcs(instructions.begin(), instructions.end(), instructions2.begin(),
             instructions2.end(), matches);
  EXPECT_EQ(3, matches.size());
  EXPECT_EQ(0x1000000010000000, matches[0].first->GetAddress());
  EXPECT_EQ(0x1000000010012302, matches[0].second->GetAddress());
  EXPECT_EQ(0x1000000010000001, matches[1].first->GetAddress());
  EXPECT_EQ(0x1000000010112334, matches[1].second->GetAddress());
  EXPECT_EQ(0x1000000010000005, matches[2].first->GetAddress());
  EXPECT_EQ(0x1000000010234207, matches[2].second->GetAddress());
  EXPECT_EQ(1236, matches[2].second->GetPrime());
}

TEST(InstructionTest, LcsEmpty) {
  Instructions instructions1;
  Instructions instructions2;
  InstructionMatches matches;
  ComputeLcs(instructions1.begin(), instructions1.end(), instructions2.begin(),
             instructions2.end(), matches);
  EXPECT_EQ(0, matches.size());
}

TEST(InstructionTest, LcsEmptyOneSided) {
  Instructions instructions1;
  Instruction::Cache cache;
  instructions1.emplace_back(&cache, 0x1000000010000000, "one", 1234);
  instructions1.emplace_back(&cache, 0x1000000010000001, "two", 1235);
  instructions1.emplace_back(&cache, 0x1000000010000005, "three", 1236);

  Instructions instructions2;
  InstructionMatches matches;
  ComputeLcs(instructions1.begin(), instructions1.end(), instructions2.begin(),
             instructions2.end(), matches);
  EXPECT_EQ(0, matches.size());
}

TEST(InstructionTest, CommonPrefix) {
  Instructions instructions1;
  Instruction::Cache cache;
  instructions1.emplace_back(&cache, 0x1000000010000000, "one", 1234);
  instructions1.emplace_back(&cache, 0x1000000010000001, "two", 1235);
  instructions1.emplace_back(&cache, 0x1000000010000005, "three", 1236);

  Instructions instructions2;
  instructions2.emplace_back(&cache, 0x1000000010000000, "one", 1234);
  instructions2.emplace_back(&cache, 0x1000000010000001, "two", 1235);
  instructions2.emplace_back(&cache, 0x1000000010000005, "unmatched", 7777);
  InstructionMatches matches;
  ComputeLcs(instructions1.begin(), instructions1.end(), instructions2.begin(),
             instructions2.end(), matches);
  EXPECT_EQ(2, matches.size());
  EXPECT_EQ(0x1000000010000000, matches[0].first->GetAddress());
  EXPECT_EQ(0x1000000010000000, matches[0].second->GetAddress());
  EXPECT_EQ(0x1000000010000001, matches[1].first->GetAddress());
  EXPECT_EQ(0x1000000010000001, matches[1].second->GetAddress());
}

}  // namespace
}  // namespace security::bindiff