1f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org/************************************************************************** 2f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * 3f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * Copyright 2010 Luca Barbieri 4f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * 5f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * Permission is hereby granted, free of charge, to any person obtaining 6f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * a copy of this software and associated documentation files (the 7f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * "Software"), to deal in the Software without restriction, including 8f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * without limitation the rights to use, copy, modify, merge, publish, 9f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * distribute, sublicense, and/or sell copies of the Software, and to 10f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * permit persons to whom the Software is furnished to do so, subject to 11f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * the following conditions: 12f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * 13f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * The above copyright notice and this permission notice (including the 14f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * next paragraph) shall be included in all copies or substantial 15f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * portions of the Software. 16f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * 17f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 18f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 19f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. 20f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * IN NO EVENT SHALL THE COPYRIGHT OWNER(S) AND/OR ITS SUPPLIERS BE 21f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION 22f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION 23f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 24f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * 25f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org **************************************************************************/ 26f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 27f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include <vector> 28f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include <set> 29f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "sm4.h" 30f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 31f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#define check(x) do {if(!(x)) return false;} while(0) 32f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 33f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgbool sm4_link_cf_insns(sm4_program& program) 34f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org{ 35f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if(program.cf_insn_linked.size()) 36f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org return true; 37f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 38f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org std::vector<int> cf_insn_linked; 39f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_insn_linked.resize(program.insns.size()); 40f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org memset(&cf_insn_linked[0], 0xff, cf_insn_linked.size() * sizeof(int)); 41f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org std::vector<unsigned> cf_stack; 42f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org for(unsigned insn_num = 0; insn_num < program.insns.size(); ++insn_num) 43f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org { 44f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org unsigned v; 45f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org switch(program.insns[insn_num]->opcode) 46f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org { 47f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_LOOP: 48f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_stack.push_back(insn_num); 49f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org break; 50f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_ENDLOOP: 51f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(!cf_stack.empty()); 52f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org v = cf_stack.back(); 53f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(program.insns[v]->opcode == SM4_OPCODE_LOOP); 54f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_insn_linked[v] = insn_num; 55f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_insn_linked[insn_num] = v; 56f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_stack.pop_back(); 57f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org break; 58f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_IF: 59f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_SWITCH: 60f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_insn_linked[insn_num] = insn_num; // later changed 61f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_stack.push_back(insn_num); 62f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org break; 63f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_ELSE: 64f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_CASE: 65f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(!cf_stack.empty()); 66f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org v = cf_stack.back(); 67f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if(program.insns[insn_num]->opcode == SM4_OPCODE_ELSE) 68f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(program.insns[v]->opcode == SM4_OPCODE_IF); 69f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else 70f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(program.insns[v]->opcode == SM4_OPCODE_SWITCH || program.insns[v]->opcode == SM4_OPCODE_CASE); 71f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_insn_linked[insn_num] = cf_insn_linked[v]; // later changed 72f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_insn_linked[v] = insn_num; 73f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_stack.back() = insn_num; 74f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org break; 75f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_ENDSWITCH: 76f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_ENDIF: 77f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(!cf_stack.empty()); 78f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org v = cf_stack.back(); 79f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if(program.insns[insn_num]->opcode == SM4_OPCODE_ENDIF) 80f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(program.insns[v]->opcode == SM4_OPCODE_IF || program.insns[v]->opcode == SM4_OPCODE_ELSE); 81f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else 82f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(program.insns[v]->opcode == SM4_OPCODE_SWITCH || program.insns[v]->opcode == SM4_OPCODE_CASE); 83f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_insn_linked[insn_num] = cf_insn_linked[v]; 84f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_insn_linked[v] = insn_num; 85f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org cf_stack.pop_back(); 86f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org break; 87f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 88f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 89f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org check(cf_stack.empty()); 90f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org program.cf_insn_linked.swap(cf_insn_linked); 91f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org return true; 92f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org} 93f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 94f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgbool sm4_find_labels(sm4_program& program) 95f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org{ 96f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if(program.labels_found) 97f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org return true; 98f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 99f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org std::vector<int> labels; 100f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org for(unsigned insn_num = 0; insn_num < program.insns.size(); ++insn_num) 101f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org { 102f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org switch(program.insns[insn_num]->opcode) 103f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org { 104f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org case SM4_OPCODE_LABEL: 105f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if(program.insns[insn_num]->num_ops > 0) 106f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org { 107f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org sm4_op& op = *program.insns[insn_num]->ops[0]; 108f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if(op.file == SM4_FILE_LABEL && op.has_simple_index()) 109f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org { 110f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org unsigned idx = (unsigned)op.indices[0].disp; 111f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if(idx >= labels.size()) 112f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org labels.resize(idx + 1); 113f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org labels[idx] = insn_num; 114f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 115f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 116f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org break; 117f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 118f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 119f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org program.label_to_insn_num.swap(labels); 120f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org program.labels_found = true; 121f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org return true; 122f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org} 123