1/*
2 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24#include "logging/log.hpp"
25#include "logging/logConfiguration.hpp"
26#include "logging/logStream.hpp"
27#include "memory/resourceArea.hpp"
28#include "runtime/os.hpp"
29#include "unittest.hpp"
30
31#define LOG_TEST_STRING_LITERAL "a (hopefully) unique log message for testing"
32
33static inline bool string_contains_substring(const char* haystack, const char* needle) {
34  return strstr(haystack, needle) != NULL;
35}
36
37static inline bool file_exists(const char* filename) {
38  struct stat st;
39  return os::stat(filename, &st) == 0;
40}
41
42static inline void delete_file(const char* filename) {
43  if (!file_exists(filename)) {
44    return;
45  }
46  int ret = remove(filename);
47  EXPECT_TRUE(ret == 0 || errno == ENOENT) << "failed to remove file '" << filename << "': "
48      << os::strerror(errno) << " (" << errno << ")";
49}
50
51static inline void create_directory(const char* name) {
52  assert(!file_exists(name), "can't create directory: %s already exists", name);
53  bool failed;
54#ifdef _WINDOWS
55  failed = !CreateDirectory(name, NULL);
56#else
57  failed = mkdir(name, 0777);
58#endif
59  assert(!failed, "failed to create directory %s", name);
60}
61
62static inline void delete_empty_directory(const char* name) {
63#ifdef _WINDOWS
64  if (!file_exists(name)) {
65    return;
66  }
67  bool failed;
68  failed = !RemoveDirectory(name);
69  EXPECT_FALSE(failed) << "failed to remove directory '" << name
70          << "': LastError = " << GetLastError();
71#else
72  delete_file(name);
73#endif
74}
75
76static inline void init_log_file(const char* filename, const char* options = "") {
77  LogStreamHandle(Error, logging) stream;
78  bool success = LogConfiguration::parse_log_arguments(filename, "logging=trace", "", options, &stream);
79  guarantee(success, "Failed to initialize log file '%s' with options '%s'", filename, options);
80  log_debug(logging)("%s", LOG_TEST_STRING_LITERAL);
81  success = LogConfiguration::parse_log_arguments(filename, "all=off", "", "", &stream);
82  guarantee(success, "Failed to disable logging to file '%s'", filename);
83}
84
85// Read a complete line from fp and return it as a resource allocated string.
86// Returns NULL on EOF.
87static inline char* read_line(FILE* fp) {
88  assert(fp != NULL, "invalid fp");
89  int buflen = 512;
90  char* buf = NEW_RESOURCE_ARRAY(char, buflen);
91  long pos = ftell(fp);
92
93  char* ret = fgets(buf, buflen, fp);
94  while (ret != NULL && buf[strlen(buf) - 1] != '\n' && !feof(fp)) {
95    // retry with a larger buffer
96    buf = REALLOC_RESOURCE_ARRAY(char, buf, buflen, buflen * 2);
97    buflen *= 2;
98    // rewind to beginning of line
99    fseek(fp, pos, SEEK_SET);
100    // retry read with new buffer
101    ret = fgets(buf, buflen, fp);
102  }
103  return ret;
104}
105
106static bool file_contains_substrings_in_order(const char* filename, const char* substrs[]) {
107  FILE* fp = fopen(filename, "r");
108  assert(fp != NULL, "error opening file %s: %s", filename, strerror(errno));
109
110  size_t idx = 0;
111  while (substrs[idx] != NULL) {
112    ResourceMark rm;
113    char* line = read_line(fp);
114    if (line == NULL) {
115      break;
116    }
117    for (char* match = strstr(line, substrs[idx]); match != NULL;) {
118      size_t match_len = strlen(substrs[idx]);
119      idx++;
120      if (substrs[idx] == NULL) {
121        break;
122      }
123      match = strstr(match + match_len, substrs[idx]);
124    }
125  }
126
127  fclose(fp);
128  return substrs[idx] == NULL;
129}
130
131static inline bool file_contains_substring(const char* filename, const char* substr) {
132  const char* strs[] = {substr, NULL};
133  return file_contains_substrings_in_order(filename, strs);
134}
135