source: src/Validate/LabelAddressFixer.cpp @ 89a5a1f

//
// Cforall Version 1.0.0 Copyright (C) 2018 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// LabelAddressFixer.cpp --
//
// Author           : Andrew Beach
// Created On       : Fri Nov 12 16:30:00 2021
// Last Modified By : Andrew Beach
// Last Modified On : Fri Nov 12 16:30:00 2021
// Update Count     : 0
//

#include "Validate/LabelAddressFixer.hpp"

#include "AST/Decl.hpp"
#include "AST/Expr.hpp"
#include "AST/Pass.hpp"
#include "AST/TranslationUnit.hpp"

#include <set>

namespace Validate {

namespace {

struct LabelFinder {
        std::set<ast::Label> & labels;
        LabelFinder( std::set<ast::Label> & labels ) : labels( labels ) {}
        void previsit( const ast::Stmt * stmt ) {
                for ( const ast::Label & label : stmt->labels ) {
                        labels.insert( label );
                }
        }
};

struct LabelAddressFixer : public ast::WithGuards {
        std::set<ast::Label> labels;
        void previsit( const ast::FunctionDecl * decl );
        const ast::Expr * postvisit( const ast::AddressExpr * expr );
};

void LabelAddressFixer::previsit( const ast::FunctionDecl * decl ) {
        GuardValue( labels );
        ast::Pass<LabelFinder>::read( decl, labels );
}

const ast::Expr * LabelAddressFixer::postvisit( const ast::AddressExpr * expr ) {
        if ( auto inner = expr->arg.as<ast::AddressExpr>() ) {
                if ( auto nameExpr = inner->arg.as<ast::NameExpr>() ) {
                        ast::Label label( nameExpr->location, nameExpr->name );
                        if ( labels.count( label ) ) {
                                return new ast::LabelAddressExpr( nameExpr->location, std::move( label ) );
                        }
                }
        }
        return expr;
}

} // namespace

void fixLabelAddresses( ast::TranslationUnit & translationUnit ) {
        ast::Pass<LabelAddressFixer>::run( translationUnit );
}

} // namespace Validate

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