source: src/ControlStruct/FixLabels.cpp

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// FixLabels.cpp -- Normalizes labels and handles multi-level exit labels.
//
// Author           : Andrew Beach
// Created On       : Mon Nov  1 09:39:00 2021
// Last Modified By : Peter A. Buhr
// Last Modified On : Sun Nov 26 15:06:51 2023
// Update Count     : 10
//

#include "FixLabels.hpp"

#include "AST/Label.hpp"
#include "AST/Pass.hpp"
#include "AST/Stmt.hpp"
#include "ControlStruct/MultiLevelExit.hpp"
using namespace ast;

namespace ControlStruct {
class FixLabelsCore final : public WithGuards {
        LabelToStmt labelTable;
  public:
        FixLabelsCore() : labelTable() {}

        void previsit( const FunctionDecl * );
        const FunctionDecl * postvisit( const FunctionDecl * );
        void previsit( const Stmt * );
        void previsit( const BranchStmt * );
        void previsit( const LabelAddressExpr * );

        void setLabelsDef( const std::vector<Label> &, const Stmt * );
        void setLabelsUsage( const Label & );
};

void FixLabelsCore::previsit( const FunctionDecl * ) {
        GuardValue( labelTable ).clear();
}

const FunctionDecl * FixLabelsCore::postvisit(
        const FunctionDecl * decl ) {
        if ( nullptr == decl->stmts ) return decl;
        for ( auto kvp : labelTable ) {
                if ( nullptr == kvp.second ) {
                        SemanticError( kvp.first.location, "Use of undefined label %s.", kvp.first.name.c_str() );
                }
        }
        return mutate_field( decl, &FunctionDecl::stmts,
                                                 multiLevelExitUpdate( decl->stmts.get(), labelTable ) );
}

void FixLabelsCore::previsit( const Stmt * stmt ) {
        if ( !stmt->labels.empty() ) {
                setLabelsDef( stmt->labels, stmt );
        }
}

void FixLabelsCore::previsit( const BranchStmt * stmt ) {
        if ( !stmt->labels.empty() ) {
                setLabelsDef( stmt->labels, stmt );
        }
        if ( !stmt->target.empty() ) {
                setLabelsUsage( stmt->target );
        }
}

void FixLabelsCore::previsit( const LabelAddressExpr * expr ) {
        assert( !expr->arg.empty() );
        setLabelsUsage( expr->arg );
}

void FixLabelsCore::setLabelsDef(
        const std::vector<Label> & labels, const Stmt * stmt ) {
        assert( !labels.empty() );
        assert( stmt );

        for ( auto label : labels ) {
                if ( labelTable.find( label ) == labelTable.end() ) {
                        // Make sure to only create the entry once.
                        labelTable[ label ] = stmt;
                } else if ( nullptr != labelTable[ label ] ) {
                        // Duplicate definition, this is an error.
                        SemanticError( label.location,
                                                   "Duplicate definition of label: " + label.name );
                } else {
                        // Perviously used, but not defined until now.
                        labelTable[ label ] = stmt;
                }
        }
}

// Label was used, if it is new add it to the table.
void FixLabelsCore::setLabelsUsage( const Label & label ) {
        if ( labelTable.find( label ) == labelTable.end() ) {
                labelTable[ label ] = nullptr;
        }
}

void fixLabels( TranslationUnit & translationUnit ) {
        Pass<FixLabelsCore>::run( translationUnit );
}
} // namespace ControlStruct

// Local Variables: //
// tab-width: 4 //
// mode: c++ //
// compile-command: "make install" //
// End: //