Fixes for the ObjC AST. Now it should be complete.

1 files changed, 17 insertions, 0 deletions

diff --git a/src/shared/cplusplus/ASTClone.cpp b/src/shared/cplusplus/ASTClone.cpp
index 1e160ae274..3340a39f25 100644
--- a/src/shared/cplusplus/ASTClone.cpp
+++ b/src/shared/cplusplus/ASTClone.cpp
@@ -1246,6 +1246,7 @@ ObjCCategoryInterfaceDeclarationAST *ObjCCategoryInterfaceDeclarationAST::clone(
     ObjCCategoryInterfaceDeclarationAST *ast = new (pool) ObjCCategoryInterfaceDeclarationAST;
     // copy DeclarationAST
     // copy ObjCCategoryInterfaceDeclarationAST
+    if (attributes) ast->attributes = attributes->clone(pool);
     ast->interface_token = interface_token;
     ast->class_identifier_token = class_identifier_token;
     if (protocol_refs) ast->protocol_refs = protocol_refs->clone(pool);
@@ -1420,6 +1421,22 @@ ObjCMethodPrototypeAST *ObjCMethodPrototypeAST::clone(MemoryPool *pool) const
     return ast;
 }
 
+ObjCMethodDeclarationAST *ObjCMethodDeclarationAST::clone(MemoryPool *pool) const
+{
+    ObjCMethodDeclarationAST *ast = new (pool) ObjCMethodDeclarationAST;
+    if (method_prototype) ast->method_prototype = method_prototype->clone(pool);
+    ast->semicolon_token = semicolon_token;
+    return ast;
+}
+
+ObjCMethodDefinitionAST *ObjCMethodDefinitionAST::clone(MemoryPool *pool) const
+{
+    ObjCMethodDefinitionAST *ast = new (pool) ObjCMethodDefinitionAST;
+    if (method_prototype) ast->method_prototype = method_prototype->clone(pool);
+    if (function_body) ast->function_body = function_body->clone(pool);
+    return ast;
+}
+
 ObjCClassImplementationAST *ObjCClassImplementationAST::clone(MemoryPool *pool) const
 {
     ObjCClassImplementationAST *ast = new (pool) ObjCClassImplementationAST;