Line data    Source code

       1              : // Copyright (C) 2024-2026 Free Software Foundation, Inc.
       2              : 
       3              : // This file is part of GCC.
       4              : 
       5              : // GCC is free software; you can redistribute it and/or modify it under
       6              : // the terms of the GNU General Public License as published by the Free
       7              : // Software Foundation; either version 3, or (at your option) any later
       8              : // version.
       9              : 
      10              : // GCC is distributed in the hope that it will be useful, but WITHOUT ANY
      11              : // WARRANTY; without even the implied warranty of MERCHANTABILITY or
      12              : // FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
      13              : // for more details.
      14              : 
      15              : // You should have received a copy of the GNU General Public License
      16              : // along with GCC; see the file COPYING3.  If not see
      17              : // <http://www.gnu.org/licenses/>.
      18              : 
      19              : #include "rust-collect-lang-items.h"
      20              : #include "optional.h"
      21              : #include "rust-ast-collector.h"
      22              : #include "rust-ast-visitor.h"
      23              : #include "rust-ast.h"
      24              : #include "rust-attribute-values.h"
      25              : #include "rust-attributes.h"
      26              : #include "rust-hir-map.h"
      27              : #include "rust-item.h"
      28              : 
      29              : namespace Rust {
      30              : namespace AST {
      31              : 
      32              : template <typename T>
      33              : tl::optional<LangItem::Kind>
      34        40132 : get_lang_item_attr (const T &maybe_lang_item)
      35              : {
      36       113733 :   for (const auto &attr : maybe_lang_item.get_outer_attrs ())
      37              :     {
      38        73601 :       const auto &str_path = attr.get_path ().as_string ();
      39        73601 :       if (!Analysis::Attributes::is_known (str_path))
      40              :         {
      41            0 :           rust_error_at (attr.get_locus (), "unknown attribute %qs",
      42              :                          str_path.c_str ());
      43              :           continue;
      44              :         }
      45              : 
      46        73601 :       bool is_lang_item = str_path == Values::Attributes::LANG;
      47              : 
      48        73601 :       if (is_lang_item)
      49              :         {
      50         3606 :           auto lang_item_type_str
      51              :             = Analysis::Attributes::extract_string_literal (attr);
      52              : 
      53         3606 :           rust_assert (lang_item_type_str.has_value ());
      54              : 
      55         3606 :           return LangItem::Parse (*lang_item_type_str);
      56         3606 :         }
      57              :     }
      58              : 
      59        36526 :   return tl::nullopt;
      60              : }
      61              : 
      62              : template <typename T>
      63              : void
      64        40132 : CollectLangItems::maybe_add_lang_item (const T &item)
      65              : {
      66        40132 :   if (auto lang_item = get_lang_item_attr (item))
      67         3589 :     mappings.insert_lang_item_node (lang_item.value (), item.get_node_id ());
      68        40132 : }
      69              : 
      70              : void
      71         4163 : CollectLangItems::visit (AST::Trait &item)
      72              : {
      73         4163 :   maybe_add_lang_item (item);
      74              : 
      75         4163 :   DefaultASTVisitor::visit (item);
      76         4163 : }
      77              : 
      78              : void
      79          786 : CollectLangItems::visit (AST::TraitItemType &item)
      80              : {
      81          786 :   maybe_add_lang_item (item);
      82              : 
      83          786 :   DefaultASTVisitor::visit (item);
      84          786 : }
      85              : 
      86              : void
      87        31499 : CollectLangItems::visit (AST::Function &item)
      88              : {
      89        31499 :   maybe_add_lang_item (item);
      90              : 
      91        31499 :   DefaultASTVisitor::visit (item);
      92        31499 : }
      93              : 
      94              : void
      95         2286 : CollectLangItems::visit (AST::StructStruct &item)
      96              : {
      97         2286 :   maybe_add_lang_item (item);
      98              : 
      99         2286 :   DefaultASTVisitor::visit (item);
     100         2286 : }
     101              : 
     102              : void
     103          541 : CollectLangItems::visit (AST::EnumItem &item)
     104              : {
     105          541 :   maybe_add_lang_item (item);
     106              : 
     107          541 :   DefaultASTVisitor::visit (item);
     108          541 : }
     109              : 
     110              : void
     111          475 : CollectLangItems::visit (AST::EnumItemTuple &item)
     112              : {
     113          475 :   maybe_add_lang_item (item);
     114              : 
     115          475 :   DefaultASTVisitor::visit (item);
     116          475 : }
     117              : 
     118              : void
     119           95 : CollectLangItems::visit (AST::EnumItemStruct &item)
     120              : {
     121           95 :   maybe_add_lang_item (item);
     122              : 
     123           95 :   DefaultASTVisitor::visit (item);
     124           95 : }
     125              : 
     126              : void
     127          287 : CollectLangItems::visit (AST::EnumItemDiscriminant &item)
     128              : {
     129          287 :   maybe_add_lang_item (item);
     130              : 
     131          287 :   DefaultASTVisitor::visit (item);
     132          287 : }
     133              : 
     134              : void
     135           28 : CollectLangItems::visit (AST::ExternCrate &extern_crate)
     136              : {
     137           28 :   auto &mappings = Analysis::Mappings::get ();
     138           28 :   auto crate_num
     139           28 :     = mappings.lookup_crate_name (extern_crate.get_referenced_crate ());
     140           28 :   if (crate_num)
     141              :     {
     142           24 :       visit (mappings.get_ast_crate (*crate_num));
     143              :     }
     144              : 
     145           28 :   DefaultASTVisitor::visit (extern_crate);
     146           28 : }
     147              : 
     148              : } // namespace AST
     149              : } // namespace Rust