|  | // Copyright 2021 The Tint Authors. | 
|  | // | 
|  | // Licensed under the Apache License, Version 2.0 (the "License"); | 
|  | // you may not use this file except in compliance with the License. | 
|  | // You may obtain a copy of the License at | 
|  | // | 
|  | //     http://www.apache.org/licenses/LICENSE-2.0 | 
|  | // | 
|  | // Unless required by applicable law or agreed to in writing, software | 
|  | // distributed under the License is distributed on an "AS IS" BASIS, | 
|  | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | // See the License for the specific language governing permissions and | 
|  | // limitations under the License. | 
|  |  | 
|  | #include "src/tint/resolver/resolver.h" | 
|  |  | 
|  | #include "gmock/gmock.h" | 
|  | #include "src/tint/ast/call_statement.h" | 
|  | #include "src/tint/resolver/resolver_test_helper.h" | 
|  |  | 
|  | using namespace tint::number_suffixes;  // NOLINT | 
|  |  | 
|  | namespace tint::resolver { | 
|  | namespace { | 
|  |  | 
|  | using ResolverCallValidationTest = ResolverTest; | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, TooFewArgs) { | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param(Sym(), ty.i32()), | 
|  | Param(Sym(), ty.f32()), | 
|  | }, | 
|  | ty.void_(), | 
|  | utils::Vector{ | 
|  | Return(), | 
|  | }); | 
|  | auto* call = Call(Source{{12, 34}}, "foo", 1_i); | 
|  | WrapInFunction(call); | 
|  |  | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), "12:34 error: too few arguments in call to 'foo', expected 2, got 1"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, TooManyArgs) { | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param(Sym(), ty.i32()), | 
|  | Param(Sym(), ty.f32()), | 
|  | }, | 
|  | ty.void_(), | 
|  | utils::Vector{ | 
|  | Return(), | 
|  | }); | 
|  | auto* call = Call(Source{{12, 34}}, "foo", 1_i, 1_f, 1_f); | 
|  | WrapInFunction(call); | 
|  |  | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), "12:34 error: too many arguments in call to 'foo', expected 2, got 3"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, MismatchedArgs) { | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param(Sym(), ty.i32()), | 
|  | Param(Sym(), ty.f32()), | 
|  | }, | 
|  | ty.void_(), | 
|  | utils::Vector{ | 
|  | Return(), | 
|  | }); | 
|  | auto* call = Call("foo", Expr(Source{{12, 34}}, true), 1_f); | 
|  | WrapInFunction(call); | 
|  |  | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), | 
|  | "12:34 error: type mismatch for argument 1 in call to 'foo', " | 
|  | "expected 'i32', got 'bool'"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, UnusedRetval) { | 
|  | // fn func() -> f32 { return 1.0; } | 
|  | // fn main() {func(); return; } | 
|  |  | 
|  | Func("func", utils::Empty, ty.f32(), | 
|  | utils::Vector{ | 
|  | Return(Expr(1_f)), | 
|  | }, | 
|  | utils::Empty); | 
|  |  | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | CallStmt(Source{{12, 34}}, Call("func")), | 
|  | Return(), | 
|  | }); | 
|  |  | 
|  | EXPECT_TRUE(r()->Resolve()) << r()->error(); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, PointerArgument_VariableIdentExpr) { | 
|  | // fn foo(p: ptr<function, i32>) {} | 
|  | // fn main() { | 
|  | //   var z: i32 = 1i; | 
|  | //   foo(&z); | 
|  | // } | 
|  | auto* param = Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)); | 
|  | Func("foo", utils::Vector{param}, ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("z", ty.i32(), Expr(1_i))), | 
|  | CallStmt(Call("foo", AddressOf(Source{{12, 34}}, Expr("z")))), | 
|  | }); | 
|  |  | 
|  | EXPECT_TRUE(r()->Resolve()) << r()->error(); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, PointerArgument_LetIdentExpr) { | 
|  | // fn foo(p: ptr<function, i32>) {} | 
|  | // fn main() { | 
|  | //   let z: i32 = 1i; | 
|  | //   foo(&z); | 
|  | // } | 
|  | auto* param = Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)); | 
|  | Func("foo", utils::Vector{param}, ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Let("z", ty.i32(), Expr(1_i))), | 
|  | CallStmt(Call("foo", AddressOf(Expr(Source{{12, 34}}, "z")))), | 
|  | }); | 
|  |  | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), "12:34 error: cannot take the address of expression"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, PointerArgument_AddressOfFunctionMember) { | 
|  | // struct S { m: i32; }; | 
|  | // fn foo(p: ptr<function, i32>) {} | 
|  | // fn main() { | 
|  | //   var v : S; | 
|  | //   foo(&v.m); | 
|  | // } | 
|  | auto* S = Structure("S", utils::Vector{ | 
|  | Member("m", ty.i32()), | 
|  | }); | 
|  | auto* param = Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)); | 
|  | Func("foo", utils::Vector{param}, ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.Of(S))), | 
|  | CallStmt(Call("foo", AddressOf(Source{{12, 34}}, MemberAccessor("v", "m")))), | 
|  | }); | 
|  |  | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), | 
|  | "12:34 error: arguments of pointer type must not point to a subset of the " | 
|  | "originating variable"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, | 
|  | PointerArgument_AddressOfFunctionMember_WithFullPtrParametersExt) { | 
|  | // enable chromium_experimental_full_ptr_parameters; | 
|  | // struct S { m: i32; }; | 
|  | // fn foo(p: ptr<function, i32>) {} | 
|  | // fn main() { | 
|  | //   var v : S; | 
|  | //   foo(&v.m); | 
|  | // } | 
|  | Enable(ast::Extension::kChromiumExperimentalFullPtrParameters); | 
|  | auto* S = Structure("S", utils::Vector{ | 
|  | Member("m", ty.i32()), | 
|  | }); | 
|  | auto* param = Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)); | 
|  | Func("foo", utils::Vector{param}, ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.Of(S))), | 
|  | CallStmt(Call("foo", AddressOf(Source{{12, 34}}, MemberAccessor("v", "m")))), | 
|  | }); | 
|  |  | 
|  | EXPECT_TRUE(r()->Resolve()) << r()->error(); | 
|  | } | 
|  | TEST_F(ResolverCallValidationTest, PointerArgument_AddressOfLetMember) { | 
|  | // struct S { m: i32; }; | 
|  | // fn foo(p: ptr<function, i32>) {} | 
|  | // fn main() { | 
|  | //   let v: S = S(); | 
|  | //   foo(&v.m); | 
|  | // } | 
|  | auto* S = Structure("S", utils::Vector{ | 
|  | Member("m", ty.i32()), | 
|  | }); | 
|  | auto* param = Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)); | 
|  | Func("foo", utils::Vector{param}, ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Let("v", ty.Of(S), Construct(ty.Of(S)))), | 
|  | CallStmt(Call("foo", AddressOf(MemberAccessor(Source{{12, 34}}, "v", "m")))), | 
|  | }); | 
|  |  | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), "12:34 error: cannot take the address of expression"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, PointerArgument_FunctionParam) { | 
|  | // fn foo(p: ptr<function, i32>) {} | 
|  | // fn bar(p: ptr<function, i32>) { | 
|  | // foo(p); | 
|  | // } | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | Func("bar", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), | 
|  | utils::Vector{ | 
|  | CallStmt(Call("foo", Expr("p"))), | 
|  | }); | 
|  |  | 
|  | EXPECT_TRUE(r()->Resolve()) << r()->error(); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, PointerArgument_FunctionParamWithMain) { | 
|  | // fn foo(p: ptr<function, i32>) {} | 
|  | // fn bar(p: ptr<function, i32>) { | 
|  | // foo(p); | 
|  | // } | 
|  | // @fragment | 
|  | // fn main() { | 
|  | //   var v: i32; | 
|  | //   bar(&v); | 
|  | // } | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | Func("bar", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), | 
|  | utils::Vector{ | 
|  | CallStmt(Call("foo", "p")), | 
|  | }); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.i32(), Expr(1_i))), | 
|  | CallStmt(Call("foo", AddressOf("v"))), | 
|  | }, | 
|  | utils::Vector{ | 
|  | Stage(ast::PipelineStage::kFragment), | 
|  | }); | 
|  |  | 
|  | EXPECT_TRUE(r()->Resolve()) << r()->error(); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, LetPointer) { | 
|  | // fn foo(p : ptr<function, i32>) {} | 
|  | // @fragment | 
|  | // fn main() { | 
|  | //   var v: i32; | 
|  | //   let p: ptr<function, i32> = &v; | 
|  | //   x(p); | 
|  | // } | 
|  | Func("x", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.i32())), | 
|  | Decl(Let("p", ty.pointer(ty.i32(), ast::AddressSpace::kFunction), AddressOf("v"))), | 
|  | CallStmt(Call("x", "p")), | 
|  | }, | 
|  | utils::Vector{ | 
|  | Stage(ast::PipelineStage::kFragment), | 
|  | }); | 
|  | EXPECT_TRUE(r()->Resolve()) << r()->error(); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, LetPointerPrivate) { | 
|  | // fn foo(p : ptr<private, i32>) {} | 
|  | // var v : i32; | 
|  | // @fragment | 
|  | // fn main() { | 
|  | //   let p : ptr<private, i32> = &v; | 
|  | //   foo(p); | 
|  | // } | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kPrivate)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | GlobalVar("v", ty.i32(), ast::AddressSpace::kPrivate); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Let("p", ty.pointer(ty.i32(), ast::AddressSpace::kPrivate), AddressOf("v"))), | 
|  | CallStmt(Call("foo", Expr(Source{{12, 34}}, "p"))), | 
|  | }, | 
|  | utils::Vector{ | 
|  | Stage(ast::PipelineStage::kFragment), | 
|  | }); | 
|  | EXPECT_TRUE(r()->Resolve()) << r()->error(); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, LetPointer_NotWholeVar) { | 
|  | // fn foo(p : ptr<function, i32>) {} | 
|  | // @fragment | 
|  | // fn main() { | 
|  | //   var v: array<i32, 4>; | 
|  | //   let p: ptr<function, i32> = &(v[0]); | 
|  | //   x(p); | 
|  | // } | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.array<i32, 4>())), | 
|  | Decl(Let("p", ty.pointer(ty.i32(), ast::AddressSpace::kFunction), | 
|  | AddressOf(IndexAccessor("v", 0_a)))), | 
|  | CallStmt(Call("foo", Expr(Source{{12, 34}}, "p"))), | 
|  | }, | 
|  | utils::Vector{ | 
|  | Stage(ast::PipelineStage::kFragment), | 
|  | }); | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), | 
|  | "12:34 error: arguments of pointer type must not point to a subset of the " | 
|  | "originating variable"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, LetPointer_NotWholeVar_WithFullPtrParametersExt) { | 
|  | // enable chromium_experimental_full_ptr_parameters; | 
|  | // fn foo(p : ptr<function, i32>) {} | 
|  | // @fragment | 
|  | // fn main() { | 
|  | //   var v: array<i32, 4>; | 
|  | //   let p: ptr<function, i32> = &(v[0]); | 
|  | //   x(p); | 
|  | // } | 
|  | Enable(ast::Extension::kChromiumExperimentalFullPtrParameters); | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.array<i32, 4>())), | 
|  | Decl(Let("p", ty.pointer(ty.i32(), ast::AddressSpace::kFunction), | 
|  | AddressOf(IndexAccessor("v", 0_a)))), | 
|  | CallStmt(Call("foo", Expr(Source{{12, 34}}, "p"))), | 
|  | }, | 
|  | utils::Vector{ | 
|  | Stage(ast::PipelineStage::kFragment), | 
|  | }); | 
|  | EXPECT_TRUE(r()->Resolve()); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, ComplexPointerChain) { | 
|  | // fn foo(p : ptr<function, array<i32, 4>>) {} | 
|  | // @fragment | 
|  | // fn main() { | 
|  | //   var v: array<i32, 4>; | 
|  | //   let p1 = &v; | 
|  | //   let p2 = p1; | 
|  | //   let p3 = &*p2; | 
|  | //   foo(&*p); | 
|  | // } | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer(ty.array<i32, 4>(), ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.array<i32, 4>())), | 
|  | Decl(Let("p1", AddressOf("v"))), | 
|  | Decl(Let("p2", Expr("p1"))), | 
|  | Decl(Let("p3", AddressOf(Deref("p2")))), | 
|  | CallStmt(Call("foo", AddressOf(Source{{12, 34}}, Deref("p3")))), | 
|  | }, | 
|  | utils::Vector{ | 
|  | Stage(ast::PipelineStage::kFragment), | 
|  | }); | 
|  | EXPECT_TRUE(r()->Resolve()) << r()->error(); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, ComplexPointerChain_NotWholeVar) { | 
|  | // fn foo(p : ptr<function, i32>) {} | 
|  | // @fragment | 
|  | // fn main() { | 
|  | //   var v: array<i32, 4>; | 
|  | //   let p1 = &v; | 
|  | //   let p2 = p1; | 
|  | //   let p3 = &(*p2)[0]; | 
|  | //   foo(&*p); | 
|  | // } | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.array<i32, 4>())), | 
|  | Decl(Let("p1", AddressOf("v"))), | 
|  | Decl(Let("p2", Expr("p1"))), | 
|  | Decl(Let("p3", AddressOf(IndexAccessor(Deref("p2"), 0_a)))), | 
|  | CallStmt(Call("foo", AddressOf(Source{{12, 34}}, Deref("p3")))), | 
|  | }, | 
|  | utils::Vector{ | 
|  | Stage(ast::PipelineStage::kFragment), | 
|  | }); | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), | 
|  | "12:34 error: arguments of pointer type must not point to a subset of the " | 
|  | "originating variable"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, ComplexPointerChain_NotWholeVar_WithFullPtrParametersExt) { | 
|  | // enable chromium_experimental_full_ptr_parameters; | 
|  | // fn foo(p : ptr<function, i32>) {} | 
|  | // @fragment | 
|  | // fn main() { | 
|  | //   var v: array<i32, 4>; | 
|  | //   let p1 = &v; | 
|  | //   let p2 = p1; | 
|  | //   let p3 = &(*p2)[0]; | 
|  | //   foo(&*p); | 
|  | // } | 
|  | Enable(ast::Extension::kChromiumExperimentalFullPtrParameters); | 
|  | Func("foo", | 
|  | utils::Vector{ | 
|  | Param("p", ty.pointer<i32>(ast::AddressSpace::kFunction)), | 
|  | }, | 
|  | ty.void_(), utils::Empty); | 
|  | Func("main", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var("v", ty.array<i32, 4>())), | 
|  | Decl(Let("p1", AddressOf("v"))), | 
|  | Decl(Let("p2", Expr("p1"))), | 
|  | Decl(Let("p3", AddressOf(IndexAccessor(Deref("p2"), 0_a)))), | 
|  | CallStmt(Call("foo", AddressOf(Source{{12, 34}}, Deref("p3")))), | 
|  | }, | 
|  | utils::Vector{ | 
|  | Stage(ast::PipelineStage::kFragment), | 
|  | }); | 
|  | EXPECT_TRUE(r()->Resolve()); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, CallVariable) { | 
|  | // var v : i32; | 
|  | // fn f() { | 
|  | //   v(); | 
|  | // } | 
|  | GlobalVar("v", ty.i32(), ast::AddressSpace::kPrivate); | 
|  | Func("f", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | CallStmt(Call(Source{{12, 34}}, "v")), | 
|  | }); | 
|  |  | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), R"(error: cannot call variable 'v' | 
|  | note: 'v' declared here)"); | 
|  | } | 
|  |  | 
|  | TEST_F(ResolverCallValidationTest, CallVariableShadowsFunction) { | 
|  | // fn x() {} | 
|  | // fn f() { | 
|  | //   var x : i32; | 
|  | //   x(); | 
|  | // } | 
|  | Func("x", utils::Empty, ty.void_(), utils::Empty); | 
|  | Func("f", utils::Empty, ty.void_(), | 
|  | utils::Vector{ | 
|  | Decl(Var(Source{{56, 78}}, "x", ty.i32())), | 
|  | CallStmt(Call(Source{{12, 34}}, "x")), | 
|  | }); | 
|  |  | 
|  | EXPECT_FALSE(r()->Resolve()); | 
|  | EXPECT_EQ(r()->error(), R"(error: cannot call variable 'x' | 
|  | 56:78 note: 'x' declared here)"); | 
|  | } | 
|  |  | 
|  | }  // namespace | 
|  | }  // namespace tint::resolver |