blob: c443519b77d0c90adc3e737b9d85e726ae61865a [file] [log] [blame]
// statements.cc -- Go frontend statements.
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
#include "go-system.h"
#include <gmp.h>
#ifndef ENABLE_BUILD_WITH_CXX
extern "C"
{
#endif
#include "intl.h"
#include "tree.h"
#include "gimple.h"
#include "convert.h"
#include "tree-iterator.h"
#include "tree-flow.h"
#include "real.h"
#ifndef ENABLE_BUILD_WITH_CXX
}
#endif
#include "go-c.h"
#include "types.h"
#include "expressions.h"
#include "gogo.h"
#include "statements.h"
// Class Statement.
Statement::Statement(Statement_classification classification,
source_location location)
: classification_(classification), location_(location)
{
}
Statement::~Statement()
{
}
// Traverse the tree. The work of walking the components is handled
// by the subclasses.
int
Statement::traverse(Block* block, size_t* pindex, Traverse* traverse)
{
if (this->classification_ == STATEMENT_ERROR)
return TRAVERSE_CONTINUE;
unsigned int traverse_mask = traverse->traverse_mask();
if ((traverse_mask & Traverse::traverse_statements) != 0)
{
int t = traverse->statement(block, pindex, this);
if (t == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
else if (t == TRAVERSE_SKIP_COMPONENTS)
return TRAVERSE_CONTINUE;
}
// No point in checking traverse_mask here--a statement may contain
// other blocks or statements, and if we got here we always want to
// walk them.
return this->do_traverse(traverse);
}
// Traverse the contents of a statement.
int
Statement::traverse_contents(Traverse* traverse)
{
return this->do_traverse(traverse);
}
// Traverse assignments.
bool
Statement::traverse_assignments(Traverse_assignments* tassign)
{
if (this->classification_ == STATEMENT_ERROR)
return false;
return this->do_traverse_assignments(tassign);
}
// Traverse an expression in a statement. This is a helper function
// for child classes.
int
Statement::traverse_expression(Traverse* traverse, Expression** expr)
{
if ((traverse->traverse_mask()
& (Traverse::traverse_types | Traverse::traverse_expressions)) == 0)
return TRAVERSE_CONTINUE;
return Expression::traverse(expr, traverse);
}
// Traverse an expression list in a statement. This is a helper
// function for child classes.
int
Statement::traverse_expression_list(Traverse* traverse,
Expression_list* expr_list)
{
if (expr_list == NULL)
return TRAVERSE_CONTINUE;
if ((traverse->traverse_mask()
& (Traverse::traverse_types | Traverse::traverse_expressions)) == 0)
return TRAVERSE_CONTINUE;
return expr_list->traverse(traverse);
}
// Traverse a type in a statement. This is a helper function for
// child classes.
int
Statement::traverse_type(Traverse* traverse, Type* type)
{
if ((traverse->traverse_mask()
& (Traverse::traverse_types | Traverse::traverse_expressions)) == 0)
return TRAVERSE_CONTINUE;
return Type::traverse(type, traverse);
}
// Set type information for unnamed constants. This is really done by
// the child class.
void
Statement::determine_types()
{
this->do_determine_types();
}
// If this is a thunk statement, return it.
Thunk_statement*
Statement::thunk_statement()
{
Thunk_statement* ret = this->convert<Thunk_statement, STATEMENT_GO>();
if (ret == NULL)
ret = this->convert<Thunk_statement, STATEMENT_DEFER>();
return ret;
}
// Get a tree for a Statement. This is really done by the child
// class.
tree
Statement::get_tree(Translate_context* context)
{
if (this->classification_ == STATEMENT_ERROR)
return error_mark_node;
return this->do_get_tree(context);
}
// Build tree nodes and set locations.
tree
Statement::build_stmt_1(int tree_code_value, tree node)
{
tree ret = build1(static_cast<tree_code>(tree_code_value),
void_type_node, node);
SET_EXPR_LOCATION(ret, this->location_);
return ret;
}
// Note that this statement is erroneous. This is called by children
// when they discover an error.
void
Statement::set_is_error()
{
this->classification_ = STATEMENT_ERROR;
}
// For children to call to report an error conveniently.
void
Statement::report_error(const char* msg)
{
error_at(this->location_, "%s", msg);
this->set_is_error();
}
// An error statement, used to avoid crashing after we report an
// error.
class Error_statement : public Statement
{
public:
Error_statement(source_location location)
: Statement(STATEMENT_ERROR, location)
{ }
protected:
int
do_traverse(Traverse*)
{ return TRAVERSE_CONTINUE; }
tree
do_get_tree(Translate_context*)
{ gcc_unreachable(); }
};
// Make an error statement.
Statement*
Statement::make_error_statement(source_location location)
{
return new Error_statement(location);
}
// Class Variable_declaration_statement.
Variable_declaration_statement::Variable_declaration_statement(
Named_object* var)
: Statement(STATEMENT_VARIABLE_DECLARATION, var->var_value()->location()),
var_(var)
{
}
// We don't actually traverse the variable here; it was traversed
// while traversing the Block.
int
Variable_declaration_statement::do_traverse(Traverse*)
{
return TRAVERSE_CONTINUE;
}
// Traverse the assignments in a variable declaration. Note that this
// traversal is different from the usual traversal.
bool
Variable_declaration_statement::do_traverse_assignments(
Traverse_assignments* tassign)
{
tassign->initialize_variable(this->var_);
return true;
}
// Return the tree for a variable declaration.
tree
Variable_declaration_statement::do_get_tree(Translate_context* context)
{
tree val = this->var_->get_tree(context->gogo(), context->function());
if (val == error_mark_node || TREE_TYPE(val) == error_mark_node)
return error_mark_node;
Variable* variable = this->var_->var_value();
tree init = variable->get_init_tree(context->gogo(), context->function());
if (init == error_mark_node)
return error_mark_node;
// If this variable lives on the heap, we need to allocate it now.
if (!variable->is_in_heap())
{
DECL_INITIAL(val) = init;
return this->build_stmt_1(DECL_EXPR, val);
}
else
{
gcc_assert(TREE_CODE(val) == INDIRECT_REF);
tree decl = TREE_OPERAND(val, 0);
gcc_assert(TREE_CODE(decl) == VAR_DECL);
tree type = TREE_TYPE(decl);
gcc_assert(POINTER_TYPE_P(type));
tree size = TYPE_SIZE_UNIT(TREE_TYPE(type));
tree space = context->gogo()->allocate_memory(variable->type(), size,
this->location());
space = fold_convert(TREE_TYPE(decl), space);
DECL_INITIAL(decl) = space;
return build2(COMPOUND_EXPR, void_type_node,
this->build_stmt_1(DECL_EXPR, decl),
build2(MODIFY_EXPR, void_type_node, val, init));
}
}
// Make a variable declaration.
Statement*
Statement::make_variable_declaration(Named_object* var)
{
return new Variable_declaration_statement(var);
}
// Class Temporary_statement.
// Return the type of the temporary variable.
Type*
Temporary_statement::type() const
{
return this->type_ != NULL ? this->type_ : this->init_->type();
}
// Return the tree for the temporary variable.
tree
Temporary_statement::get_decl() const
{
if (this->decl_ == NULL)
{
gcc_assert(saw_errors());
return error_mark_node;
}
return this->decl_;
}
// Traversal.
int
Temporary_statement::do_traverse(Traverse* traverse)
{
if (this->type_ != NULL
&& this->traverse_type(traverse, this->type_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
if (this->init_ == NULL)
return TRAVERSE_CONTINUE;
else
return this->traverse_expression(traverse, &this->init_);
}
// Traverse assignments.
bool
Temporary_statement::do_traverse_assignments(Traverse_assignments* tassign)
{
if (this->init_ == NULL)
return false;
tassign->value(&this->init_, true, true);
return true;
}
// Determine types.
void
Temporary_statement::do_determine_types()
{
if (this->type_ != NULL && this->type_->is_abstract())
this->type_ = this->type_->make_non_abstract_type();
if (this->init_ != NULL)
{
if (this->type_ == NULL)
this->init_->determine_type_no_context();
else
{
Type_context context(this->type_, false);
this->init_->determine_type(&context);
}
}
if (this->type_ == NULL)
{
this->type_ = this->init_->type();
gcc_assert(!this->type_->is_abstract());
}
}
// Check types.
void
Temporary_statement::do_check_types(Gogo*)
{
if (this->type_ != NULL && this->init_ != NULL)
{
std::string reason;
if (!Type::are_assignable(this->type_, this->init_->type(), &reason))
{
if (reason.empty())
error_at(this->location(), "incompatible types in assignment");
else
error_at(this->location(), "incompatible types in assignment (%s)",
reason.c_str());
this->set_is_error();
}
}
}
// Return a tree.
tree
Temporary_statement::do_get_tree(Translate_context* context)
{
gcc_assert(this->decl_ == NULL_TREE);
tree type_tree = this->type()->get_tree(context->gogo());
tree init_tree = (this->init_ == NULL
? NULL_TREE
: this->init_->get_tree(context));
if (type_tree == error_mark_node || init_tree == error_mark_node)
{
this->decl_ = error_mark_node;
return error_mark_node;
}
// We can only use create_tmp_var if the type is not addressable.
if (!TREE_ADDRESSABLE(type_tree))
{
this->decl_ = create_tmp_var(type_tree, "GOTMP");
DECL_SOURCE_LOCATION(this->decl_) = this->location();
}
else
{
gcc_assert(context->function() != NULL && context->block() != NULL);
tree decl = build_decl(this->location(), VAR_DECL,
create_tmp_var_name("GOTMP"),
type_tree);
DECL_ARTIFICIAL(decl) = 1;
DECL_IGNORED_P(decl) = 1;
TREE_USED(decl) = 1;
gcc_assert(current_function_decl != NULL_TREE);
DECL_CONTEXT(decl) = current_function_decl;
// We have to add this variable to the block so that it winds up
// in a BIND_EXPR.
tree block_tree = context->block_tree();
gcc_assert(block_tree != NULL_TREE);
DECL_CHAIN(decl) = BLOCK_VARS(block_tree);
BLOCK_VARS(block_tree) = decl;
this->decl_ = decl;
}
if (init_tree != NULL_TREE)
DECL_INITIAL(this->decl_) =
Expression::convert_for_assignment(context, this->type(),
this->init_->type(), init_tree,
this->location());
if (this->is_address_taken_)
TREE_ADDRESSABLE(this->decl_) = 1;
return this->build_stmt_1(DECL_EXPR, this->decl_);
}
// Make and initialize a temporary variable in BLOCK.
Temporary_statement*
Statement::make_temporary(Type* type, Expression* init,
source_location location)
{
return new Temporary_statement(type, init, location);
}
// An assignment statement.
class Assignment_statement : public Statement
{
public:
Assignment_statement(Expression* lhs, Expression* rhs,
source_location location)
: Statement(STATEMENT_ASSIGNMENT, location),
lhs_(lhs), rhs_(rhs)
{ }
protected:
int
do_traverse(Traverse* traverse);
bool
do_traverse_assignments(Traverse_assignments*);
void
do_determine_types();
void
do_check_types(Gogo*);
tree
do_get_tree(Translate_context*);
private:
// Left hand side--the lvalue.
Expression* lhs_;
// Right hand side--the rvalue.
Expression* rhs_;
};
// Traversal.
int
Assignment_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression(traverse, &this->lhs_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->traverse_expression(traverse, &this->rhs_);
}
bool
Assignment_statement::do_traverse_assignments(Traverse_assignments* tassign)
{
tassign->assignment(&this->lhs_, &this->rhs_);
return true;
}
// Set types for the assignment.
void
Assignment_statement::do_determine_types()
{
this->lhs_->determine_type_no_context();
Type_context context(this->lhs_->type(), false);
this->rhs_->determine_type(&context);
}
// Check types for an assignment.
void
Assignment_statement::do_check_types(Gogo*)
{
// The left hand side must be either addressable, a map index
// expression, or the blank identifier.
if (!this->lhs_->is_addressable()
&& this->lhs_->map_index_expression() == NULL
&& !this->lhs_->is_sink_expression())
{
if (!this->lhs_->type()->is_error_type())
this->report_error(_("invalid left hand side of assignment"));
return;
}
Type* lhs_type = this->lhs_->type();
Type* rhs_type = this->rhs_->type();
std::string reason;
if (!Type::are_assignable(lhs_type, rhs_type, &reason))
{
if (reason.empty())
error_at(this->location(), "incompatible types in assignment");
else
error_at(this->location(), "incompatible types in assignment (%s)",
reason.c_str());
this->set_is_error();
}
if (lhs_type->is_error_type()
|| rhs_type->is_error_type()
|| lhs_type->is_undefined()
|| rhs_type->is_undefined())
{
// Make sure we get the error for an undefined type.
lhs_type->base();
rhs_type->base();
this->set_is_error();
}
}
// Build a tree for an assignment statement.
tree
Assignment_statement::do_get_tree(Translate_context* context)
{
tree rhs_tree = this->rhs_->get_tree(context);
if (this->lhs_->is_sink_expression())
return rhs_tree;
tree lhs_tree = this->lhs_->get_tree(context);
if (lhs_tree == error_mark_node || rhs_tree == error_mark_node)
return error_mark_node;
rhs_tree = Expression::convert_for_assignment(context, this->lhs_->type(),
this->rhs_->type(), rhs_tree,
this->location());
if (rhs_tree == error_mark_node)
return error_mark_node;
return fold_build2_loc(this->location(), MODIFY_EXPR, void_type_node,
lhs_tree, rhs_tree);
}
// Make an assignment statement.
Statement*
Statement::make_assignment(Expression* lhs, Expression* rhs,
source_location location)
{
return new Assignment_statement(lhs, rhs, location);
}
// The Move_ordered_evals class is used to find any subexpressions of
// an expression that have an evaluation order dependency. It creates
// temporary variables to hold them.
class Move_ordered_evals : public Traverse
{
public:
Move_ordered_evals(Block* block)
: Traverse(traverse_expressions),
block_(block)
{ }
protected:
int
expression(Expression**);
private:
// The block where new temporary variables should be added.
Block* block_;
};
int
Move_ordered_evals::expression(Expression** pexpr)
{
// We have to look at subexpressions first.
if ((*pexpr)->traverse_subexpressions(this) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
if ((*pexpr)->must_eval_in_order())
{
source_location loc = (*pexpr)->location();
Temporary_statement* temp = Statement::make_temporary(NULL, *pexpr, loc);
this->block_->add_statement(temp);
*pexpr = Expression::make_temporary_reference(temp, loc);
}
return TRAVERSE_SKIP_COMPONENTS;
}
// An assignment operation statement.
class Assignment_operation_statement : public Statement
{
public:
Assignment_operation_statement(Operator op, Expression* lhs, Expression* rhs,
source_location location)
: Statement(STATEMENT_ASSIGNMENT_OPERATION, location),
op_(op), lhs_(lhs), rhs_(rhs)
{ }
protected:
int
do_traverse(Traverse*);
bool
do_traverse_assignments(Traverse_assignments*)
{ gcc_unreachable(); }
Statement*
do_lower(Gogo*, Block*);
tree
do_get_tree(Translate_context*)
{ gcc_unreachable(); }
private:
// The operator (OPERATOR_PLUSEQ, etc.).
Operator op_;
// Left hand side.
Expression* lhs_;
// Right hand side.
Expression* rhs_;
};
// Traversal.
int
Assignment_operation_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression(traverse, &this->lhs_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->traverse_expression(traverse, &this->rhs_);
}
// Lower an assignment operation statement to a regular assignment
// statement.
Statement*
Assignment_operation_statement::do_lower(Gogo*, Block* enclosing)
{
source_location loc = this->location();
// We have to evaluate the left hand side expression only once. We
// do this by moving out any expression with side effects.
Block* b = new Block(enclosing, loc);
Move_ordered_evals moe(b);
this->lhs_->traverse_subexpressions(&moe);
Expression* lval = this->lhs_->copy();
Operator op;
switch (this->op_)
{
case OPERATOR_PLUSEQ:
op = OPERATOR_PLUS;
break;
case OPERATOR_MINUSEQ:
op = OPERATOR_MINUS;
break;
case OPERATOR_OREQ:
op = OPERATOR_OR;
break;
case OPERATOR_XOREQ:
op = OPERATOR_XOR;
break;
case OPERATOR_MULTEQ:
op = OPERATOR_MULT;
break;
case OPERATOR_DIVEQ:
op = OPERATOR_DIV;
break;
case OPERATOR_MODEQ:
op = OPERATOR_MOD;
break;
case OPERATOR_LSHIFTEQ:
op = OPERATOR_LSHIFT;
break;
case OPERATOR_RSHIFTEQ:
op = OPERATOR_RSHIFT;
break;
case OPERATOR_ANDEQ:
op = OPERATOR_AND;
break;
case OPERATOR_BITCLEAREQ:
op = OPERATOR_BITCLEAR;
break;
default:
gcc_unreachable();
}
Expression* binop = Expression::make_binary(op, lval, this->rhs_, loc);
Statement* s = Statement::make_assignment(this->lhs_, binop, loc);
if (b->statements()->empty())
{
delete b;
return s;
}
else
{
b->add_statement(s);
return Statement::make_block_statement(b, loc);
}
}
// Make an assignment operation statement.
Statement*
Statement::make_assignment_operation(Operator op, Expression* lhs,
Expression* rhs, source_location location)
{
return new Assignment_operation_statement(op, lhs, rhs, location);
}
// A tuple assignment statement. This differs from an assignment
// statement in that the right-hand-side expressions are evaluated in
// parallel.
class Tuple_assignment_statement : public Statement
{
public:
Tuple_assignment_statement(Expression_list* lhs, Expression_list* rhs,
source_location location)
: Statement(STATEMENT_TUPLE_ASSIGNMENT, location),
lhs_(lhs), rhs_(rhs)
{ }
protected:
int
do_traverse(Traverse* traverse);
bool
do_traverse_assignments(Traverse_assignments*)
{ gcc_unreachable(); }
Statement*
do_lower(Gogo*, Block*);
tree
do_get_tree(Translate_context*)
{ gcc_unreachable(); }
private:
// Left hand side--a list of lvalues.
Expression_list* lhs_;
// Right hand side--a list of rvalues.
Expression_list* rhs_;
};
// Traversal.
int
Tuple_assignment_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression_list(traverse, this->lhs_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->traverse_expression_list(traverse, this->rhs_);
}
// Lower a tuple assignment. We use temporary variables to split it
// up into a set of single assignments.
Statement*
Tuple_assignment_statement::do_lower(Gogo*, Block* enclosing)
{
source_location loc = this->location();
Block* b = new Block(enclosing, loc);
// First move out any subexpressions on the left hand side. The
// right hand side will be evaluated in the required order anyhow.
Move_ordered_evals moe(b);
for (Expression_list::const_iterator plhs = this->lhs_->begin();
plhs != this->lhs_->end();
++plhs)
(*plhs)->traverse_subexpressions(&moe);
std::vector<Temporary_statement*> temps;
temps.reserve(this->lhs_->size());
Expression_list::const_iterator prhs = this->rhs_->begin();
for (Expression_list::const_iterator plhs = this->lhs_->begin();
plhs != this->lhs_->end();
++plhs, ++prhs)
{
gcc_assert(prhs != this->rhs_->end());
if ((*plhs)->is_error_expression()
|| (*plhs)->type()->is_error_type()
|| (*prhs)->is_error_expression()
|| (*prhs)->type()->is_error_type())
continue;
if ((*plhs)->is_sink_expression())
{
b->add_statement(Statement::make_statement(*prhs));
continue;
}
Temporary_statement* temp = Statement::make_temporary((*plhs)->type(),
*prhs, loc);
b->add_statement(temp);
temps.push_back(temp);
}
gcc_assert(prhs == this->rhs_->end());
prhs = this->rhs_->begin();
std::vector<Temporary_statement*>::const_iterator ptemp = temps.begin();
for (Expression_list::const_iterator plhs = this->lhs_->begin();
plhs != this->lhs_->end();
++plhs, ++prhs)
{
if ((*plhs)->is_error_expression()
|| (*plhs)->type()->is_error_type()
|| (*prhs)->is_error_expression()
|| (*prhs)->type()->is_error_type())
continue;
if ((*plhs)->is_sink_expression())
continue;
Expression* ref = Expression::make_temporary_reference(*ptemp, loc);
Statement* s = Statement::make_assignment(*plhs, ref, loc);
b->add_statement(s);
++ptemp;
}
gcc_assert(ptemp == temps.end());
return Statement::make_block_statement(b, loc);
}
// Make a tuple assignment statement.
Statement*
Statement::make_tuple_assignment(Expression_list* lhs, Expression_list* rhs,
source_location location)
{
return new Tuple_assignment_statement(lhs, rhs, location);
}
// A tuple assignment from a map index expression.
// v, ok = m[k]
class Tuple_map_assignment_statement : public Statement
{
public:
Tuple_map_assignment_statement(Expression* val, Expression* present,
Expression* map_index,
source_location location)
: Statement(STATEMENT_TUPLE_MAP_ASSIGNMENT, location),
val_(val), present_(present), map_index_(map_index)
{ }
protected:
int
do_traverse(Traverse* traverse);
bool
do_traverse_assignments(Traverse_assignments*)
{ gcc_unreachable(); }
Statement*
do_lower(Gogo*, Block*);
tree
do_get_tree(Translate_context*)
{ gcc_unreachable(); }
private:
// Lvalue which receives the value from the map.
Expression* val_;
// Lvalue which receives whether the key value was present.
Expression* present_;
// The map index expression.
Expression* map_index_;
};
// Traversal.
int
Tuple_map_assignment_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT
|| this->traverse_expression(traverse, &this->present_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->traverse_expression(traverse, &this->map_index_);
}
// Lower a tuple map assignment.
Statement*
Tuple_map_assignment_statement::do_lower(Gogo*, Block* enclosing)
{
source_location loc = this->location();
Map_index_expression* map_index = this->map_index_->map_index_expression();
if (map_index == NULL)
{
this->report_error(_("expected map index on right hand side"));
return Statement::make_error_statement(loc);
}
Map_type* map_type = map_index->get_map_type();
if (map_type == NULL)
return Statement::make_error_statement(loc);
Block* b = new Block(enclosing, loc);
// Move out any subexpressions to make sure that functions are
// called in the required order.
Move_ordered_evals moe(b);
this->val_->traverse_subexpressions(&moe);
this->present_->traverse_subexpressions(&moe);
// Copy the key value into a temporary so that we can take its
// address without pushing the value onto the heap.
// var key_temp KEY_TYPE = MAP_INDEX
Temporary_statement* key_temp =
Statement::make_temporary(map_type->key_type(), map_index->index(), loc);
b->add_statement(key_temp);
// var val_temp VAL_TYPE
Temporary_statement* val_temp =
Statement::make_temporary(map_type->val_type(), NULL, loc);
b->add_statement(val_temp);
// var present_temp bool
Temporary_statement* present_temp =
Statement::make_temporary(Type::lookup_bool_type(), NULL, loc);
b->add_statement(present_temp);
// func mapaccess2(hmap map[k]v, key *k, val *v) bool
source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("hmap", map_type, bloc));
Type* pkey_type = Type::make_pointer_type(map_type->key_type());
param_types->push_back(Typed_identifier("key", pkey_type, bloc));
Type* pval_type = Type::make_pointer_type(map_type->val_type());
param_types->push_back(Typed_identifier("val", pval_type, bloc));
Typed_identifier_list* ret_types = new Typed_identifier_list();
ret_types->push_back(Typed_identifier("", Type::lookup_bool_type(), bloc));
Function_type* fntype = Type::make_function_type(NULL, param_types,
ret_types, bloc);
Named_object* mapaccess2 =
Named_object::make_function_declaration("mapaccess2", NULL, fntype, bloc);
mapaccess2->func_declaration_value()->set_asm_name("runtime.mapaccess2");
// present_temp = mapaccess2(MAP, &key_temp, &val_temp)
Expression* func = Expression::make_func_reference(mapaccess2, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(map_index->map());
Expression* ref = Expression::make_temporary_reference(key_temp, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
ref = Expression::make_temporary_reference(val_temp, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
Expression* call = Expression::make_call(func, params, false, loc);
ref = Expression::make_temporary_reference(present_temp, loc);
Statement* s = Statement::make_assignment(ref, call, loc);
b->add_statement(s);
// val = val_temp
ref = Expression::make_temporary_reference(val_temp, loc);
s = Statement::make_assignment(this->val_, ref, loc);
b->add_statement(s);
// present = present_temp
ref = Expression::make_temporary_reference(present_temp, loc);
s = Statement::make_assignment(this->present_, ref, loc);
b->add_statement(s);
return Statement::make_block_statement(b, loc);
}
// Make a map assignment statement which returns a pair of values.
Statement*
Statement::make_tuple_map_assignment(Expression* val, Expression* present,
Expression* map_index,
source_location location)
{
return new Tuple_map_assignment_statement(val, present, map_index, location);
}
// Assign a pair of entries to a map.
// m[k] = v, p
class Map_assignment_statement : public Statement
{
public:
Map_assignment_statement(Expression* map_index,
Expression* val, Expression* should_set,
source_location location)
: Statement(STATEMENT_MAP_ASSIGNMENT, location),
map_index_(map_index), val_(val), should_set_(should_set)
{ }
protected:
int
do_traverse(Traverse* traverse);
bool
do_traverse_assignments(Traverse_assignments*)
{ gcc_unreachable(); }
Statement*
do_lower(Gogo*, Block*);
tree
do_get_tree(Translate_context*)
{ gcc_unreachable(); }
private:
// A reference to the map index which should be set or deleted.
Expression* map_index_;
// The value to add to the map.
Expression* val_;
// Whether or not to add the value.
Expression* should_set_;
};
// Traverse a map assignment.
int
Map_assignment_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression(traverse, &this->map_index_) == TRAVERSE_EXIT
|| this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->traverse_expression(traverse, &this->should_set_);
}
// Lower a map assignment to a function call.
Statement*
Map_assignment_statement::do_lower(Gogo*, Block* enclosing)
{
source_location loc = this->location();
Map_index_expression* map_index = this->map_index_->map_index_expression();
if (map_index == NULL)
{
this->report_error(_("expected map index on left hand side"));
return Statement::make_error_statement(loc);
}
Map_type* map_type = map_index->get_map_type();
if (map_type == NULL)
return Statement::make_error_statement(loc);
Block* b = new Block(enclosing, loc);
// Evaluate the map first to get order of evaluation right.
// map_temp := m // we are evaluating m[k] = v, p
Temporary_statement* map_temp = Statement::make_temporary(map_type,
map_index->map(),
loc);
b->add_statement(map_temp);
// var key_temp MAP_KEY_TYPE = k
Temporary_statement* key_temp =
Statement::make_temporary(map_type->key_type(), map_index->index(), loc);
b->add_statement(key_temp);
// var val_temp MAP_VAL_TYPE = v
Temporary_statement* val_temp =
Statement::make_temporary(map_type->val_type(), this->val_, loc);
b->add_statement(val_temp);
// func mapassign2(hmap map[k]v, key *k, val *v, p)
source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("hmap", map_type, bloc));
Type* pkey_type = Type::make_pointer_type(map_type->key_type());
param_types->push_back(Typed_identifier("key", pkey_type, bloc));
Type* pval_type = Type::make_pointer_type(map_type->val_type());
param_types->push_back(Typed_identifier("val", pval_type, bloc));
param_types->push_back(Typed_identifier("p", Type::lookup_bool_type(), bloc));
Function_type* fntype = Type::make_function_type(NULL, param_types,
NULL, bloc);
Named_object* mapassign2 =
Named_object::make_function_declaration("mapassign2", NULL, fntype, bloc);
mapassign2->func_declaration_value()->set_asm_name("runtime.mapassign2");
// mapassign2(map_temp, &key_temp, &val_temp, p)
Expression* func = Expression::make_func_reference(mapassign2, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(Expression::make_temporary_reference(map_temp, loc));
Expression* ref = Expression::make_temporary_reference(key_temp, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
ref = Expression::make_temporary_reference(val_temp, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
params->push_back(this->should_set_);
Expression* call = Expression::make_call(func, params, false, loc);
Statement* s = Statement::make_statement(call);
b->add_statement(s);
return Statement::make_block_statement(b, loc);
}
// Make a statement which assigns a pair of entries to a map.
Statement*
Statement::make_map_assignment(Expression* map_index,
Expression* val, Expression* should_set,
source_location location)
{
return new Map_assignment_statement(map_index, val, should_set, location);
}
// A tuple assignment from a receive statement.
class Tuple_receive_assignment_statement : public Statement
{
public:
Tuple_receive_assignment_statement(Expression* val, Expression* success,
Expression* channel,
source_location location)
: Statement(STATEMENT_TUPLE_RECEIVE_ASSIGNMENT, location),
val_(val), success_(success), channel_(channel)
{ }
protected:
int
do_traverse(Traverse* traverse);
bool
do_traverse_assignments(Traverse_assignments*)
{ gcc_unreachable(); }
Statement*
do_lower(Gogo*, Block*);
tree
do_get_tree(Translate_context*)
{ gcc_unreachable(); }
private:
// Lvalue which receives the value from the channel.
Expression* val_;
// Lvalue which receives whether the read succeeded or failed.
Expression* success_;
// The channel on which we receive the value.
Expression* channel_;
};
// Traversal.
int
Tuple_receive_assignment_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT
|| this->traverse_expression(traverse, &this->success_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->traverse_expression(traverse, &this->channel_);
}
// Lower to a function call.
Statement*
Tuple_receive_assignment_statement::do_lower(Gogo*, Block* enclosing)
{
source_location loc = this->location();
Channel_type* channel_type = this->channel_->type()->channel_type();
if (channel_type == NULL)
{
this->report_error(_("expected channel"));
return Statement::make_error_statement(loc);
}
if (!channel_type->may_receive())
{
this->report_error(_("invalid receive on send-only channel"));
return Statement::make_error_statement(loc);
}
Block* b = new Block(enclosing, loc);
// Make sure that any subexpressions on the left hand side are
// evaluated in the right order.
Move_ordered_evals moe(b);
this->val_->traverse_subexpressions(&moe);
this->success_->traverse_subexpressions(&moe);
// var val_temp ELEMENT_TYPE
Temporary_statement* val_temp =
Statement::make_temporary(channel_type->element_type(), NULL, loc);
b->add_statement(val_temp);
// var success_temp bool
Temporary_statement* success_temp =
Statement::make_temporary(Type::lookup_bool_type(), NULL, loc);
b->add_statement(success_temp);
// func chanrecv2(c chan T, val *T) bool
source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("c", channel_type, bloc));
Type* pelement_type = Type::make_pointer_type(channel_type->element_type());
param_types->push_back(Typed_identifier("val", pelement_type, bloc));
Typed_identifier_list* ret_types = new Typed_identifier_list();
ret_types->push_back(Typed_identifier("", Type::lookup_bool_type(), bloc));
Function_type* fntype = Type::make_function_type(NULL, param_types,
ret_types, bloc);
Named_object* chanrecv2 =
Named_object::make_function_declaration("chanrecv2", NULL, fntype, bloc);
chanrecv2->func_declaration_value()->set_asm_name("runtime.chanrecv2");
// success_temp = chanrecv2(channel, &val_temp)
Expression* func = Expression::make_func_reference(chanrecv2, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(this->channel_);
Expression* ref = Expression::make_temporary_reference(val_temp, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
Expression* call = Expression::make_call(func, params, false, loc);
ref = Expression::make_temporary_reference(success_temp, loc);
Statement* s = Statement::make_assignment(ref, call, loc);
b->add_statement(s);
// val = val_temp
ref = Expression::make_temporary_reference(val_temp, loc);
s = Statement::make_assignment(this->val_, ref, loc);
b->add_statement(s);
// success = success_temp
ref = Expression::make_temporary_reference(success_temp, loc);
s = Statement::make_assignment(this->success_, ref, loc);
b->add_statement(s);
return Statement::make_block_statement(b, loc);
}
// Make a nonblocking receive statement.
Statement*
Statement::make_tuple_receive_assignment(Expression* val, Expression* success,
Expression* channel,
source_location location)
{
return new Tuple_receive_assignment_statement(val, success, channel,
location);
}
// An assignment to a pair of values from a type guard. This is a
// conditional type guard. v, ok = i.(type).
class Tuple_type_guard_assignment_statement : public Statement
{
public:
Tuple_type_guard_assignment_statement(Expression* val, Expression* ok,
Expression* expr, Type* type,
source_location location)
: Statement(STATEMENT_TUPLE_TYPE_GUARD_ASSIGNMENT, location),
val_(val), ok_(ok), expr_(expr), type_(type)
{ }
protected:
int
do_traverse(Traverse*);
bool
do_traverse_assignments(Traverse_assignments*)
{ gcc_unreachable(); }
Statement*
do_lower(Gogo*, Block*);
tree
do_get_tree(Translate_context*)
{ gcc_unreachable(); }
private:
Call_expression*
lower_to_empty_interface(const char*);
Call_expression*
lower_to_type(const char*);
void
lower_to_object_type(Block*, const char*);
// The variable which recieves the converted value.
Expression* val_;
// The variable which receives the indication of success.
Expression* ok_;
// The expression being converted.
Expression* expr_;
// The type to which the expression is being converted.
Type* type_;
};
// Traverse a type guard tuple assignment.
int
Tuple_type_guard_assignment_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT
|| this->traverse_expression(traverse, &this->ok_) == TRAVERSE_EXIT
|| this->traverse_type(traverse, this->type_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->traverse_expression(traverse, &this->expr_);
}
// Lower to a function call.
Statement*
Tuple_type_guard_assignment_statement::do_lower(Gogo*, Block* enclosing)
{
source_location loc = this->location();
Type* expr_type = this->expr_->type();
if (expr_type->interface_type() == NULL)
{
if (!expr_type->is_error_type() && !this->type_->is_error_type())
this->report_error(_("type assertion only valid for interface types"));
return Statement::make_error_statement(loc);
}
Block* b = new Block(enclosing, loc);
// Make sure that any subexpressions on the left hand side are
// evaluated in the right order.
Move_ordered_evals moe(b);
this->val_->traverse_subexpressions(&moe);
this->ok_->traverse_subexpressions(&moe);
bool expr_is_empty = expr_type->interface_type()->is_empty();
Call_expression* call;
if (this->type_->interface_type() != NULL)
{
if (this->type_->interface_type()->is_empty())
call = this->lower_to_empty_interface(expr_is_empty
? "ifaceE2E2"
: "ifaceI2E2");
else
call = this->lower_to_type(expr_is_empty ? "ifaceE2I2" : "ifaceI2I2");
}
else if (this->type_->points_to() != NULL)
call = this->lower_to_type(expr_is_empty ? "ifaceE2T2P" : "ifaceI2T2P");
else
{
this->lower_to_object_type(b, expr_is_empty ? "ifaceE2T2" : "ifaceI2T2");
call = NULL;
}
if (call != NULL)
{
Expression* res = Expression::make_call_result(call, 0);
Statement* s = Statement::make_assignment(this->val_, res, loc);
b->add_statement(s);
res = Expression::make_call_result(call, 1);
s = Statement::make_assignment(this->ok_, res, loc);
b->add_statement(s);
}
return Statement::make_block_statement(b, loc);
}
// Lower a conversion to an empty interface type.
Call_expression*
Tuple_type_guard_assignment_statement::lower_to_empty_interface(
const char *fnname)
{
source_location loc = this->location();
// func FNNAME(interface) (empty, bool)
source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("i", this->expr_->type(), bloc));
Typed_identifier_list* ret_types = new Typed_identifier_list();
ret_types->push_back(Typed_identifier("ret", this->type_, bloc));
ret_types->push_back(Typed_identifier("ok", Type::lookup_bool_type(), bloc));
Function_type* fntype = Type::make_function_type(NULL, param_types,
ret_types, bloc);
Named_object* fn =
Named_object::make_function_declaration(fnname, NULL, fntype, bloc);
std::string asm_name = "runtime.";
asm_name += fnname;
fn->func_declaration_value()->set_asm_name(asm_name);
// val, ok = FNNAME(expr)
Expression* func = Expression::make_func_reference(fn, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(this->expr_);
return Expression::make_call(func, params, false, loc);
}
// Lower a conversion to a non-empty interface type or a pointer type.
Call_expression*
Tuple_type_guard_assignment_statement::lower_to_type(const char* fnname)
{
source_location loc = this->location();
// func FNNAME(*descriptor, interface) (interface, bool)
source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("inter",
Type::make_type_descriptor_ptr_type(),
bloc));
param_types->push_back(Typed_identifier("i", this->expr_->type(), bloc));
Typed_identifier_list* ret_types = new Typed_identifier_list();
ret_types->push_back(Typed_identifier("ret", this->type_, bloc));
ret_types->push_back(Typed_identifier("ok", Type::lookup_bool_type(), bloc));
Function_type* fntype = Type::make_function_type(NULL, param_types,
ret_types, bloc);
Named_object* fn =
Named_object::make_function_declaration(fnname, NULL, fntype, bloc);
std::string asm_name = "runtime.";
asm_name += fnname;
fn->func_declaration_value()->set_asm_name(asm_name);
// val, ok = FNNAME(type_descriptor, expr)
Expression* func = Expression::make_func_reference(fn, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(Expression::make_type_descriptor(this->type_, loc));
params->push_back(this->expr_);
return Expression::make_call(func, params, false, loc);
}
// Lower a conversion to a non-interface non-pointer type.
void
Tuple_type_guard_assignment_statement::lower_to_object_type(Block* b,
const char *fnname)
{
source_location loc = this->location();
// var val_temp TYPE
Temporary_statement* val_temp = Statement::make_temporary(this->type_,
NULL, loc);
b->add_statement(val_temp);
// func FNNAME(*descriptor, interface, *T) bool
source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("inter",
Type::make_type_descriptor_ptr_type(),
bloc));
param_types->push_back(Typed_identifier("i", this->expr_->type(), bloc));
Type* ptype = Type::make_pointer_type(this->type_);
param_types->push_back(Typed_identifier("v", ptype, bloc));
Typed_identifier_list* ret_types = new Typed_identifier_list();
ret_types->push_back(Typed_identifier("ok", Type::lookup_bool_type(), bloc));
Function_type* fntype = Type::make_function_type(NULL, param_types,
ret_types, bloc);
Named_object* fn =
Named_object::make_function_declaration(fnname, NULL, fntype, bloc);
std::string asm_name = "runtime.";
asm_name += fnname;
fn->func_declaration_value()->set_asm_name(asm_name);
// ok = FNNAME(type_descriptor, expr, &val_temp)
Expression* func = Expression::make_func_reference(fn, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(Expression::make_type_descriptor(this->type_, loc));
params->push_back(this->expr_);
Expression* ref = Expression::make_temporary_reference(val_temp, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
Expression* call = Expression::make_call(func, params, false, loc);
Statement* s = Statement::make_assignment(this->ok_, call, loc);
b->add_statement(s);
// val = val_temp
ref = Expression::make_temporary_reference(val_temp, loc);
s = Statement::make_assignment(this->val_, ref, loc);
b->add_statement(s);
}
// Make an assignment from a type guard to a pair of variables.
Statement*
Statement::make_tuple_type_guard_assignment(Expression* val, Expression* ok,
Expression* expr, Type* type,
source_location location)
{
return new Tuple_type_guard_assignment_statement(val, ok, expr, type,
location);
}
// An expression statement.
class Expression_statement : public Statement
{
public:
Expression_statement(Expression* expr)
: Statement(STATEMENT_EXPRESSION, expr->location()),
expr_(expr)
{ }
protected:
int
do_traverse(Traverse* traverse)
{ return this->traverse_expression(traverse, &this->expr_); }
void
do_determine_types()
{ this->expr_->determine_type_no_context(); }
bool
do_may_fall_through() const;
tree
do_get_tree(Translate_context* context)
{ return this->expr_->get_tree(context); }
private:
Expression* expr_;
};
// An expression statement may fall through unless it is a call to a
// function which does not return.
bool
Expression_statement::do_may_fall_through() const
{
const Call_expression* call = this->expr_->call_expression();
if (call != NULL)
{
const Expression* fn = call->fn();
const Func_expression* fe = fn->func_expression();
if (fe != NULL)
{
const Named_object* no = fe->named_object();
Function_type* fntype;
if (no->is_function())
fntype = no->func_value()->type();
else if (no->is_function_declaration())
fntype = no->func_declaration_value()->type();
else
fntype = NULL;
// The builtin function panic does not return.
if (fntype != NULL && fntype->is_builtin() && no->name() == "panic")
return false;
}
}
return true;
}
// Make an expression statement from an Expression.
Statement*
Statement::make_statement(Expression* expr)
{
return new Expression_statement(expr);
}
// A block statement--a list of statements which may include variable
// definitions.
class Block_statement : public Statement
{
public:
Block_statement(Block* block, source_location location)
: Statement(STATEMENT_BLOCK, location),
block_(block)
{ }
protected:
int
do_traverse(Traverse* traverse)
{ return this->block_->traverse(traverse); }
void
do_determine_types()
{ this->block_->determine_types(); }
bool
do_may_fall_through() const
{ return this->block_->may_fall_through(); }
tree
do_get_tree(Translate_context* context)
{ return this->block_->get_tree(context); }
private:
Block* block_;
};
// Make a block statement.
Statement*
Statement::make_block_statement(Block* block, source_location location)
{
return new Block_statement(block, location);
}
// An increment or decrement statement.
class Inc_dec_statement : public Statement
{
public:
Inc_dec_statement(bool is_inc, Expression* expr)
: Statement(STATEMENT_INCDEC, expr->location()),
expr_(expr), is_inc_(is_inc)
{ }
protected:
int
do_traverse(Traverse* traverse)
{ return this->traverse_expression(traverse, &this->expr_); }
bool
do_traverse_assignments(Traverse_assignments*)
{ gcc_unreachable(); }
Statement*
do_lower(Gogo*, Block*);
tree
do_get_tree(Translate_context*)
{ gcc_unreachable(); }
private:
// The l-value to increment or decrement.
Expression* expr_;
// Whether to increment or decrement.
bool is_inc_;
};
// Lower to += or -=.
Statement*
Inc_dec_statement::do_lower(Gogo*, Block*)
{
source_location loc = this->location();
mpz_t oval;
mpz_init_set_ui(oval, 1UL);
Expression* oexpr = Expression::make_integer(&oval, NULL, loc);
mpz_clear(oval);
Operator op = this->is_inc_ ? OPERATOR_PLUSEQ : OPERATOR_MINUSEQ;
return Statement::make_assignment_operation(op, this->expr_, oexpr, loc);
}
// Make an increment statement.
Statement*
Statement::make_inc_statement(Expression* expr)
{
return new Inc_dec_statement(true, expr);
}
// Make a decrement statement.
Statement*
Statement::make_dec_statement(Expression* expr)
{
return new Inc_dec_statement(false, expr);
}
// Class Thunk_statement. This is the base class for go and defer
// statements.
const char* const Thunk_statement::thunk_field_fn = "fn";
const char* const Thunk_statement::thunk_field_receiver = "receiver";
// Constructor.
Thunk_statement::Thunk_statement(Statement_classification classification,
Call_expression* call,
source_location location)
: Statement(classification, location),
call_(call), struct_type_(NULL)
{
}
// Return whether this is a simple statement which does not require a
// thunk.
bool
Thunk_statement::is_simple(Function_type* fntype) const
{
// We need a thunk to call a method, or to pass a variable number of
// arguments.
if (fntype->is_method() || fntype->is_varargs())
return false;
// A defer statement requires a thunk to set up for whether the
// function can call recover.
if (this->classification() == STATEMENT_DEFER)
return false;
// We can only permit a single parameter of pointer type.
const Typed_identifier_list* parameters = fntype->parameters();
if (parameters != NULL
&& (parameters->size() > 1
|| (parameters->size() == 1
&& parameters->begin()->type()->points_to() == NULL)))
return false;
// If the function returns multiple values, or returns a type other
// than integer, floating point, or pointer, then it may get a
// hidden first parameter, in which case we need the more
// complicated approach. This is true even though we are going to
// ignore the return value.
const Typed_identifier_list* results = fntype->results();
if (results != NULL
&& (results->size() > 1
|| (results->size() == 1
&& !results->begin()->type()->is_basic_type()
&& results->begin()->type()->points_to() == NULL)))
return false;
// If this calls something which is not a simple function, then we
// need a thunk.
Expression* fn = this->call_->call_expression()->fn();
if (fn->bound_method_expression() != NULL
|| fn->interface_field_reference_expression() != NULL)
return false;
return true;
}
// Traverse a thunk statement.
int
Thunk_statement::do_traverse(Traverse* traverse)
{
return this->traverse_expression(traverse, &this->call_);
}
// We implement traverse_assignment for a thunk statement because it
// effectively copies the function call.
bool
Thunk_statement::do_traverse_assignments(Traverse_assignments* tassign)
{
Expression* fn = this->call_->call_expression()->fn();
Expression* fn2 = fn;
tassign->value(&fn2, true, false);
return true;
}
// Determine types in a thunk statement.
void
Thunk_statement::do_determine_types()
{
this->call_->determine_type_no_context();
// Now that we know the types of the call, build the struct used to
// pass parameters.
Call_expression* ce = this->call_->call_expression();
if (ce == NULL)
return;
Function_type* fntype = ce->get_function_type();
if (fntype != NULL && !this->is_simple(fntype))
this->struct_type_ = this->build_struct(fntype);
}
// Check types in a thunk statement.
void
Thunk_statement::do_check_types(Gogo*)
{
Call_expression* ce = this->call_->call_expression();
if (ce == NULL)
{
if (!this->call_->is_error_expression())
this->report_error("expected call expression");
return;
}
Function_type* fntype = ce->get_function_type();
if (fntype != NULL && fntype->is_method())
{
Expression* fn = ce->fn();
if (fn->bound_method_expression() == NULL
&& fn->interface_field_reference_expression() == NULL)
this->report_error(_("no object for method call"));
}
}
// The Traverse class used to find and simplify thunk statements.
class Simplify_thunk_traverse : public Traverse
{
public:
Simplify_thunk_traverse(Gogo* gogo)
: Traverse(traverse_blocks),
gogo_(gogo)
{ }
int
block(Block*);
private:
Gogo* gogo_;
};
int
Simplify_thunk_traverse::block(Block* b)
{
// The parser ensures that thunk statements always appear at the end
// of a block.
if (b->statements()->size() < 1)
return TRAVERSE_CONTINUE;
Thunk_statement* stat = b->statements()->back()->thunk_statement();
if (stat == NULL)
return TRAVERSE_CONTINUE;
if (stat->simplify_statement(this->gogo_, b))
return TRAVERSE_SKIP_COMPONENTS;
return TRAVERSE_CONTINUE;
}
// Simplify all thunk statements.
void
Gogo::simplify_thunk_statements()
{
Simplify_thunk_traverse thunk_traverse(this);
this->traverse(&thunk_traverse);
}
// Simplify complex thunk statements into simple ones. A complicated
// thunk statement is one which takes anything other than zero
// parameters or a single pointer parameter. We rewrite it into code
// which allocates a struct, stores the parameter values into the
// struct, and does a simple go or defer statement which passes the
// struct to a thunk. The thunk does the real call.
bool
Thunk_statement::simplify_statement(Gogo* gogo, Block* block)
{
if (this->classification() == STATEMENT_ERROR)
return false;
if (this->call_->is_error_expression())
return false;
Call_expression* ce = this->call_->call_expression();
Function_type* fntype = ce->get_function_type();
if (fntype == NULL)
{
gcc_assert(saw_errors());
this->set_is_error();
return false;
}
if (this->is_simple(fntype))
return false;
Expression* fn = ce->fn();
Bound_method_expression* bound_method = fn->bound_method_expression();
Interface_field_reference_expression* interface_method =
fn->interface_field_reference_expression();
const bool is_method = bound_method != NULL || interface_method != NULL;
source_location location = this->location();
std::string thunk_name = Gogo::thunk_name();
// Build the thunk.
this->build_thunk(gogo, thunk_name, fntype);
// Generate code to call the thunk.
// Get the values to store into the struct which is the single
// argument to the thunk.
Expression_list* vals = new Expression_list();
if (fntype->is_builtin())
;
else if (!is_method)
vals->push_back(fn);
else if (interface_method != NULL)
vals->push_back(interface_method->expr());
else if (bound_method != NULL)
{
vals->push_back(bound_method->method());
Expression* first_arg = bound_method->first_argument();
// We always pass a pointer when calling a method.
if (first_arg->type()->points_to() == NULL)
first_arg = Expression::make_unary(OPERATOR_AND, first_arg, location);
// If we are calling a method which was inherited from an
// embedded struct, and the method did not get a stub, then the
// first type may be wrong.
Type* fatype = bound_method->first_argument_type();
if (fatype != NULL)
{
if (fatype->points_to() == NULL)
fatype = Type::make_pointer_type(fatype);
Type* unsafe = Type::make_pointer_type(Type::make_void_type());
first_arg = Expression::make_cast(unsafe, first_arg, location);
first_arg = Expression::make_cast(fatype, first_arg, location);
}
vals->push_back(first_arg);
}
else
gcc_unreachable();
if (ce->args() != NULL)
{
for (Expression_list::const_iterator p = ce->args()->begin();
p != ce->args()->end();
++p)
vals->push_back(*p);
}
// Build the struct.
Expression* constructor =
Expression::make_struct_composite_literal(this->struct_type_, vals,
location);
// Allocate the initialized struct on the heap.
constructor = Expression::make_heap_composite(constructor, location);
// Look up the thunk.
Named_object* named_thunk = gogo->lookup(thunk_name, NULL);
gcc_assert(named_thunk != NULL && named_thunk->is_function());
// Build the call.
Expression* func = Expression::make_func_reference(named_thunk, NULL,
location);
Expression_list* params = new Expression_list();
params->push_back(constructor);
Call_expression* call = Expression::make_call(func, params, false, location);
// Build the simple go or defer statement.
Statement* s;
if (this->classification() == STATEMENT_GO)
s = Statement::make_go_statement(call, location);
else if (this->classification() == STATEMENT_DEFER)
s = Statement::make_defer_statement(call, location);
else
gcc_unreachable();
// The current block should end with the go statement.
gcc_assert(block->statements()->size() >= 1);
gcc_assert(block->statements()->back() == this);
block->replace_statement(block->statements()->size() - 1, s);
// We already ran the determine_types pass, so we need to run it now
// for the new statement.
s->determine_types();
// Sanity check.
gogo->check_types_in_block(block);
// Return true to tell the block not to keep looking at statements.
return true;
}
// Set the name to use for thunk parameter N.
void
Thunk_statement::thunk_field_param(int n, char* buf, size_t buflen)
{
snprintf(buf, buflen, "a%d", n);
}
// Build a new struct type to hold the parameters for a complicated
// thunk statement. FNTYPE is the type of the function call.
Struct_type*
Thunk_statement::build_struct(Function_type* fntype)
{
source_location location = this->location();
Struct_field_list* fields = new Struct_field_list();
Call_expression* ce = this->call_->call_expression();
Expression* fn = ce->fn();
Interface_field_reference_expression* interface_method =
fn->interface_field_reference_expression();
if (interface_method != NULL)
{
// If this thunk statement calls a method on an interface, we
// pass the interface object to the thunk.
Typed_identifier tid(Thunk_statement::thunk_field_fn,
interface_method->expr()->type(),
location);
fields->push_back(Struct_field(tid));
}
else if (!fntype->is_builtin())
{
// The function to call.
Typed_identifier tid(Go_statement::thunk_field_fn, fntype, location);
fields->push_back(Struct_field(tid));
}
else if (ce->is_recover_call())
{
// The predeclared recover function has no argument. However,
// we add an argument when building recover thunks. Handle that
// here.
fields->push_back(Struct_field(Typed_identifier("can_recover",
Type::lookup_bool_type(),
location)));
}
if (fn->bound_method_expression() != NULL)
{
gcc_assert(fntype->is_method());
Type* rtype = fntype->receiver()->type();
// We always pass the receiver as a pointer.
if (rtype->points_to() == NULL)
rtype = Type::make_pointer_type(rtype);
Typed_identifier tid(Thunk_statement::thunk_field_receiver, rtype,
location);
fields->push_back(Struct_field(tid));
}
const Expression_list* args = ce->args();
if (args != NULL)
{
int i = 0;
for (Expression_list::const_iterator p = args->begin();
p != args->end();
++p, ++i)
{
char buf[50];
this->thunk_field_param(i, buf, sizeof buf);
fields->push_back(Struct_field(Typed_identifier(buf, (*p)->type(),
location)));
}
}
return Type::make_struct_type(fields, location);
}
// Build the thunk we are going to call. This is a brand new, albeit
// artificial, function.
void
Thunk_statement::build_thunk(Gogo* gogo, const std::string& thunk_name,
Function_type* fntype)
{
source_location location = this->location();
Call_expression* ce = this->call_->call_expression();
bool may_call_recover = false;
if (this->classification() == STATEMENT_DEFER)
{
Func_expression* fn = ce->fn()->func_expression();
if (fn == NULL)
may_call_recover = true;
else
{
const Named_object* no = fn->named_object();
if (!no->is_function())
may_call_recover = true;
else
may_call_recover = no->func_value()->calls_recover();
}
}
// Build the type of the thunk. The thunk takes a single parameter,
// which is a pointer to the special structure we build.
const char* const parameter_name = "__go_thunk_parameter";
Typed_identifier_list* thunk_parameters = new Typed_identifier_list();
Type* pointer_to_struct_type = Type::make_pointer_type(this->struct_type_);
thunk_parameters->push_back(Typed_identifier(parameter_name,
pointer_to_struct_type,
location));
Typed_identifier_list* thunk_results = NULL;
if (may_call_recover)
{
// When deferring a function which may call recover, add a
// return value, to disable tail call optimizations which will
// break the way we check whether recover is permitted.
thunk_results = new Typed_identifier_list();
thunk_results->push_back(Typed_identifier("", Type::lookup_bool_type(),
location));
}
Function_type* thunk_type = Type::make_function_type(NULL, thunk_parameters,
thunk_results,
location);
// Start building the thunk.
Named_object* function = gogo->start_function(thunk_name, thunk_type, true,
location);
// For a defer statement, start with a call to
// __go_set_defer_retaddr. */
Label* retaddr_label = NULL;
if (may_call_recover)
{
retaddr_label = gogo->add_label_reference("retaddr");
Expression* arg = Expression::make_label_addr(retaddr_label, location);
Expression_list* args = new Expression_list();
args->push_back(arg);
static Named_object* set_defer_retaddr;
if (set_defer_retaddr == NULL)
{
const source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types = new Typed_identifier_list();
Type *voidptr_type = Type::make_pointer_type(Type::make_void_type());
param_types->push_back(Typed_identifier("r", voidptr_type, bloc));
Typed_identifier_list* result_types = new Typed_identifier_list();
result_types->push_back(Typed_identifier("",
Type::lookup_bool_type(),
bloc));
Function_type* t = Type::make_function_type(NULL, param_types,
result_types, bloc);
set_defer_retaddr =
Named_object::make_function_declaration("__go_set_defer_retaddr",
NULL, t, bloc);
const char* n = "__go_set_defer_retaddr";
set_defer_retaddr->func_declaration_value()->set_asm_name(n);
}
Expression* fn = Expression::make_func_reference(set_defer_retaddr,
NULL, location);
Expression* call = Expression::make_call(fn, args, false, location);
// This is a hack to prevent the middle-end from deleting the
// label.
gogo->start_block(location);
gogo->add_statement(Statement::make_goto_statement(retaddr_label,
location));
Block* then_block = gogo->finish_block(location);
then_block->determine_types();
Statement* s = Statement::make_if_statement(call, then_block, NULL,
location);
s->determine_types();
gogo->add_statement(s);
}
// Get a reference to the parameter.
Named_object* named_parameter = gogo->lookup(parameter_name, NULL);
gcc_assert(named_parameter != NULL && named_parameter->is_variable());
// Build the call. Note that the field names are the same as the
// ones used in build_struct.
Expression* thunk_parameter = Expression::make_var_reference(named_parameter,
location);
thunk_parameter = Expression::make_unary(OPERATOR_MULT, thunk_parameter,
location);
Bound_method_expression* bound_method = ce->fn()->bound_method_expression();
Interface_field_reference_expression* interface_method =
ce->fn()->interface_field_reference_expression();
Expression* func_to_call;
unsigned int next_index;
if (!fntype->is_builtin())
{
func_to_call = Expression::make_field_reference(thunk_parameter,
0, location);
next_index = 1;
}
else
{
gcc_assert(bound_method == NULL && interface_method == NULL);
func_to_call = ce->fn();
next_index = 0;
}
if (bound_method != NULL)
{
Expression* r = Expression::make_field_reference(thunk_parameter, 1,
location);
// The main program passes in a function pointer from the
// interface expression, so here we can make a bound method in
// all cases.
func_to_call = Expression::make_bound_method(r, func_to_call,
location);
next_index = 2;
}
else if (interface_method != NULL)
{
// The main program passes the interface object.
const std::string& name(interface_method->name());
func_to_call = Expression::make_interface_field_reference(func_to_call,
name,
location);
}
Expression_list* call_params = new Expression_list();
const Struct_field_list* fields = this->struct_type_->fields();
Struct_field_list::const_iterator p = fields->begin();
for (unsigned int i = 0; i < next_index; ++i)
++p;
bool is_recover_call = ce->is_recover_call();
Expression* recover_arg = NULL;
for (; p != fields->end(); ++p, ++next_index)
{
Expression* thunk_param = Expression::make_var_reference(named_parameter,
location);
thunk_param = Expression::make_unary(OPERATOR_MULT, thunk_param,
location);
Expression* param = Expression::make_field_reference(thunk_param,
next_index,
location);
if (!is_recover_call)
call_params->push_back(param);
else
{
gcc_assert(call_params->empty());
recover_arg = param;
}
}
if (call_params->empty())
{
delete call_params;
call_params = NULL;
}
Expression* call = Expression::make_call(func_to_call, call_params, false,
location);
// We need to lower in case this is a builtin function.
call = call->lower(gogo, function, -1);
Call_expression* call_ce = call->call_expression();
if (call_ce != NULL && may_call_recover)
call_ce->set_is_deferred();
Statement* call_statement = Statement::make_statement(call);
// We already ran the determine_types pass, so we need to run it
// just for this statement now.
call_statement->determine_types();
// Sanity check.
call->check_types(gogo);
if (call_ce != NULL && recover_arg != NULL)
call_ce->set_recover_arg(recover_arg);
gogo->add_statement(call_statement);
// If this is a defer statement, the label comes immediately after
// the call.
if (may_call_recover)
{
gogo->add_label_definition("retaddr", location);
Expression_list* vals = new Expression_list();
vals->push_back(Expression::make_boolean(false, location));
const Typed_identifier_list* results =
function->func_value()->type()->results();
gogo->add_statement(Statement::make_return_statement(results, vals,
location));
}
// That is all the thunk has to do.
gogo->finish_function(location);
}
// Get the function and argument trees.
void
Thunk_statement::get_fn_and_arg(Translate_context* context, tree* pfn,
tree* parg)
{
if (this->call_->is_error_expression())
{
*pfn = error_mark_node;
*parg = error_mark_node;
return;
}
Call_expression* ce = this->call_->call_expression();
Expression* fn = ce->fn();
*pfn = fn->get_tree(context);
const Expression_list* args = ce->args();
if (args == NULL || args->empty())
*parg = null_pointer_node;
else
{
gcc_assert(args->size() == 1);
*parg = args->front()->get_tree(context);
}
}
// Class Go_statement.
tree
Go_statement::do_get_tree(Translate_context* context)
{
tree fn_tree;
tree arg_tree;
this->get_fn_and_arg(context, &fn_tree, &arg_tree);
static tree go_fndecl;
tree fn_arg_type = NULL_TREE;
if (go_fndecl == NULL_TREE)
{
// Only build FN_ARG_TYPE if we need it.
tree subargtypes = tree_cons(NULL_TREE, ptr_type_node, void_list_node);
tree subfntype = build_function_type(ptr_type_node, subargtypes);
fn_arg_type = build_pointer_type(subfntype);
}
return Gogo::call_builtin(&go_fndecl,
this->location(),
"__go_go",
2,
void_type_node,
fn_arg_type,
fn_tree,
ptr_type_node,
arg_tree);
}
// Make a go statement.
Statement*
Statement::make_go_statement(Call_expression* call, source_location location)
{
return new Go_statement(call, location);
}
// Class Defer_statement.
tree
Defer_statement::do_get_tree(Translate_context* context)
{
source_location loc = this->location();
tree fn_tree;
tree arg_tree;
this->get_fn_and_arg(context, &fn_tree, &arg_tree);
if (fn_tree == error_mark_node || arg_tree == error_mark_node)
return error_mark_node;
static tree defer_fndecl;
tree fn_arg_type = NULL_TREE;
if (defer_fndecl == NULL_TREE)
{
// Only build FN_ARG_TYPE if we need it.
tree subargtypes = tree_cons(NULL_TREE, ptr_type_node, void_list_node);
tree subfntype = build_function_type(ptr_type_node, subargtypes);
fn_arg_type = build_pointer_type(subfntype);
}
tree defer_stack = context->function()->func_value()->defer_stack(loc);
return Gogo::call_builtin(&defer_fndecl,
loc,
"__go_defer",
3,
void_type_node,
ptr_type_node,
defer_stack,
fn_arg_type,
fn_tree,
ptr_type_node,
arg_tree);
}
// Make a defer statement.
Statement*
Statement::make_defer_statement(Call_expression* call,
source_location location)
{
return new Defer_statement(call, location);
}
// Class Return_statement.
// Traverse assignments. We treat each return value as a top level
// RHS in an expression.
bool
Return_statement::do_traverse_assignments(Traverse_assignments* tassign)
{
Expression_list* vals = this->vals_;
if (vals != NULL)
{
for (Expression_list::iterator p = vals->begin();
p != vals->end();
++p)
tassign->value(&*p, true, true);
}
return true;
}
// Lower a return statement. If we are returning a function call
// which returns multiple values which match the current function,
// split up the call's results. If the function has named result
// variables, and the return statement lists explicit values, then
// implement it by assigning the values to the result variables and
// changing the statement to not list any values. This lets
// panic/recover work correctly.
Statement*
Return_statement::do_lower(Gogo*, Block* enclosing)
{
if (this->vals_ == NULL)
return this;
const Typed_identifier_list* results = this->results_;
if (results == NULL || results->empty())
return this;
// If the current function has multiple return values, and we are
// returning a single call expression, split up the call expression.
size_t results_count = results->size();
if (results_count > 1
&& this->vals_->size() == 1
&& this->vals_->front()->call_expression() != NULL)
{
Call_expression* call = this->vals_->front()->call_expression();
size_t count = results->size();
Expression_list* vals = new Expression_list;
for (size_t i = 0; i < count; ++i)
vals->push_back(Expression::make_call_result(call, i));
delete this->vals_;
this->vals_ = vals;
}
if (results->front().name().empty())
return this;
if (results_count != this->vals_->size())
{
// Presumably an error which will be reported in check_types.
return this;
}
// Assign to named return values and then return them.
source_location loc = this->location();
const Block* top = enclosing;
while (top->enclosing() != NULL)
top = top->enclosing();
const Bindings *bindings = top->bindings();
Block* b = new Block(enclosing, loc);
Expression_list* lhs = new Expression_list();
Expression_list* rhs = new Expression_list();
Expression_list::const_iterator pe = this->vals_->begin();
int i = 1;
for (Typed_identifier_list::const_iterator pr = results->begin();
pr != results->end();
++pr, ++pe, ++i)
{
Named_object* rv = bindings->lookup_local(pr->name());
if (rv == NULL || !rv->is_result_variable())
{
// Presumably an error.
delete b;
delete lhs;
delete rhs;
return this;
}
Expression* e = *pe;
// Check types now so that we give a good error message. The
// result type is known. We determine the expression type
// early.
Type *rvtype = rv->result_var_value()->type();
Type_context type_context(rvtype, false);
e->determine_type(&type_context);
std::string reason;
if (Type::are_assignable(rvtype, e->type(), &reason))
{
Expression* ve = Expression::make_var_reference(rv, e->location());
lhs->push_back(ve);
rhs->push_back(e);
}
else
{
if (reason.empty())
error_at(e->location(), "incompatible type for return value %d", i);
else
error_at(e->location(),
"incompatible type for return value %d (%s)",
i, reason.c_str());
}
}
gcc_assert(lhs->size() == rhs->size());
if (lhs->empty())
;
else if (lhs->size() == 1)
{
b->add_statement(Statement::make_assignment(lhs->front(), rhs->front(),
loc));
delete lhs;
delete rhs;
}
else
b->add_statement(Statement::make_tuple_assignment(lhs, rhs, loc));
b->add_statement(Statement::make_return_statement(this->results_, NULL,
loc));
return Statement::make_block_statement(b, loc);
}
// Determine types.
void
Return_statement::do_determine_types()
{
if (this->vals_ == NULL)
return;
const Typed_identifier_list* results = this->results_;
Typed_identifier_list::const_iterator pt;
if (results != NULL)
pt = results->begin();
for (Expression_list::iterator pe = this->vals_->begin();
pe != this->vals_->end();
++pe)
{
if (results == NULL || pt == results->end())
(*pe)->determine_type_no_context();
else
{
Type_context context(pt->type(), false);
(*pe)->determine_type(&context);
++pt;
}
}
}
// Check types.
void
Return_statement::do_check_types(Gogo*)
{
if (this->vals_ == NULL)
return;
const Typed_identifier_list* results = this->results_;
if (results == NULL)
{
this->report_error(_("return with value in function "
"with no return type"));
return;
}
int i = 1;
Typed_identifier_list::const_iterator pt = results->begin();
for (Expression_list::const_iterator pe = this->vals_->begin();
pe != this->vals_->end();
++pe, ++pt, ++i)
{
if (pt == results->end())
{
this->report_error(_("too many values in return statement"));
return;
}
std::string reason;
if (!Type::are_assignable(pt->type(), (*pe)->type(), &reason))
{
if (reason.empty())
error_at(this->location(),
"incompatible type for return value %d",
i);
else
error_at(this->location(),
"incompatible type for return value %d (%s)",
i, reason.c_str());
this->set_is_error();
}
else if (pt->type()->is_error_type()
|| (*pe)->type()->is_error_type()
|| pt->type()->is_undefined()
|| (*pe)->type()->is_undefined())
{
// Make sure we get the error for an undefined type.
pt->type()->base();
(*pe)->type()->base();
this->set_is_error();
}
}
if (pt != results->end())
this->report_error(_("not enough values in return statement"));
}
// Build a RETURN_EXPR tree.
tree
Return_statement::do_get_tree(Translate_context* context)
{
Function* function = context->function()->func_value();
tree fndecl = function->get_decl();
if (fndecl == error_mark_node || DECL_RESULT(fndecl) == error_mark_node)
return error_mark_node;
const Typed_identifier_list* results = this->results_;
if (this->vals_ == NULL)
{
tree stmt_list = NULL_TREE;
tree retval = function->return_value(context->gogo(),
context->function(),
this->location(),
&stmt_list);
tree set;
if (retval == NULL_TREE)
set = NULL_TREE;
else if (retval == error_mark_node)
return error_mark_node;
else
set = fold_build2_loc(this->location(), MODIFY_EXPR, void_type_node,
DECL_RESULT(fndecl), retval);
append_to_statement_list(this->build_stmt_1(RETURN_EXPR, set),
&stmt_list);
return stmt_list;
}
else if (this->vals_->size() == 1)
{
gcc_assert(!VOID_TYPE_P(TREE_TYPE(TREE_TYPE(fndecl))));
tree val = (*this->vals_->begin())->get_tree(context);
gcc_assert(results != NULL && results->size() == 1);
val = Expression::convert_for_assignment(context,
results->begin()->type(),
(*this->vals_->begin())->type(),
val, this->location());
if (val == error_mark_node)
return error_mark_node;
tree set = build2(MODIFY_EXPR, void_type_node,
DECL_RESULT(fndecl), val);
SET_EXPR_LOCATION(set, this->location());
return this->build_stmt_1(RETURN_EXPR, set);
}
else
{
gcc_assert(!VOID_TYPE_P(TREE_TYPE(TREE_TYPE(fndecl))));
tree stmt_list = NULL_TREE;
tree rettype = TREE_TYPE(DECL_RESULT(fndecl));
tree retvar = create_tmp_var(rettype, "RESULT");
gcc_assert(results != NULL && results->size() == this->vals_->size());
Expression_list::const_iterator pv = this->vals_->begin();
Typed_identifier_list::const_iterator pr = results->begin();
for (tree field = TYPE_FIELDS(rettype);
field != NULL_TREE;
++pv, ++pr, field = DECL_CHAIN(field))
{
gcc_assert(pv != this->vals_->end());
tree val = (*pv)->get_tree(context);
val = Expression::convert_for_assignment(context, pr->type(),
(*pv)->type(), val,
this->location());
if (val == error_mark_node)
return error_mark_node;
tree set = build2(MODIFY_EXPR, void_type_node,
build3(COMPONENT_REF, TREE_TYPE(field),
retvar, field, NULL_TREE),
val);
SET_EXPR_LOCATION(set, this->location());
append_to_statement_list(set, &stmt_list);
}
tree set = build2(MODIFY_EXPR, void_type_node, DECL_RESULT(fndecl),
retvar);
append_to_statement_list(this->build_stmt_1(RETURN_EXPR, set),
&stmt_list);
return stmt_list;
}
}
// Make a return statement.
Statement*
Statement::make_return_statement(const Typed_identifier_list* results,
Expression_list* vals,
source_location location)
{
return new Return_statement(results, vals, location);
}
// A break or continue statement.
class Bc_statement : public Statement
{
public:
Bc_statement(bool is_break, Unnamed_label* label, source_location location)
: Statement(STATEMENT_BREAK_OR_CONTINUE, location),
label_(label), is_break_(is_break)
{ }
bool
is_break() const
{ return this->is_break_; }
protected:
int
do_traverse(Traverse*)
{ return TRAVERSE_CONTINUE; }
bool
do_may_fall_through() const
{ return false; }
tree
do_get_tree(Translate_context*)
{ return this->label_->get_goto(this->location()); }
private:
// The label that this branches to.
Unnamed_label* label_;
// True if this is "break", false if it is "continue".
bool is_break_;
};
// Make a break statement.
Statement*
Statement::make_break_statement(Unnamed_label* label, source_location location)
{
return new Bc_statement(true, label, location);
}
// Make a continue statement.
Statement*
Statement::make_continue_statement(Unnamed_label* label,
source_location location)
{
return new Bc_statement(false, label, location);
}
// A goto statement.
class Goto_statement : public Statement
{
public:
Goto_statement(Label* label, source_location location)
: Statement(STATEMENT_GOTO, location),
label_(label)
{ }
protected:
int
do_traverse(Traverse*)
{ return TRAVERSE_CONTINUE; }
void
do_check_types(Gogo*);
bool
do_may_fall_through() const
{ return false; }
tree
do_get_tree(Translate_context*);
private:
Label* label_;
};
// Check types for a label. There aren't any types per se, but we use
// this to give an error if the label was never defined.
void
Goto_statement::do_check_types(Gogo*)
{
if (!this->label_->is_defined())
{
error_at(this->location(), "reference to undefined label %qs",
Gogo::message_name(this->label_->name()).c_str());
this->set_is_error();
}
}
// Return the tree for the goto statement.
tree
Goto_statement::do_get_tree(Translate_context*)
{
return this->build_stmt_1(GOTO_EXPR, this->label_->get_decl());
}
// Make a goto statement.
Statement*
Statement::make_goto_statement(Label* label, source_location location)
{
return new Goto_statement(label, location);
}
// A goto statement to an unnamed label.
class Goto_unnamed_statement : public Statement
{
public:
Goto_unnamed_statement(Unnamed_label* label, source_location location)
: Statement(STATEMENT_GOTO_UNNAMED, location),
label_(label)
{ }
protected:
int
do_traverse(Traverse*)
{ return TRAVERSE_CONTINUE; }
bool
do_may_fall_through() const
{ return false; }
tree
do_get_tree(Translate_context*)
{ return this->label_->get_goto(this->location()); }
private:
Unnamed_label* label_;
};
// Make a goto statement to an unnamed label.
Statement*
Statement::make_goto_unnamed_statement(Unnamed_label* label,
source_location location)
{
return new Goto_unnamed_statement(label, location);
}
// Class Label_statement.
// Traversal.
int
Label_statement::do_traverse(Traverse*)
{
return TRAVERSE_CONTINUE;
}
// Return a tree defining this label.
tree
Label_statement::do_get_tree(Translate_context*)
{
return this->build_stmt_1(LABEL_EXPR, this->label_->get_decl());
}
// Make a label statement.
Statement*
Statement::make_label_statement(Label* label, source_location location)
{
return new Label_statement(label, location);
}
// An unnamed label statement.
class Unnamed_label_statement : public Statement
{
public:
Unnamed_label_statement(Unnamed_label* label)
: Statement(STATEMENT_UNNAMED_LABEL, label->location()),
label_(label)
{ }
protected:
int
do_traverse(Traverse*)
{ return TRAVERSE_CONTINUE; }
tree
do_get_tree(Translate_context*)
{ return this->label_->get_definition(); }
private:
// The label.
Unnamed_label* label_;
};
// Make an unnamed label statement.
Statement*
Statement::make_unnamed_label_statement(Unnamed_label* label)
{
return new Unnamed_label_statement(label);
}
// An if statement.
class If_statement : public Statement
{
public:
If_statement(Expression* cond, Block* then_block, Block* else_block,
source_location location)
: Statement(STATEMENT_IF, location),
cond_(cond), then_block_(then_block), else_block_(else_block)
{ }
protected:
int
do_traverse(Traverse*);
void
do_determine_types();
void
do_check_types(Gogo*);
bool
do_may_fall_through() const;
tree
do_get_tree(Translate_context*);
private:
Expression* cond_;
Block* then_block_;
Block* else_block_;
};
// Traversal.
int
If_statement::do_traverse(Traverse* traverse)
{
if (this->cond_ != NULL)
{
if (this->traverse_expression(traverse, &this->cond_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
if (this->then_block_->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
if (this->else_block_ != NULL)
{
if (this->else_block_->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
return TRAVERSE_CONTINUE;
}
void
If_statement::do_determine_types()
{
if (this->cond_ != NULL)
{
Type_context context(Type::lookup_bool_type(), false);
this->cond_->determine_type(&context);
}
this->then_block_->determine_types();
if (this->else_block_ != NULL)
this->else_block_->determine_types();
}
// Check types.
void
If_statement::do_check_types(Gogo*)
{
if (this->cond_ != NULL)
{
Type* type = this->cond_->type();
if (type->is_error_type())
this->set_is_error();
else if (!type->is_boolean_type())
this->report_error(_("expected boolean expression"));
}
}
// Whether the overall statement may fall through.
bool
If_statement::do_may_fall_through() const
{
return (this->else_block_ == NULL
|| this->then_block_->may_fall_through()
|| this->else_block_->may_fall_through());
}
// Get tree.
tree
If_statement::do_get_tree(Translate_context* context)
{
gcc_assert(this->cond_ == NULL
|| this->cond_->type()->is_boolean_type()
|| this->cond_->type()->is_error_type());
tree cond_tree = (this->cond_ == NULL
? boolean_true_node
: this->cond_->get_tree(context));
tree then_tree = this->then_block_->get_tree(context);
tree else_tree = (this->else_block_ == NULL
? NULL_TREE
: this->else_block_->get_tree(context));
if (cond_tree == error_mark_node
|| then_tree == error_mark_node
|| else_tree == error_mark_node)
return error_mark_node;
tree ret = build3(COND_EXPR, void_type_node, cond_tree, then_tree,
else_tree);
SET_EXPR_LOCATION(ret, this->location());
return ret;
}
// Make an if statement.
Statement*
Statement::make_if_statement(Expression* cond, Block* then_block,
Block* else_block, source_location location)
{
return new If_statement(cond, then_block, else_block, location);
}
// Class Case_clauses::Case_clause.
// Traversal.
int
Case_clauses::Case_clause::traverse(Traverse* traverse)
{
if (this->cases_ != NULL
&& (traverse->traverse_mask()
& (Traverse::traverse_types | Traverse::traverse_expressions)) != 0)
{
if (this->cases_->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
if (this->statements_ != NULL)
{
if (this->statements_->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
return TRAVERSE_CONTINUE;
}
// Check whether all the case expressions are integer constants.
bool
Case_clauses::Case_clause::is_constant() const
{
if (this->cases_ != NULL)
{
for (Expression_list::const_iterator p = this->cases_->begin();
p != this->cases_->end();
++p)
if (!(*p)->is_constant() || (*p)->type()->integer_type() == NULL)
return false;
}
return true;
}
// Lower a case clause for a nonconstant switch. VAL_TEMP is the
// value we are switching on; it may be NULL. If START_LABEL is not
// NULL, it goes at the start of the statements, after the condition
// test. We branch to FINISH_LABEL at the end of the statements.
void
Case_clauses::Case_clause::lower(Block* b, Temporary_statement* val_temp,
Unnamed_label* start_label,
Unnamed_label* finish_label) const
{
source_location loc = this->location_;
Unnamed_label* next_case_label;
if (this->cases_ == NULL || this->cases_->empty())
{
gcc_assert(this->is_default_);
next_case_label = NULL;
}
else
{
Expression* cond = NULL;
for (Expression_list::const_iterator p = this->cases_->begin();
p != this->cases_->end();
++p)
{
Expression* this_cond;
if (val_temp == NULL)
this_cond = *p;
else
{
Expression* ref = Expression::make_temporary_reference(val_temp,
loc);
this_cond = Expression::make_binary(OPERATOR_EQEQ, ref, *p, loc);
}
if (cond == NULL)
cond = this_cond;
else
cond = Expression::make_binary(OPERATOR_OROR, cond, this_cond, loc);
}
Block* then_block = new Block(b, loc);
next_case_label = new Unnamed_label(UNKNOWN_LOCATION);
Statement* s = Statement::make_goto_unnamed_statement(next_case_label,
loc);
then_block->add_statement(s);
// if !COND { goto NEXT_CASE_LABEL }
cond = Expression::make_unary(OPERATOR_NOT, cond, loc);
s = Statement::make_if_statement(cond, then_block, NULL, loc);
b->add_statement(s);
}
if (start_label != NULL)
b->add_statement(Statement::make_unnamed_label_statement(start_label));
if (this->statements_ != NULL)
b->add_statement(Statement::make_block_statement(this->statements_, loc));
Statement* s = Statement::make_goto_unnamed_statement(finish_label, loc);
b->add_statement(s);
if (next_case_label != NULL)
b->add_statement(Statement::make_unnamed_label_statement(next_case_label));
}
// Determine types.
void
Case_clauses::Case_clause::determine_types(Type* type)
{
if (this->cases_ != NULL)
{
Type_context case_context(type, false);
for (Expression_list::iterator p = this->cases_->begin();
p != this->cases_->end();
++p)
(*p)->determine_type(&case_context);
}
if (this->statements_ != NULL)
this->statements_->determine_types();
}
// Check types. Returns false if there was an error.
bool
Case_clauses::Case_clause::check_types(Type* type)
{
if (this->cases_ != NULL)
{
for (Expression_list::iterator p = this->cases_->begin();
p != this->cases_->end();
++p)
{
if (!Type::are_assignable(type, (*p)->type(), NULL)
&& !Type::are_assignable((*p)->type(), type, NULL))
{
error_at((*p)->location(),
"type mismatch between switch value and case clause");
return false;
}
}
}
return true;
}
// Return true if this clause may fall through to the following
// statements. Note that this is not the same as whether the case
// uses the "fallthrough" keyword.
bool
Case_clauses::Case_clause::may_fall_through() const
{
if (this->statements_ == NULL)
return true;
return this->statements_->may_fall_through();
}
// Build up the body of a SWITCH_EXPR.
void
Case_clauses::Case_clause::get_constant_tree(Translate_context* context,
Unnamed_label* break_label,
Case_constants* case_constants,
tree* stmt_list) const
{
if (this->cases_ != NULL)
{
for (Expression_list::const_iterator p = this->cases_->begin();
p != this->cases_->end();
++p)
{
Type* itype;
mpz_t ival;
mpz_init(ival);
if (!(*p)->integer_constant_value(true, ival, &itype))
{
// Something went wrong. This can happen with a
// negative constant and an unsigned switch value.
gcc_assert(saw_errors());
continue;
}
gcc_assert(itype != NULL);
tree type_tree = itype->get_tree(context->gogo());
tree val = Expression::integer_constant_tree(ival, type_tree);
mpz_clear(ival);
if (val != error_mark_node)
{
gcc_assert(TREE_CODE(val) == INTEGER_CST);
std::pair<Case_constants::iterator, bool> ins =
case_constants->insert(val);
if (!ins.second)
{
// Value was already present.
warning_at(this->location_, 0,
"duplicate case value will never match");
continue;
}
tree label = create_artificial_label(this->location_);
append_to_statement_list(build3(CASE_LABEL_EXPR, void_type_node,
val, NULL_TREE, label),
stmt_list);
}
}
}
if (this->is_default_)
{
tree label = create_artificial_label(this->location_);
append_to_statement_list(build3(CASE_LABEL_EXPR, void_type_node,
NULL_TREE, NULL_TREE, label),
stmt_list);
}
if (this->statements_ != NULL)
{
tree block_tree = this->statements_->get_tree(context);
if (block_tree != error_mark_node)
append_to_statement_list(block_tree, stmt_list);
}
if (!this->is_fallthrough_)
append_to_statement_list(break_label->get_goto(this->location_), stmt_list);
}
// Class Case_clauses.
// Traversal.
int
Case_clauses::traverse(Traverse* traverse)
{
for (Clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
if (p->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
return TRAVERSE_CONTINUE;
}
// Check whether all the case expressions are constant.
bool
Case_clauses::is_constant() const
{
for (Clauses::const_iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
if (!p->is_constant())
return false;
return true;
}
// Lower case clauses for a nonconstant switch.
void
Case_clauses::lower(Block* b, Temporary_statement* val_temp,
Unnamed_label* break_label) const
{
// The default case.
const Case_clause* default_case = NULL;
// The label for the fallthrough of the previous case.
Unnamed_label* last_fallthrough_label = NULL;
// The label for the start of the default case. This is used if the
// case before the default case falls through.
Unnamed_label* default_start_label = NULL;
// The label for the end of the default case. This normally winds
// up as BREAK_LABEL, but it will be different if the default case
// falls through.
Unnamed_label* default_finish_label = NULL;
for (Clauses::const_iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
// The label to use for the start of the statements for this
// case. This is NULL unless the previous case falls through.
Unnamed_label* start_label = last_fallthrough_label;
// The label to jump to after the end of the statements for this
// case.
Unnamed_label* finish_label = break_label;
last_fallthrough_label = NULL;
if (p->is_fallthrough() && p + 1 != this->clauses_.end())
{
finish_label = new Unnamed_label(p->location());
last_fallthrough_label = finish_label;
}
if (!p->is_default())
p->lower(b, val_temp, start_label, finish_label);
else
{
// We have to move the default case to the end, so that we
// only use it if all the other tests fail.
default_case = &*p;
default_start_label = start_label;
default_finish_label = finish_label;
}
}
if (default_case != NULL)
default_case->lower(b, val_temp, default_start_label,
default_finish_label);
}
// Determine types.
void
Case_clauses::determine_types(Type* type)
{
for (Clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
p->determine_types(type);
}
// Check types. Returns false if there was an error.
bool
Case_clauses::check_types(Type* type)
{
bool ret = true;
for (Clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
if (!p->check_types(type))
ret = false;
}
return ret;
}
// Return true if these clauses may fall through to the statements
// following the switch statement.
bool
Case_clauses::may_fall_through() const
{
bool found_default = false;
for (Clauses::const_iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
if (p->may_fall_through() && !p->is_fallthrough())
return true;
if (p->is_default())
found_default = true;
}
return !found_default;
}
// Return a tree when all case expressions are constants.
tree
Case_clauses::get_constant_tree(Translate_context* context,
Unnamed_label* break_label) const
{
Case_constants case_constants;
tree stmt_list = NULL_TREE;
for (Clauses::const_iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
p->get_constant_tree(context, break_label, &case_constants,
&stmt_list);
return stmt_list;
}
// A constant switch statement. A Switch_statement is lowered to this
// when all the cases are constants.
class Constant_switch_statement : public Statement
{
public:
Constant_switch_statement(Expression* val, Case_clauses* clauses,
Unnamed_label* break_label,
source_location location)
: Statement(STATEMENT_CONSTANT_SWITCH, location),
val_(val), clauses_(clauses), break_label_(break_label)
{ }
protected:
int
do_traverse(Traverse*);
void
do_determine_types();
void
do_check_types(Gogo*);
bool
do_may_fall_through() const;
tree
do_get_tree(Translate_context*);
private:
// The value to switch on.
Expression* val_;
// The case clauses.
Case_clauses* clauses_;
// The break label, if needed.
Unnamed_label* break_label_;
};
// Traversal.
int
Constant_switch_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->clauses_->traverse(traverse);
}
// Determine types.
void
Constant_switch_statement::do_determine_types()
{
this->val_->determine_type_no_context();
this->clauses_->determine_types(this->val_->type());
}
// Check types.
void
Constant_switch_statement::do_check_types(Gogo*)
{
if (!this->clauses_->check_types(this->val_->type()))
this->set_is_error();
}
// Return whether this switch may fall through.
bool
Constant_switch_statement::do_may_fall_through() const
{
if (this->clauses_ == NULL)
return true;
// If we have a break label, then some case needed it. That implies
// that the switch statement as a whole can fall through.
if (this->break_label_ != NULL)
return true;
return this->clauses_->may_fall_through();
}
// Convert to GENERIC.
tree
Constant_switch_statement::do_get_tree(Translate_context* context)
{
tree switch_val_tree = this->val_->get_tree(context);
Unnamed_label* break_label = this->break_label_;
if (break_label == NULL)
break_label = new Unnamed_label(this->location());
tree stmt_list = NULL_TREE;
tree s = build3(SWITCH_EXPR, void_type_node, switch_val_tree,
this->clauses_->get_constant_tree(context, break_label),
NULL_TREE);
SET_EXPR_LOCATION(s, this->location());
append_to_statement_list(s, &stmt_list);
append_to_statement_list(break_label->get_definition(), &stmt_list);
return stmt_list;
}
// Class Switch_statement.
// Traversal.
int
Switch_statement::do_traverse(Traverse* traverse)
{
if (this->val_ != NULL)
{
if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
return this->clauses_->traverse(traverse);
}
// Lower a Switch_statement to a Constant_switch_statement or a series
// of if statements.
Statement*
Switch_statement::do_lower(Gogo*, Block* enclosing)
{
source_location loc = this->location();
if (this->val_ != NULL
&& (this->val_->is_error_expression()
|| this->val_->type()->is_error_type()))
return Statement::make_error_statement(loc);
if (this->val_ != NULL
&& this->val_->type()->integer_type() != NULL
&& !this->clauses_->empty()
&& this->clauses_->is_constant())
return new Constant_switch_statement(this->val_, this->clauses_,
this->break_label_, loc);
Block* b = new Block(enclosing, loc);
if (this->clauses_->empty())
{
Expression* val = this->val_;
if (val == NULL)
val = Expression::make_boolean(true, loc);
return Statement::make_statement(val);
}
Temporary_statement* val_temp;
if (this->val_ == NULL)
val_temp = NULL;
else
{
// var val_temp VAL_TYPE = VAL
val_temp = Statement::make_temporary(NULL, this->val_, loc);
b->add_statement(val_temp);
}
this->clauses_->lower(b, val_temp, this->break_label());
Statement* s = Statement::make_unnamed_label_statement(this->break_label_);
b->add_statement(s);
return Statement::make_block_statement(b, loc);
}
// Return the break label for this switch statement, creating it if
// necessary.
Unnamed_label*
Switch_statement::break_label()
{
if (this->break_label_ == NULL)
this->break_label_ = new Unnamed_label(this->location());
return this->break_label_;
}
// Make a switch statement.
Switch_statement*
Statement::make_switch_statement(Expression* val, source_location location)
{
return new Switch_statement(val, location);
}
// Class Type_case_clauses::Type_case_clause.
// Traversal.
int
Type_case_clauses::Type_case_clause::traverse(Traverse* traverse)
{
if (!this->is_default_
&& ((traverse->traverse_mask()
& (Traverse::traverse_types | Traverse::traverse_expressions)) != 0)
&& Type::traverse(this->type_, traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
if (this->statements_ != NULL)
return this->statements_->traverse(traverse);
return TRAVERSE_CONTINUE;
}
// Lower one clause in a type switch. Add statements to the block B.
// The type descriptor we are switching on is in DESCRIPTOR_TEMP.
// BREAK_LABEL is the label at the end of the type switch.
// *STMTS_LABEL, if not NULL, is a label to put at the start of the
// statements.
void
Type_case_clauses::Type_case_clause::lower(Block* b,
Temporary_statement* descriptor_temp,
Unnamed_label* break_label,
Unnamed_label** stmts_label) const
{
source_location loc = this->location_;
Unnamed_label* next_case_label = NULL;
if (!this->is_default_)
{
Type* type = this->type_;
Expression* cond;
// The language permits case nil, which is of course a constant
// rather than a type. It will appear here as an invalid
// forwarding type.
if (type->is_nil_constant_as_type())
{
Expression* ref =
Expression::make_temporary_reference(descriptor_temp, loc);
cond = Expression::make_binary(OPERATOR_EQEQ, ref,
Expression::make_nil(loc),
loc);
}
else
{
Expression* func;
if (type->interface_type() == NULL)
{
// func ifacetypeeq(*descriptor, *descriptor) bool
static Named_object* ifacetypeeq;
if (ifacetypeeq == NULL)
{
const source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types =
new Typed_identifier_list();
Type* descriptor_type = Type::make_type_descriptor_ptr_type();
param_types->push_back(Typed_identifier("a", descriptor_type,
bloc));
param_types->push_back(Typed_identifier("b", descriptor_type,
bloc));
Typed_identifier_list* ret_types =
new Typed_identifier_list();
Type* bool_type = Type::lookup_bool_type();
ret_types->push_back(Typed_identifier("", bool_type, bloc));
Function_type* fntype = Type::make_function_type(NULL,
param_types,
ret_types,
bloc);
ifacetypeeq =
Named_object::make_function_declaration("ifacetypeeq", NULL,
fntype, bloc);
const char* n = "runtime.ifacetypeeq";
ifacetypeeq->func_declaration_value()->set_asm_name(n);
}
// ifacetypeeq(descriptor_temp, DESCRIPTOR)
func = Expression::make_func_reference(ifacetypeeq, NULL, loc);
}
else
{
// func ifaceI2Tp(*descriptor, *descriptor) bool
static Named_object* ifaceI2Tp;
if (ifaceI2Tp == NULL)
{
const source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types =
new Typed_identifier_list();
Type* descriptor_type = Type::make_type_descriptor_ptr_type();
param_types->push_back(Typed_identifier("a", descriptor_type,
bloc));
param_types->push_back(Typed_identifier("b", descriptor_type,
bloc));
Typed_identifier_list* ret_types =
new Typed_identifier_list();
Type* bool_type = Type::lookup_bool_type();
ret_types->push_back(Typed_identifier("", bool_type, bloc));
Function_type* fntype = Type::make_function_type(NULL,
param_types,
ret_types,
bloc);
ifaceI2Tp =
Named_object::make_function_declaration("ifaceI2Tp", NULL,
fntype, bloc);
const char* n = "runtime.ifaceI2Tp";
ifaceI2Tp->func_declaration_value()->set_asm_name(n);
}
// ifaceI2Tp(descriptor_temp, DESCRIPTOR)
func = Expression::make_func_reference(ifaceI2Tp, NULL, loc);
}
Expression_list* params = new Expression_list();
params->push_back(Expression::make_type_descriptor(type, loc));
Expression* ref =
Expression::make_temporary_reference(descriptor_temp, loc);
params->push_back(ref);
cond = Expression::make_call(func, params, false, loc);
}
Unnamed_label* dest;
if (!this->is_fallthrough_)
{
// if !COND { goto NEXT_CASE_LABEL }
next_case_label = new Unnamed_label(UNKNOWN_LOCATION);
dest = next_case_label;
cond = Expression::make_unary(OPERATOR_NOT, cond, loc);
}
else
{
// if COND { goto STMTS_LABEL }
gcc_assert(stmts_label != NULL);
if (*stmts_label == NULL)
*stmts_label = new Unnamed_label(UNKNOWN_LOCATION);
dest = *stmts_label;
}
Block* then_block = new Block(b, loc);
Statement* s = Statement::make_goto_unnamed_statement(dest, loc);
then_block->add_statement(s);
s = Statement::make_if_statement(cond, then_block, NULL, loc);
b->add_statement(s);
}
if (this->statements_ != NULL
|| (!this->is_fallthrough_
&& stmts_label != NULL
&& *stmts_label != NULL))
{
gcc_assert(!this->is_fallthrough_);
if (stmts_label != NULL && *stmts_label != NULL)
{
gcc_assert(!this->is_default_);
if (this->statements_ != NULL)
(*stmts_label)->set_location(this->statements_->start_location());
Statement* s = Statement::make_unnamed_label_statement(*stmts_label);
b->add_statement(s);
*stmts_label = NULL;
}
if (this->statements_ != NULL)
b->add_statement(Statement::make_block_statement(this->statements_,
loc));
}
if (this->is_fallthrough_)
gcc_assert(next_case_label == NULL);
else
{
source_location gloc = (this->statements_ == NULL
? loc
: this->statements_->end_location());
b->add_statement(Statement::make_goto_unnamed_statement(break_label,
gloc));
if (next_case_label != NULL)
{
Statement* s =
Statement::make_unnamed_label_statement(next_case_label);
b->add_statement(s);
}
}
}
// Class Type_case_clauses.
// Traversal.
int
Type_case_clauses::traverse(Traverse* traverse)
{
for (Type_clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
if (p->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
return TRAVERSE_CONTINUE;
}
// Check for duplicate types.
void
Type_case_clauses::check_duplicates() const
{
typedef Unordered_set_hash(const Type*, Type_hash_identical,
Type_identical) Types_seen;
Types_seen types_seen;
for (Type_clauses::const_iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
Type* t = p->type();
if (t == NULL)
continue;
if (t->is_nil_constant_as_type())
t = Type::make_nil_type();
std::pair<Types_seen::iterator, bool> ins = types_seen.insert(t);
if (!ins.second)
error_at(p->location(), "duplicate type in switch");
}
}
// Lower the clauses in a type switch. Add statements to the block B.
// The type descriptor we are switching on is in DESCRIPTOR_TEMP.
// BREAK_LABEL is the label at the end of the type switch.
void
Type_case_clauses::lower(Block* b, Temporary_statement* descriptor_temp,
Unnamed_label* break_label) const
{
const Type_case_clause* default_case = NULL;
Unnamed_label* stmts_label = NULL;
for (Type_clauses::const_iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
if (!p->is_default())
p->lower(b, descriptor_temp, break_label, &stmts_label);
else
{
// We are generating a series of tests, which means that we
// need to move the default case to the end.
default_case = &*p;
}
}
gcc_assert(stmts_label == NULL);
if (default_case != NULL)
default_case->lower(b, descriptor_temp, break_label, NULL);
}
// Class Type_switch_statement.
// Traversal.
int
Type_switch_statement::do_traverse(Traverse* traverse)
{
if (this->var_ == NULL)
{
if (this->traverse_expression(traverse, &this->expr_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
if (this->clauses_ != NULL)
return this->clauses_->traverse(traverse);
return TRAVERSE_CONTINUE;
}
// Lower a type switch statement to a series of if statements. The gc
// compiler is able to generate a table in some cases. However, that
// does not work for us because we may have type descriptors in
// different shared libraries, so we can't compare them with simple
// equality testing.
Statement*
Type_switch_statement::do_lower(Gogo*, Block* enclosing)
{
const source_location loc = this->location();
if (this->clauses_ != NULL)
this->clauses_->check_duplicates();
Block* b = new Block(enclosing, loc);
Type* val_type = (this->var_ != NULL
? this->var_->var_value()->type()
: this->expr_->type());
// var descriptor_temp DESCRIPTOR_TYPE
Type* descriptor_type = Type::make_type_descriptor_ptr_type();
Temporary_statement* descriptor_temp =
Statement::make_temporary(descriptor_type, NULL, loc);
b->add_statement(descriptor_temp);
if (val_type->interface_type() == NULL)
{
// Doing a type switch on a non-interface type. Should we issue
// a warning for this case?
Expression* lhs = Expression::make_temporary_reference(descriptor_temp,
loc);
Expression* rhs;
if (val_type->is_nil_type())
rhs = Expression::make_nil(loc);
else
{
if (val_type->is_abstract())
val_type = val_type->make_non_abstract_type();
rhs = Expression::make_type_descriptor(val_type, loc);
}
Statement* s = Statement::make_assignment(lhs, rhs, loc);
b->add_statement(s);
}
else
{
const source_location bloc = BUILTINS_LOCATION;
// func {efacetype,ifacetype}(*interface) *descriptor
// FIXME: This should be inlined.
Typed_identifier_list* param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("i", val_type, bloc));
Typed_identifier_list* ret_types = new Typed_identifier_list();
ret_types->push_back(Typed_identifier("", descriptor_type, bloc));
Function_type* fntype = Type::make_function_type(NULL, param_types,
ret_types, bloc);
bool is_empty = val_type->interface_type()->is_empty();
const char* fnname = is_empty ? "efacetype" : "ifacetype";
Named_object* fn =
Named_object::make_function_declaration(fnname, NULL, fntype, bloc);
const char* asm_name = (is_empty
? "runtime.efacetype"
: "runtime.ifacetype");
fn->func_declaration_value()->set_asm_name(asm_name);
// descriptor_temp = ifacetype(val_temp)
Expression* func = Expression::make_func_reference(fn, NULL, loc);
Expression_list* params = new Expression_list();
Expression* ref;
if (this->var_ == NULL)
ref = this->expr_;
else
ref = Expression::make_var_reference(this->var_, loc);
params->push_back(ref);
Expression* call = Expression::make_call(func, params, false, loc);
Expression* lhs = Expression::make_temporary_reference(descriptor_temp,
loc);
Statement* s = Statement::make_assignment(lhs, call, loc);
b->add_statement(s);
}
if (this->clauses_ != NULL)
this->clauses_->lower(b, descriptor_temp, this->break_label());
Statement* s = Statement::make_unnamed_label_statement(this->break_label_);
b->add_statement(s);
return Statement::make_block_statement(b, loc);
}
// Return the break label for this type switch statement, creating it
// if necessary.
Unnamed_label*
Type_switch_statement::break_label()
{
if (this->break_label_ == NULL)
this->break_label_ = new Unnamed_label(this->location());
return this->break_label_;
}
// Make a type switch statement.
Type_switch_statement*
Statement::make_type_switch_statement(Named_object* var, Expression* expr,
source_location location)
{
return new Type_switch_statement(var, expr, location);
}
// Class Select_clauses::Select_clause.
// Traversal.
int
Select_clauses::Select_clause::traverse(Traverse* traverse)
{
if (!this->is_lowered_
&& (traverse->traverse_mask()
& (Traverse::traverse_types | Traverse::traverse_expressions)) != 0)
{
if (this->channel_ != NULL)
{
if (Expression::traverse(&this->channel_, traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
if (this->val_ != NULL)
{
if (Expression::traverse(&this->val_, traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
}
if (this->statements_ != NULL)
{
if (this->statements_->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
return TRAVERSE_CONTINUE;
}
// Lowering. Here we pull out the channel and the send values, to
// enforce the order of evaluation. We also add explicit send and
// receive statements to the clauses.
void
Select_clauses::Select_clause::lower(Block* b)
{
if (this->is_default_)
{
gcc_assert(this->channel_ == NULL && this->val_ == NULL);
this->is_lowered_ = true;
return;
}
source_location loc = this->location_;
// Evaluate the channel before the select statement.
Temporary_statement* channel_temp = Statement::make_temporary(NULL,
this->channel_,
loc);
b->add_statement(channel_temp);
this->channel_ = Expression::make_temporary_reference(channel_temp, loc);
// If this is a send clause, evaluate the value to send before the
// select statement.
Temporary_statement* val_temp = NULL;
if (this->is_send_)
{
val_temp = Statement::make_temporary(NULL, this->val_, loc);
b->add_statement(val_temp);
}
// Add the send or receive before the rest of the statements if any.
Block *init = new Block(b, loc);
Expression* ref = Expression::make_temporary_reference(channel_temp, loc);
if (this->is_send_)
{
Expression* ref2 = Expression::make_temporary_reference(val_temp, loc);
Send_expression* send = Expression::make_send(ref, ref2, loc);
send->discarding_value();
send->set_for_select();
init->add_statement(Statement::make_statement(send));
}
else
{
Receive_expression* recv = Expression::make_receive(ref, loc);
recv->set_for_select();
if (this->val_ != NULL)
{
gcc_assert(this->var_ == NULL);
init->add_statement(Statement::make_assignment(this->val_, recv,
loc));
}
else if (this->var_ != NULL)
{
this->var_->var_value()->set_init(recv);
this->var_->var_value()->clear_type_from_chan_element();
}
else
{
recv->discarding_value();
init->add_statement(Statement::make_statement(recv));
}
}
if (this->statements_ != NULL)
init->add_statement(Statement::make_block_statement(this->statements_,
loc));
this->statements_ = init;
// Now all references should be handled through the statements, not
// through here.
this->is_lowered_ = true;
this->val_ = NULL;
this->var_ = NULL;
}
// Determine types.
void
Select_clauses::Select_clause::determine_types()
{
gcc_assert(this->is_lowered_);
if (this->statements_ != NULL)
this->statements_->determine_types();
}
// Whether this clause may fall through to the statement which follows
// the overall select statement.
bool
Select_clauses::Select_clause::may_fall_through() const
{
if (this->statements_ == NULL)
return true;
return this->statements_->may_fall_through();
}
// Return a tree for the statements to execute.
tree
Select_clauses::Select_clause::get_statements_tree(Translate_context* context)
{
if (this->statements_ == NULL)
return NULL_TREE;
return this->statements_->get_tree(context);
}
// Class Select_clauses.
// Traversal.
int
Select_clauses::traverse(Traverse* traverse)
{
for (Clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
if (p->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
return TRAVERSE_CONTINUE;
}
// Lowering. Here we pull out the channel and the send values, to
// enforce the order of evaluation. We also add explicit send and
// receive statements to the clauses.
void
Select_clauses::lower(Block* b)
{
for (Clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
p->lower(b);
}
// Determine types.
void
Select_clauses::determine_types()
{
for (Clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
p->determine_types();
}
// Return whether these select clauses fall through to the statement
// following the overall select statement.
bool
Select_clauses::may_fall_through() const
{
for (Clauses::const_iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
if (p->may_fall_through())
return true;
return false;
}
// Return a tree. We build a call to
// size_t __go_select(size_t count, _Bool has_default,
// channel* channels, _Bool* is_send)
//
// There are COUNT entries in the CHANNELS and IS_SEND arrays. The
// value in the IS_SEND array is true for send, false for receive.
// __go_select returns an integer from 0 to COUNT, inclusive. A
// return of 0 means that the default case should be run; this only
// happens if HAS_DEFAULT is non-zero. Otherwise the number indicates
// the case to run.
// FIXME: This doesn't handle channels which send interface types
// where the receiver has a static type which matches that interface.
tree
Select_clauses::get_tree(Translate_context* context,
Unnamed_label *break_label,
source_location location)
{
size_t count = this->clauses_.size();
VEC(constructor_elt, gc)* chan_init = VEC_alloc(constructor_elt, gc, count);
VEC(constructor_elt, gc)* is_send_init = VEC_alloc(constructor_elt, gc,
count);
Select_clause* default_clause = NULL;
tree final_stmt_list = NULL_TREE;
tree channel_type_tree = NULL_TREE;
size_t i = 0;
for (Clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
if (p->is_default())
{
default_clause = &*p;
--count;
continue;
}
if (p->channel()->type()->channel_type() == NULL)
{
// We should have given an error in the send or receive
// statement we created via lowering.
gcc_assert(saw_errors());
return error_mark_node;
}
tree channel_tree = p->channel()->get_tree(context);
if (channel_tree == error_mark_node)
return error_mark_node;
channel_type_tree = TREE_TYPE(channel_tree);
constructor_elt* elt = VEC_quick_push(constructor_elt, chan_init, NULL);
elt->index = build_int_cstu(sizetype, i);
elt->value = channel_tree;
elt = VEC_quick_push(constructor_elt, is_send_init, NULL);
elt->index = build_int_cstu(sizetype, i);
elt->value = p->is_send() ? boolean_true_node : boolean_false_node;
++i;
}
gcc_assert(i == count);
if (i == 0 && default_clause != NULL)
{
// There is only a default clause.
gcc_assert(final_stmt_list == NULL_TREE);
tree stmt_list = NULL_TREE;
append_to_statement_list(default_clause->get_statements_tree(context),
&stmt_list);
append_to_statement_list(break_label->get_definition(), &stmt_list);
return stmt_list;
}
tree pointer_chan_type_tree = (channel_type_tree == NULL_TREE
? ptr_type_node
: build_pointer_type(channel_type_tree));
tree chans_arg;
tree pointer_boolean_type_tree = build_pointer_type(boolean_type_node);
tree is_sends_arg;
if (i == 0)
{
chans_arg = fold_convert_loc(location, pointer_chan_type_tree,
null_pointer_node);
is_sends_arg = fold_convert_loc(location, pointer_boolean_type_tree,
null_pointer_node);
}
else
{
tree index_type_tree = build_index_type(size_int(count - 1));
tree chan_array_type_tree = build_array_type(channel_type_tree,
index_type_tree);
tree chan_constructor = build_constructor(chan_array_type_tree,
chan_init);
tree chan_var = create_tmp_var(chan_array_type_tree, "CHAN");
DECL_IGNORED_P(chan_var) = 0;
DECL_INITIAL(chan_var) = chan_constructor;
DECL_SOURCE_LOCATION(chan_var) = location;
TREE_ADDRESSABLE(chan_var) = 1;
tree decl_expr = build1(DECL_EXPR, void_type_node, chan_var);
SET_EXPR_LOCATION(decl_expr, location);
append_to_statement_list(decl_expr, &final_stmt_list);
tree is_send_array_type_tree = build_array_type(boolean_type_node,
index_type_tree);
tree is_send_constructor = build_constructor(is_send_array_type_tree,
is_send_init);
tree is_send_var = create_tmp_var(is_send_array_type_tree, "ISSEND");
DECL_IGNORED_P(is_send_var) = 0;
DECL_INITIAL(is_send_var) = is_send_constructor;
DECL_SOURCE_LOCATION(is_send_var) = location;
TREE_ADDRESSABLE(is_send_var) = 1;
decl_expr = build1(DECL_EXPR, void_type_node, is_send_var);
SET_EXPR_LOCATION(decl_expr, location);
append_to_statement_list(decl_expr, &final_stmt_list);
chans_arg = fold_convert_loc(location, pointer_chan_type_tree,
build_fold_addr_expr_loc(location,
chan_var));
is_sends_arg = fold_convert_loc(location, pointer_boolean_type_tree,
build_fold_addr_expr_loc(location,
is_send_var));
}
static tree select_fndecl;
tree call = Gogo::call_builtin(&select_fndecl,
location,
"__go_select",
4,
sizetype,
sizetype,
size_int(count),
boolean_type_node,
(default_clause == NULL
? boolean_false_node
: boolean_true_node),
pointer_chan_type_tree,
chans_arg,
pointer_boolean_type_tree,
is_sends_arg);
if (call == error_mark_node)
return error_mark_node;
tree stmt_list = NULL_TREE;
if (default_clause != NULL)
this->add_clause_tree(context, 0, default_clause, break_label, &stmt_list);
i = 1;
for (Clauses::iterator p = this->clauses_.begin();
p != this->clauses_.end();
++p)
{
if (!p->is_default())
{
this->add_clause_tree(context, i, &*p, break_label, &stmt_list);
++i;
}
}
append_to_statement_list(break_label->get_definition(), &stmt_list);
tree switch_stmt = build3(SWITCH_EXPR, sizetype, call, stmt_list, NULL_TREE);
SET_EXPR_LOCATION(switch_stmt, location);
append_to_statement_list(switch_stmt, &final_stmt_list);
return final_stmt_list;
}
// Add the tree for CLAUSE to STMT_LIST.
void
Select_clauses::add_clause_tree(Translate_context* context, int case_index,
Select_clause* clause,
Unnamed_label* bottom_label, tree* stmt_list)
{
tree label = create_artificial_label(clause->location());
append_to_statement_list(build3(CASE_LABEL_EXPR, void_type_node,
build_int_cst(sizetype, case_index),
NULL_TREE, label),
stmt_list);
append_to_statement_list(clause->get_statements_tree(context), stmt_list);
tree g = bottom_label->get_goto(clause->statements() == NULL
? clause->location()
: clause->statements()->end_location());
append_to_statement_list(g, stmt_list);
}
// Class Select_statement.
// Return the break label for this switch statement, creating it if
// necessary.
Unnamed_label*
Select_statement::break_label()
{
if (this->break_label_ == NULL)
this->break_label_ = new Unnamed_label(this->location());
return this->break_label_;
}
// Lower a select statement. This will still return a select
// statement, but it will be modified to implement the order of
// evaluation rules, and to include the send and receive statements as
// explicit statements in the clauses.
Statement*
Select_statement::do_lower(Gogo*, Block* enclosing)
{
if (this->is_lowered_)
return this;
Block* b = new Block(enclosing, this->location());
this->clauses_->lower(b);
this->is_lowered_ = true;
b->add_statement(this);
return Statement::make_block_statement(b, this->location());
}
// Return the tree for a select statement.
tree
Select_statement::do_get_tree(Translate_context* context)
{
return this->clauses_->get_tree(context, this->break_label(),
this->location());
}
// Make a select statement.
Select_statement*
Statement::make_select_statement(source_location location)
{
return new Select_statement(location);
}
// Class For_statement.
// Traversal.
int
For_statement::do_traverse(Traverse* traverse)
{
if (this->init_ != NULL)
{
if (this->init_->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
if (this->cond_ != NULL)
{
if (this->traverse_expression(traverse, &this->cond_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
if (this->post_ != NULL)
{
if (this->post_->traverse(traverse) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
return this->statements_->traverse(traverse);
}
// Lower a For_statement into if statements and gotos. Getting rid of
// complex statements make it easier to handle garbage collection.
Statement*
For_statement::do_lower(Gogo*, Block* enclosing)
{
Statement* s;
source_location loc = this->location();
Block* b = new Block(enclosing, this->location());
if (this->init_ != NULL)
{
s = Statement::make_block_statement(this->init_,
this->init_->start_location());
b->add_statement(s);
}
Unnamed_label* entry = NULL;
if (this->cond_ != NULL)
{
entry = new Unnamed_label(this->location());
b->add_statement(Statement::make_goto_unnamed_statement(entry, loc));
}
Unnamed_label* top = new Unnamed_label(this->location());
b->add_statement(Statement::make_unnamed_label_statement(top));
s = Statement::make_block_statement(this->statements_,
this->statements_->start_location());
b->add_statement(s);
source_location end_loc = this->statements_->end_location();
Unnamed_label* cont = this->continue_label_;
if (cont != NULL)
b->add_statement(Statement::make_unnamed_label_statement(cont));
if (this->post_ != NULL)
{
s = Statement::make_block_statement(this->post_,
this->post_->start_location());
b->add_statement(s);
end_loc = this->post_->end_location();
}
if (this->cond_ == NULL)
b->add_statement(Statement::make_goto_unnamed_statement(top, end_loc));
else
{
b->add_statement(Statement::make_unnamed_label_statement(entry));
source_location cond_loc = this->cond_->location();
Block* then_block = new Block(b, cond_loc);
s = Statement::make_goto_unnamed_statement(top, cond_loc);
then_block->add_statement(s);
s = Statement::make_if_statement(this->cond_, then_block, NULL, cond_loc);
b->add_statement(s);
}
Unnamed_label* brk = this->break_label_;
if (brk != NULL)
b->add_statement(Statement::make_unnamed_label_statement(brk));
b->set_end_location(end_loc);
return Statement::make_block_statement(b, loc);
}
// Return the break label, creating it if necessary.
Unnamed_label*
For_statement::break_label()
{
if (this->break_label_ == NULL)
this->break_label_ = new Unnamed_label(this->location());
return this->break_label_;
}
// Return the continue LABEL_EXPR.
Unnamed_label*
For_statement::continue_label()
{
if (this->continue_label_ == NULL)
this->continue_label_ = new Unnamed_label(this->location());
return this->continue_label_;
}
// Set the break and continue labels a for statement. This is used
// when lowering a for range statement.
void
For_statement::set_break_continue_labels(Unnamed_label* break_label,
Unnamed_label* continue_label)
{
gcc_assert(this->break_label_ == NULL && this->continue_label_ == NULL);
this->break_label_ = break_label;
this->continue_label_ = continue_label;
}
// Make a for statement.
For_statement*
Statement::make_for_statement(Block* init, Expression* cond, Block* post,
source_location location)
{
return new For_statement(init, cond, post, location);
}
// Class For_range_statement.
// Traversal.
int
For_range_statement::do_traverse(Traverse* traverse)
{
if (this->traverse_expression(traverse, &this->index_var_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
if (this->value_var_ != NULL)
{
if (this->traverse_expression(traverse, &this->value_var_)
== TRAVERSE_EXIT)
return TRAVERSE_EXIT;
}
if (this->traverse_expression(traverse, &this->range_) == TRAVERSE_EXIT)
return TRAVERSE_EXIT;
return this->statements_->traverse(traverse);
}
// Lower a for range statement. For simplicity we lower this into a
// for statement, which will then be lowered in turn to goto
// statements.
Statement*
For_range_statement::do_lower(Gogo* gogo, Block* enclosing)
{
Type* range_type = this->range_->type();
if (range_type->points_to() != NULL
&& range_type->points_to()->array_type() != NULL
&& !range_type->points_to()->is_open_array_type())
range_type = range_type->points_to();
Type* index_type;
Type* value_type = NULL;
if (range_type->array_type() != NULL)
{
index_type = Type::lookup_integer_type("int");
value_type = range_type->array_type()->element_type();
}
else if (range_type->is_string_type())
{
index_type = Type::lookup_integer_type("int");
value_type = index_type;
}
else if (range_type->map_type() != NULL)
{
index_type = range_type->map_type()->key_type();
value_type = range_type->map_type()->val_type();
}
else if (range_type->channel_type() != NULL)
{
index_type = range_type->channel_type()->element_type();
if (this->value_var_ != NULL)
{
if (!this->value_var_->type()->is_error_type())
this->report_error(_("too many variables for range clause "
"with channel"));
return Statement::make_error_statement(this->location());
}
}
else
{
this->report_error(_("range clause must have "
"array, slice, setring, map, or channel type"));
return Statement::make_error_statement(this->location());
}
source_location loc = this->location();
Block* temp_block = new Block(enclosing, loc);
Named_object* range_object = NULL;
Temporary_statement* range_temp = NULL;
Var_expression* ve = this->range_->var_expression();
if (ve != NULL)
range_object = ve->named_object();
else
{
range_temp = Statement::make_temporary(NULL, this->range_, loc);
temp_block->add_statement(range_temp);
}
Temporary_statement* index_temp = Statement::make_temporary(index_type,
NULL, loc);
temp_block->add_statement(index_temp);
Temporary_statement* value_temp = NULL;
if (this->value_var_ != NULL)
{
value_temp = Statement::make_temporary(value_type, NULL, loc);
temp_block->add_statement(value_temp);
}
Block* body = new Block(temp_block, loc);
Block* init;
Expression* cond;
Block* iter_init;
Block* post;
// Arrange to do a loop appropriate for the type. We will produce
// for INIT ; COND ; POST {
// ITER_INIT
// INDEX = INDEX_TEMP
// VALUE = VALUE_TEMP // If there is a value
// original statements
// }
if (range_type->array_type() != NULL)
this->lower_range_array(gogo, temp_block, body, range_object, range_temp,
index_temp, value_temp, &init, &cond, &iter_init,
&post);
else if (range_type->is_string_type())
this->lower_range_string(gogo, temp_block, body, range_object, range_temp,
index_temp, value_temp, &init, &cond, &iter_init,
&post);
else if (range_type->map_type() != NULL)
this->lower_range_map(gogo, temp_block, body, range_object, range_temp,
index_temp, value_temp, &init, &cond, &iter_init,
&post);
else if (range_type->channel_type() != NULL)
this->lower_range_channel(gogo, temp_block, body, range_object, range_temp,
index_temp, value_temp, &init, &cond, &iter_init,
&post);
else
gcc_unreachable();
if (iter_init != NULL)
body->add_statement(Statement::make_block_statement(iter_init, loc));
Statement* assign;
Expression* index_ref = Expression::make_temporary_reference(index_temp, loc);
if (this->value_var_ == NULL)
{
assign = Statement::make_assignment(this->index_var_, index_ref, loc);
}
else
{
Expression_list* lhs = new Expression_list();
lhs->push_back(this->index_var_);
lhs->push_back(this->value_var_);
Expression_list* rhs = new Expression_list();
rhs->push_back(index_ref);
rhs->push_back(Expression::make_temporary_reference(value_temp, loc));
assign = Statement::make_tuple_assignment(lhs, rhs, loc);
}
body->add_statement(assign);
body->add_statement(Statement::make_block_statement(this->statements_, loc));
body->set_end_location(this->statements_->end_location());
For_statement* loop = Statement::make_for_statement(init, cond, post,
this->location());
loop->add_statements(body);
loop->set_break_continue_labels(this->break_label_, this->continue_label_);
temp_block->add_statement(loop);
return Statement::make_block_statement(temp_block, loc);
}
// Return a reference to the range, which may be in RANGE_OBJECT or in
// RANGE_TEMP.
Expression*
For_range_statement::make_range_ref(Named_object* range_object,
Temporary_statement* range_temp,
source_location loc)
{
if (range_object != NULL)
return Expression::make_var_reference(range_object, loc);
else
return Expression::make_temporary_reference(range_temp, loc);
}
// Return a call to the predeclared function FUNCNAME passing a
// reference to the temporary variable ARG.
Expression*
For_range_statement::call_builtin(Gogo* gogo, const char* funcname,
Expression* arg,
source_location loc)
{
Named_object* no = gogo->lookup_global(funcname);
gcc_assert(no != NULL && no->is_function_declaration());
Expression* func = Expression::make_func_reference(no, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(arg);
return Expression::make_call(func, params, false, loc);
}
// Lower a for range over an array or slice.
void
For_range_statement::lower_range_array(Gogo* gogo,
Block* enclosing,
Block* body_block,
Named_object* range_object,
Temporary_statement* range_temp,
Temporary_statement* index_temp,
Temporary_statement* value_temp,
Block** pinit,
Expression** pcond,
Block** piter_init,
Block** ppost)
{
source_location loc = this->location();
// The loop we generate:
// len_temp := len(range)
// for index_temp = 0; index_temp < len_temp; index_temp++ {
// value_temp = range[index_temp]
// index = index_temp
// value = value_temp
// original body
// }
// Set *PINIT to
// var len_temp int
// len_temp = len(range)
// index_temp = 0
Block* init = new Block(enclosing, loc);
Expression* ref = this->make_range_ref(range_object, range_temp, loc);
Expression* len_call = this->call_builtin(gogo, "len", ref, loc);
Temporary_statement* len_temp = Statement::make_temporary(index_temp->type(),
len_call, loc);
init->add_statement(len_temp);
mpz_t zval;
mpz_init_set_ui(zval, 0UL);
Expression* zexpr = Expression::make_integer(&zval, NULL, loc);
mpz_clear(zval);
ref = Expression::make_temporary_reference(index_temp, loc);
Statement* s = Statement::make_assignment(ref, zexpr, loc);
init->add_statement(s);
*pinit = init;
// Set *PCOND to
// index_temp < len_temp
ref = Expression::make_temporary_reference(index_temp, loc);
Expression* ref2 = Expression::make_temporary_reference(len_temp, loc);
Expression* lt = Expression::make_binary(OPERATOR_LT, ref, ref2, loc);
*pcond = lt;
// Set *PITER_INIT to
// value_temp = range[index_temp]
Block* iter_init = NULL;
if (value_temp != NULL)
{
iter_init = new Block(body_block, loc);
ref = this->make_range_ref(range_object, range_temp, loc);
Expression* ref2 = Expression::make_temporary_reference(index_temp, loc);
Expression* index = Expression::make_index(ref, ref2, NULL, loc);
ref = Expression::make_temporary_reference(value_temp, loc);
s = Statement::make_assignment(ref, index, loc);
iter_init->add_statement(s);
}
*piter_init = iter_init;
// Set *PPOST to
// index_temp++
Block* post = new Block(enclosing, loc);
ref = Expression::make_temporary_reference(index_temp, loc);
s = Statement::make_inc_statement(ref);
post->add_statement(s);
*ppost = post;
}
// Lower a for range over a string.
void
For_range_statement::lower_range_string(Gogo* gogo,
Block* enclosing,
Block* body_block,
Named_object* range_object,
Temporary_statement* range_temp,
Temporary_statement* index_temp,
Temporary_statement* value_temp,
Block** pinit,
Expression** pcond,
Block** piter_init,
Block** ppost)
{
source_location loc = this->location();
// The loop we generate:
// var next_index_temp int
// for index_temp = 0; ; index_temp = next_index_temp {
// next_index_temp, value_temp = stringiter2(range, index_temp)
// if next_index_temp == 0 {
// break
// }
// index = index_temp
// value = value_temp
// original body
// }
// Set *PINIT to
// var next_index_temp int
// index_temp = 0
Block* init = new Block(enclosing, loc);
Temporary_statement* next_index_temp =
Statement::make_temporary(index_temp->type(), NULL, loc);
init->add_statement(next_index_temp);
mpz_t zval;
mpz_init_set_ui(zval, 0UL);
Expression* zexpr = Expression::make_integer(&zval, NULL, loc);
Expression* ref = Expression::make_temporary_reference(index_temp, loc);
Statement* s = Statement::make_assignment(ref, zexpr, loc);
init->add_statement(s);
*pinit = init;
// The loop has no condition.
*pcond = NULL;
// Set *PITER_INIT to
// next_index_temp = runtime.stringiter(range, index_temp)
// or
// next_index_temp, value_temp = runtime.stringiter2(range, index_temp)
// followed by
// if next_index_temp == 0 {
// break
// }
Block* iter_init = new Block(body_block, loc);
Named_object* no;
if (value_temp == NULL)
{
static Named_object* stringiter;
if (stringiter == NULL)
{
source_location bloc = BUILTINS_LOCATION;
Type* int_type = gogo->lookup_global("int")->type_value();
Typed_identifier_list* params = new Typed_identifier_list();
params->push_back(Typed_identifier("s", Type::make_string_type(),
bloc));
params->push_back(Typed_identifier("k", int_type, bloc));
Typed_identifier_list* results = new Typed_identifier_list();
results->push_back(Typed_identifier("", int_type, bloc));
Function_type* fntype = Type::make_function_type(NULL, params,
results, bloc);
stringiter = Named_object::make_function_declaration("stringiter",
NULL, fntype,
bloc);
const char* n = "runtime.stringiter";
stringiter->func_declaration_value()->set_asm_name(n);
}
no = stringiter;
}
else
{
static Named_object* stringiter2;
if (stringiter2 == NULL)
{
source_location bloc = BUILTINS_LOCATION;
Type* int_type = gogo->lookup_global("int")->type_value();
Typed_identifier_list* params = new Typed_identifier_list();
params->push_back(Typed_identifier("s", Type::make_string_type(),
bloc));
params->push_back(Typed_identifier("k", int_type, bloc));
Typed_identifier_list* results = new Typed_identifier_list();
results->push_back(Typed_identifier("", int_type, bloc));
results->push_back(Typed_identifier("", int_type, bloc));
Function_type* fntype = Type::make_function_type(NULL, params,
results, bloc);
stringiter2 = Named_object::make_function_declaration("stringiter",
NULL, fntype,
bloc);
const char* n = "runtime.stringiter2";
stringiter2->func_declaration_value()->set_asm_name(n);
}
no = stringiter2;
}
Expression* func = Expression::make_func_reference(no, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(this->make_range_ref(range_object, range_temp, loc));
params->push_back(Expression::make_temporary_reference(index_temp, loc));
Call_expression* call = Expression::make_call(func, params, false, loc);
if (value_temp == NULL)
{
ref = Expression::make_temporary_reference(next_index_temp, loc);
s = Statement::make_assignment(ref, call, loc);
}
else
{
Expression_list* lhs = new Expression_list();
lhs->push_back(Expression::make_temporary_reference(next_index_temp,
loc));
lhs->push_back(Expression::make_temporary_reference(value_temp, loc));
Expression_list* rhs = new Expression_list();
rhs->push_back(Expression::make_call_result(call, 0));
rhs->push_back(Expression::make_call_result(call, 1));
s = Statement::make_tuple_assignment(lhs, rhs, loc);
}
iter_init->add_statement(s);
ref = Expression::make_temporary_reference(next_index_temp, loc);
zexpr = Expression::make_integer(&zval, NULL, loc);
mpz_clear(zval);
Expression* equals = Expression::make_binary(OPERATOR_EQEQ, ref, zexpr, loc);
Block* then_block = new Block(iter_init, loc);
s = Statement::make_break_statement(this->break_label(), loc);
then_block->add_statement(s);
s = Statement::make_if_statement(equals, then_block, NULL, loc);
iter_init->add_statement(s);
*piter_init = iter_init;
// Set *PPOST to
// index_temp = next_index_temp
Block* post = new Block(enclosing, loc);
Expression* lhs = Expression::make_temporary_reference(index_temp, loc);
Expression* rhs = Expression::make_temporary_reference(next_index_temp, loc);
s = Statement::make_assignment(lhs, rhs, loc);
post->add_statement(s);
*ppost = post;
}
// Lower a for range over a map.
void
For_range_statement::lower_range_map(Gogo* gogo,
Block* enclosing,
Block* body_block,
Named_object* range_object,
Temporary_statement* range_temp,
Temporary_statement* index_temp,
Temporary_statement* value_temp,
Block** pinit,
Expression** pcond,
Block** piter_init,
Block** ppost)
{
source_location loc = this->location();
// The runtime uses a struct to handle ranges over a map. The
// struct is four pointers long. The first pointer is NULL when we
// have completed the iteration.
// The loop we generate:
// var hiter map_iteration_struct
// for mapiterinit(range, &hiter); hiter[0] != nil; mapiternext(&hiter) {
// mapiter2(hiter, &index_temp, &value_temp)
// index = index_temp
// value = value_temp
// original body
// }
// Set *PINIT to
// var hiter map_iteration_struct
// runtime.mapiterinit(range, &hiter)
Block* init = new Block(enclosing, loc);
const unsigned long map_iteration_size = 4;
mpz_t ival;
mpz_init_set_ui(ival, map_iteration_size);
Expression* iexpr = Expression::make_integer(&ival, NULL, loc);
mpz_clear(ival);
Type* byte_type = gogo->lookup_global("byte")->type_value();
Type* ptr_type = Type::make_pointer_type(byte_type);
Type* map_iteration_type = Type::make_array_type(ptr_type, iexpr);
Type* map_iteration_ptr = Type::make_pointer_type(map_iteration_type);
Temporary_statement* hiter = Statement::make_temporary(map_iteration_type,
NULL, loc);
init->add_statement(hiter);
source_location bloc = BUILTINS_LOCATION;
Typed_identifier_list* param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("map", this->range_->type(), bloc));
param_types->push_back(Typed_identifier("it", map_iteration_ptr, bloc));
Function_type* fntype = Type::make_function_type(NULL, param_types, NULL,
bloc);
Named_object* mapiterinit =
Named_object::make_function_declaration("mapiterinit", NULL, fntype, bloc);
const char* n = "runtime.mapiterinit";
mapiterinit->func_declaration_value()->set_asm_name(n);
Expression* func = Expression::make_func_reference(mapiterinit, NULL, loc);
Expression_list* params = new Expression_list();
params->push_back(this->make_range_ref(range_object, range_temp, loc));
Expression* ref = Expression::make_temporary_reference(hiter, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
Expression* call = Expression::make_call(func, params, false, loc);
init->add_statement(Statement::make_statement(call));
*pinit = init;
// Set *PCOND to
// hiter[0] != nil
ref = Expression::make_temporary_reference(hiter, loc);
mpz_t zval;
mpz_init_set_ui(zval, 0UL);
Expression* zexpr = Expression::make_integer(&zval, NULL, loc);
mpz_clear(zval);
Expression* index = Expression::make_index(ref, zexpr, NULL, loc);
Expression* ne = Expression::make_binary(OPERATOR_NOTEQ, index,
Expression::make_nil(loc),
loc);
*pcond = ne;
// Set *PITER_INIT to
// mapiter1(hiter, &index_temp)
// or
// mapiter2(hiter, &index_temp, &value_temp)
Block* iter_init = new Block(body_block, loc);
param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("hiter", map_iteration_ptr, bloc));
Type* pkey_type = Type::make_pointer_type(index_temp->type());
param_types->push_back(Typed_identifier("key", pkey_type, bloc));
if (value_temp != NULL)
{
Type* pval_type = Type::make_pointer_type(value_temp->type());
param_types->push_back(Typed_identifier("val", pval_type, bloc));
}
fntype = Type::make_function_type(NULL, param_types, NULL, bloc);
n = value_temp == NULL ? "mapiter1" : "mapiter2";
Named_object* mapiter = Named_object::make_function_declaration(n, NULL,
fntype, bloc);
n = value_temp == NULL ? "runtime.mapiter1" : "runtime.mapiter2";
mapiter->func_declaration_value()->set_asm_name(n);
func = Expression::make_func_reference(mapiter, NULL, loc);
params = new Expression_list();
ref = Expression::make_temporary_reference(hiter, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
ref = Expression::make_temporary_reference(index_temp, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
if (value_temp != NULL)
{
ref = Expression::make_temporary_reference(value_temp, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
}
call = Expression::make_call(func, params, false, loc);
iter_init->add_statement(Statement::make_statement(call));
*piter_init = iter_init;
// Set *PPOST to
// mapiternext(&hiter)
Block* post = new Block(enclosing, loc);
static Named_object* mapiternext;
if (mapiternext == NULL)
{
param_types = new Typed_identifier_list();
param_types->push_back(Typed_identifier("it", map_iteration_ptr, bloc));
fntype = Type::make_function_type(NULL, param_types, NULL, bloc);
mapiternext = Named_object::make_function_declaration("mapiternext",
NULL, fntype,
bloc);
const char* n = "runtime.mapiternext";
mapiternext->func_declaration_value()->set_asm_name(n);
}
func = Expression::make_func_reference(mapiternext, NULL, loc);
params = new Expression_list();
ref = Expression::make_temporary_reference(hiter, loc);
params->push_back(Expression::make_unary(OPERATOR_AND, ref, loc));
call = Expression::make_call(func, params, false, loc);
post->add_statement(Statement::make_statement(call));
*ppost = post;
}
// Lower a for range over a channel.
void
For_range_statement::lower_range_channel(Gogo* gogo,
Block*,
Block* body_block,
Named_object* range_object,
Temporary_statement* range_temp,
Temporary_statement* index_temp,
Temporary_statement* value_temp,
Block** pinit,
Expression** pcond,
Block** piter_init,
Block** ppost)
{
gcc_assert(value_temp == NULL);
source_location loc = this->location();
// The loop we generate:
// for {
// index_temp = <-range
// if closed(range) {
// break
// }
// index = index_temp
// value = value_temp
// original body
// }
// We have no initialization code, no condition, and no post code.
*pinit = NULL;
*pcond = NULL;
*ppost = NULL;
// Set *PITER_INIT to
// index_temp = <-range
// if closed(range) {
// break
// }
Block* iter_init = new Block(body_block, loc);
Expression* ref = this->make_range_ref(range_object, range_temp, loc);
Expression* cond = this->call_builtin(gogo, "closed", ref, loc);
ref = this->make_range_ref(range_object, range_temp, loc);
Expression* recv = Expression::make_receive(ref, loc);
ref = Expression::make_temporary_reference(index_temp, loc);
Statement* s = Statement::make_assignment(ref, recv, loc);
iter_init->add_statement(s);
Block* then_block = new Block(iter_init, loc);
s = Statement::make_break_statement(this->break_label(), loc);
then_block->add_statement(s);
s = Statement::make_if_statement(cond, then_block, NULL, loc);
iter_init->add_statement(s);
*piter_init = iter_init;
}
// Return the break LABEL_EXPR.
Unnamed_label*
For_range_statement::break_label()
{
if (this->break_label_ == NULL)
this->break_label_ = new Unnamed_label(this->location());
return this->break_label_;
}
// Return the continue LABEL_EXPR.
Unnamed_label*
For_range_statement::continue_label()
{
if (this->continue_label_ == NULL)
this->continue_label_ = new Unnamed_label(this->location());
return this->continue_label_;
}
// Make a for statement with a range clause.
For_range_statement*
Statement::make_for_range_statement(Expression* index_var,
Expression* value_var,
Expression* range,
source_location location)
{
return new For_range_statement(index_var, value_var, range, location);
}