courgette_flow.cc 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "courgette/courgette_flow.h"
  5. #include <stdarg.h>
  6. #include <memory>
  7. #include "base/check.h"
  8. #include "base/files/file_path.h"
  9. #include "base/memory/ptr_util.h"
  10. #include "base/notreached.h"
  11. #include "base/strings/stringprintf.h"
  12. #include "courgette/assembly_program.h"
  13. #include "courgette/disassembler.h"
  14. #include "courgette/encoded_program.h"
  15. #include "courgette/program_detector.h"
  16. namespace courgette {
  17. /******** CourgetteFlow::Data ********/
  18. CourgetteFlow::Data::Data() = default;
  19. CourgetteFlow::Data::~Data() = default;
  20. /******** CourgetteFlow ********/
  21. CourgetteFlow::CourgetteFlow() = default;
  22. CourgetteFlow::~CourgetteFlow() = default;
  23. // static
  24. const char* CourgetteFlow::name(Group group) {
  25. switch (group) {
  26. case ONLY:
  27. return "input";
  28. case OLD:
  29. return "'old' input";
  30. case NEW:
  31. return "'new' input";
  32. default:
  33. NOTREACHED();
  34. break;
  35. }
  36. return nullptr;
  37. }
  38. CourgetteFlow::Data* CourgetteFlow::data(Group group) {
  39. switch (group) {
  40. case ONLY:
  41. return &data_only_;
  42. case OLD:
  43. return &data_old_;
  44. case NEW:
  45. return &data_new_;
  46. default:
  47. NOTREACHED();
  48. break;
  49. }
  50. return nullptr;
  51. }
  52. bool CourgetteFlow::ok() {
  53. return status_ == C_OK;
  54. }
  55. bool CourgetteFlow::failed() {
  56. return status_ != C_OK;
  57. }
  58. Status CourgetteFlow::status() {
  59. return status_;
  60. }
  61. const std::string& CourgetteFlow::message() {
  62. return message_;
  63. }
  64. void CourgetteFlow::ReadSourceStreamSetFromBuffer(Group group,
  65. const BasicBuffer& buffer) {
  66. if (failed())
  67. return;
  68. Data* d = data(group);
  69. if (!check(d->sources.Init(buffer.data(), buffer.length()),
  70. C_GENERAL_ERROR)) {
  71. setMessage("Cannot read %s as SourceStreamSet.", name(group));
  72. }
  73. }
  74. void CourgetteFlow::ReadDisassemblerFromBuffer(Group group,
  75. const BasicBuffer& buffer) {
  76. if (failed())
  77. return;
  78. Data* d = data(group);
  79. d->disassembler = DetectDisassembler(buffer.data(), buffer.length());
  80. if (!check(d->disassembler.get() != nullptr, C_INPUT_NOT_RECOGNIZED))
  81. setMessage("Cannot detect program for %s.", name(group));
  82. }
  83. void CourgetteFlow::ReadEncodedProgramFromSourceStreamSet(
  84. Group group,
  85. SourceStreamSet* opt_sources /* nullptr */) {
  86. if (failed())
  87. return;
  88. Data* d = data(group);
  89. SourceStreamSet* sources = opt_sources ? opt_sources : &d->sources;
  90. if (!check(ReadEncodedProgram(sources, &d->encoded)))
  91. setMessage("Cannot read %s as encoded program.", name(group));
  92. }
  93. void CourgetteFlow::CreateAssemblyProgramFromDisassembler(Group group,
  94. bool annotate) {
  95. if (failed())
  96. return;
  97. Data* d = data(group);
  98. d->program = d->disassembler->CreateProgram(annotate);
  99. if (!check(d->program.get() != nullptr, C_DISASSEMBLY_FAILED))
  100. setMessage("Cannot create AssemblyProgram for %s.", name(group));
  101. }
  102. void CourgetteFlow::CreateEncodedProgramFromDisassemblerAndAssemblyProgram(
  103. Group group) {
  104. if (failed())
  105. return;
  106. Data* d = data(group);
  107. d->encoded = std::make_unique<EncodedProgram>();
  108. if (!check(d->disassembler->DisassembleAndEncode(d->program.get(),
  109. d->encoded.get()))) {
  110. setMessage("Cannot disassemble to form EncodedProgram for %s.",
  111. name(group));
  112. }
  113. }
  114. void CourgetteFlow::WriteSinkStreamFromSinkStreamSet(Group group,
  115. SinkStream* sink) {
  116. DCHECK(sink);
  117. if (failed())
  118. return;
  119. if (!check(data(group)->sinks.CopyTo(sink), C_GENERAL_ERROR))
  120. setMessage("Cannnot combine serialized streams for %s.", name(group));
  121. }
  122. void CourgetteFlow::WriteSinkStreamSetFromEncodedProgram(
  123. Group group,
  124. SinkStreamSet* opt_sinks /* nullptr */) {
  125. if (failed())
  126. return;
  127. Data* d = data(group);
  128. SinkStreamSet* sinks = opt_sinks ? opt_sinks : &d->sinks;
  129. if (!check(WriteEncodedProgram(d->encoded.get(), sinks)))
  130. setMessage("Cannot serialize encoded %s.", name(group));
  131. }
  132. void CourgetteFlow::WriteExecutableFromEncodedProgram(Group group,
  133. SinkStream* sink) {
  134. DCHECK(sink);
  135. if (failed())
  136. return;
  137. if (!check(Assemble(data(group)->encoded.get(), sink)))
  138. setMessage("Cannot assemble %s.", name(group));
  139. }
  140. void CourgetteFlow::AdjustNewAssemblyProgramToMatchOld() {
  141. if (failed())
  142. return;
  143. if (!check(Adjust(*data_old_.program, data_new_.program.get())))
  144. setMessage("Cannot adjust %s to match %s.", name(OLD), name(NEW));
  145. }
  146. void CourgetteFlow::DestroyDisassembler(Group group) {
  147. if (failed())
  148. return;
  149. data(group)->disassembler.reset();
  150. }
  151. void CourgetteFlow::DestroyAssemblyProgram(Group group) {
  152. if (failed())
  153. return;
  154. data(group)->program.reset();
  155. }
  156. void CourgetteFlow::DestroyEncodedProgram(Group group) {
  157. if (failed())
  158. return;
  159. data(group)->encoded.reset();
  160. }
  161. bool CourgetteFlow::check(Status new_status) {
  162. if (new_status == C_OK)
  163. return true;
  164. status_ = new_status;
  165. return false;
  166. }
  167. bool CourgetteFlow::check(bool success, Status failure_mode) {
  168. if (success)
  169. return true;
  170. status_ = failure_mode;
  171. return false;
  172. }
  173. void CourgetteFlow::setMessage(const char* format, ...) {
  174. va_list args;
  175. va_start(args, format);
  176. message_ = base::StringPrintV(format, args);
  177. va_end(args);
  178. }
  179. } // namespace courgette