source: src/Validate/CheckAssertions.cpp

//
// 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.
//
// CheckAssertions.cpp --
//
// Author           : Andrew Beach
// Created On       : Thu Sep  5 14:43:00 2024
// Last Modified By : Andrew Beach
// Last Modified On : Thu Sep  5 14:43:00 2024
// Update Count     : 0
//

#include "CheckAssertions.hpp"

#include "AST/Decl.hpp"
#include "AST/Pass.hpp"
#include "AST/Type.hpp"
#include "Common/SemanticError.hpp"
#include "GenPoly/GenPoly.hpp"

namespace Validate {

namespace {

/// Check for dynamic (sized polymorphic) variadic assertions.
struct VariadicAssertionCore final {
        void checkAssertion( GenPoly::TypeVarMap const & typeVars,
                        ast::FunctionDecl const * assert ) {
                if ( ast::FixedArgs != assert->type->isVarArgs
                                && needsAdapter( assert->type, typeVars ) ) {
                        SemanticError( assert,
                                "Cannot assert a function that both has by-value polymorphic "
                                "parameters or return values, and also takes variadic (...) "
                                "parameters. Consider using a va_list parameter instead.\n" );
                }
        }

        template<typename T>
        void checkList(
                        GenPoly::TypeVarMap const & typeVars,
                        SemanticErrorException & errors,
                        std::vector<ast::ptr<T>> const & members ) {
                for ( const ast::ptr<T> & member : members ) {
                        try {
                                if ( auto func = member.template as<ast::FunctionDecl>() ) {
                                        checkAssertion( typeVars, func );
                                }
                        } catch ( SemanticErrorException & error ) {
                                errors.append( error );
                        }
                }
        }

        void postvisit( ast::FunctionDecl const * decl ) {
                GenPoly::TypeVarMap typeVars;
                SemanticErrorException errors;
                makeTypeVarMap( decl, typeVars );
                checkList( typeVars, errors, decl->assertions );
                errors.throwIfNonEmpty();
        }

        void checkAggr( bool checkMembers,
                        ast::AggregateDecl const * decl ) {
                SemanticErrorException errors;
                GenPoly::TypeVarMap typeVars;
                for ( auto & param : decl->params ) {
                        addToTypeVarMap( param, typeVars );
                        checkList( typeVars, errors, param->assertions );
                }
                if ( checkMembers ) checkList( typeVars, errors, decl->members );
                errors.throwIfNonEmpty();
        }

        void postvisit( ast::StructDecl const * decl ) {
        checkAggr( false, decl );
        }

        void postvisit( ast::UnionDecl const * decl ) {
                checkAggr( false, decl );
        }

        void postvisit( ast::TraitDecl const * decl ) {
                checkAggr( true, decl );
        }
};

} // namespace

void checkAssertions( ast::TranslationUnit & transUnit ) {
        ast::Pass<VariadicAssertionCore>::run( transUnit );
}

} // namespace Validate