1 /*
2 * Copyright (C) 2020 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <fcntl.h>
18 #include <gtest/gtest.h>
19 #include <sys/stat.h>
20
21 #include "profile-extras.h"
22
23 static int flush_count = 0;
24
25 extern "C" {
__llvm_profile_write_file()26 int __llvm_profile_write_file() {
27 flush_count++;
28 return 0;
29 }
30 }
31
TEST(profile_extras,smoke)32 TEST(profile_extras, smoke) {
33 flush_count = 0;
34
35 ASSERT_EQ(0, flush_count);
36 kill(getpid(), COVERAGE_FLUSH_SIGNAL);
37 sleep(2);
38 ASSERT_EQ(1, flush_count);
39 }
40
41 static const char* OPEN_AT_TEST_FNAME = "/data/misc/trace/test.profraw";
TEST(profile_extras,openat)42 TEST(profile_extras, openat) {
43 mode_t old_umask = umask(0077);
44 unlink(OPEN_AT_TEST_FNAME);
45
46 int fd = open(OPEN_AT_TEST_FNAME, O_RDWR | O_CREAT, 0666);
47 ASSERT_NE(fd, -1);
48 close(fd);
49 umask(old_umask);
50
51 struct stat stat_buf;
52 ASSERT_EQ(stat(OPEN_AT_TEST_FNAME, &stat_buf), 0);
53 ASSERT_EQ(stat_buf.st_mode & 0777, 0666);
54 unlink(OPEN_AT_TEST_FNAME);
55 }
56