You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
120 lines
4.7 KiB
120 lines
4.7 KiB
(*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*)
|
|
|
|
open! IStd
|
|
module L = Logging
|
|
|
|
(** In this module an ObjC category declaration or implementation is processed. The category is
|
|
saved in the tenv as a struct with the corresponding fields and methods , and the class it
|
|
belongs to *)
|
|
|
|
let cat_class_decl dr =
|
|
match dr.Clang_ast_t.dr_name with Some n -> CAst_utils.get_qualified_name n | _ -> assert false
|
|
|
|
|
|
let get_classname decl_ref_opt =
|
|
match decl_ref_opt with Some dr -> cat_class_decl dr | _ -> assert false
|
|
|
|
|
|
let get_classname_from_category_decl ocdi = get_classname ocdi.Clang_ast_t.odi_class_interface
|
|
|
|
let get_classname_from_category_impl ocidi = get_classname ocidi.Clang_ast_t.ocidi_class_interface
|
|
|
|
let add_category_decl qual_type_to_sil_type tenv category_impl_info =
|
|
let decl_ref_opt = category_impl_info.Clang_ast_t.ocidi_category_decl in
|
|
CAst_utils.add_type_from_decl_ref_opt qual_type_to_sil_type tenv decl_ref_opt true
|
|
|
|
|
|
let add_class_decl qual_type_to_sil_type tenv category_decl_info =
|
|
let decl_ref_opt = category_decl_info.Clang_ast_t.odi_class_interface in
|
|
CAst_utils.add_type_from_decl_ref_opt qual_type_to_sil_type tenv decl_ref_opt true
|
|
|
|
|
|
let add_category_implementation qual_type_to_sil_type tenv category_decl_info =
|
|
let decl_ref_opt = category_decl_info.Clang_ast_t.odi_implementation in
|
|
CAst_utils.add_type_from_decl_ref_opt qual_type_to_sil_type tenv decl_ref_opt false
|
|
|
|
|
|
let get_base_class_name_from_category decl =
|
|
let open Clang_ast_t in
|
|
let base_class_pointer_opt =
|
|
match decl with
|
|
| ObjCCategoryDecl (_, _, _, _, cdi) ->
|
|
cdi.Clang_ast_t.odi_class_interface
|
|
| ObjCCategoryImplDecl (_, _, _, _, cii) ->
|
|
cii.Clang_ast_t.ocidi_class_interface
|
|
| _ ->
|
|
None
|
|
in
|
|
match base_class_pointer_opt with
|
|
| Some decl_ref -> (
|
|
match CAst_utils.get_decl decl_ref.Clang_ast_t.dr_decl_pointer with
|
|
| Some (ObjCInterfaceDecl (_, name_info, _, _, _)) ->
|
|
Some (Typ.Name.Objc.from_qual_name (CAst_utils.get_qualified_name name_info))
|
|
| _ ->
|
|
None )
|
|
| None ->
|
|
None
|
|
|
|
|
|
(* Add potential extra fields and methods defined only in the category *)
|
|
(* to the corresponding class. Update the tenv accordingly.*)
|
|
let process_category qual_type_to_sil_type procname_from_decl tenv class_name decl_info decl_list =
|
|
let class_tn_name = Typ.Name.Objc.from_qual_name class_name in
|
|
let decl_fields = CField_decl.get_fields qual_type_to_sil_type tenv class_tn_name decl_list in
|
|
let decl_methods = ObjcMethod_decl.get_methods procname_from_decl tenv decl_list in
|
|
let class_tn_desc = Typ.Tstruct class_tn_name in
|
|
let decl_key = Clang_ast_extend.DeclPtr decl_info.Clang_ast_t.di_pointer in
|
|
CAst_utils.update_sil_types_map decl_key class_tn_desc ;
|
|
( match Tenv.lookup tenv class_tn_name with
|
|
| Some ({fields; methods} as struct_typ) ->
|
|
let new_fields = CGeneral_utils.append_no_duplicates_fields decl_fields fields in
|
|
let new_methods = CGeneral_utils.append_no_duplicates_methods decl_methods methods in
|
|
ignore
|
|
(Tenv.mk_struct tenv ~default:struct_typ ~fields:new_fields ~methods:new_methods
|
|
class_tn_name) ;
|
|
L.(debug Capture Verbose)
|
|
" Updating info for class '%a' in tenv@\n" QualifiedCppName.pp class_name
|
|
| _ ->
|
|
() ) ;
|
|
class_tn_desc
|
|
|
|
|
|
let category_decl qual_type_to_sil_type procname_from_decl tenv decl =
|
|
let open Clang_ast_t in
|
|
match decl with
|
|
| ObjCCategoryDecl (decl_info, name_info, decl_list, _, cdi) ->
|
|
let name = CAst_utils.get_qualified_name name_info in
|
|
let class_name = get_classname_from_category_decl cdi in
|
|
L.(debug Capture Verbose) "ADDING: ObjCCategoryDecl for '%a'@\n" QualifiedCppName.pp name ;
|
|
add_class_decl qual_type_to_sil_type tenv cdi ;
|
|
let typ =
|
|
process_category qual_type_to_sil_type procname_from_decl tenv class_name decl_info
|
|
decl_list
|
|
in
|
|
add_category_implementation qual_type_to_sil_type tenv cdi ;
|
|
typ
|
|
| _ ->
|
|
assert false
|
|
|
|
|
|
let category_impl_decl qual_type_to_sil_type procname_from_decl tenv decl =
|
|
let open Clang_ast_t in
|
|
match decl with
|
|
| ObjCCategoryImplDecl (decl_info, name_info, decl_list, _, cii) ->
|
|
let name = CAst_utils.get_qualified_name name_info in
|
|
let class_name = get_classname_from_category_impl cii in
|
|
L.(debug Capture Verbose) "ADDING: ObjCCategoryImplDecl for '%a'@\n" QualifiedCppName.pp name ;
|
|
add_category_decl qual_type_to_sil_type tenv cii ;
|
|
let typ =
|
|
process_category qual_type_to_sil_type procname_from_decl tenv class_name decl_info
|
|
decl_list
|
|
in
|
|
typ
|
|
| _ ->
|
|
assert false
|