1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220
|
// -*- C++ -*-
// ============================================================================
/**
* @file Test_Output.cpp
*
* $Id: Test_Output.cpp 93622 2011-03-22 15:45:57Z johnnyw $
*
* This file factors out common macros and other utilities used by the
* ACE automated regression tests.
*
* @author Prashant Jain <pjain@cs.wustl.edu>
* @author Tim Harrison <harrison@cs.wustl.edu>
* @author David Levine <levine@cs.wustl.edu>
* @author Don Hinton <dhinton@dresystems.com>
*/
// ============================================================================
#include "test_config.h"
#include "ace/OS_NS_stdio.h"
#include "ace/OS_NS_string.h"
#include "ace/OS_NS_sys_stat.h"
#include "ace/Guard_T.h"
#include "ace/Object_Manager.h"
// FUZZ: disable check_for_streams_include
#include "ace/streams.h"
#include "ace/Framework_Component.h"
#include "ace/Log_Msg.h"
#include "ace/ACE.h"
#if defined (VXWORKS)
# include "ace/OS_NS_unistd.h"
# include "ace/OS_NS_fcntl.h"
#endif /* VXWORKS */
ACE_Test_Output *ACE_Test_Output::instance_ = 0;
ACE_Test_Output::ACE_Test_Output (void)
: output_file_ (0)
{
#if !defined (ACE_LACKS_IOSTREAM_TOTALLY)
this->output_file_ = new OFSTREAM;
#endif /* ACE_LACKS_IOSTREAM_TOTALLY */
}
ACE_Test_Output::~ACE_Test_Output (void)
{
#if !defined (ACE_LACKS_IOSTREAM_TOTALLY)
ACE_LOG_MSG->msg_ostream (&cerr, 0);
#endif /* ! ACE_LACKS_IOSTREAM_TOTALLY */
ACE_LOG_MSG->clr_flags (ACE_Log_Msg::OSTREAM);
ACE_LOG_MSG->set_flags (ACE_Log_Msg::STDERR);
#if !defined (ACE_LACKS_IOSTREAM_TOTALLY) && \
(!defined (ACE_HAS_PHARLAP) || defined (ACE_PHARLAP_TESTLOG_TO_FILE))
delete this->output_file_;
#endif /* ! ACE_LACKS_IOSTREAM_TOTALLY */
}
OFSTREAM *
ACE_Test_Output::output_file (void)
{
// the output_file_ is loaned to ACE_LOG_MSG
// and something else might destroy and/or change the stream
// so return what ACE_LOG_MSG is using.
#if defined (ACE_LACKS_IOSTREAM_TOTALLY)
return reinterpret_cast<OFSTREAM*>(ACE_LOG_MSG->msg_ostream ());
#else
return dynamic_cast<OFSTREAM*>(ACE_LOG_MSG->msg_ostream ());
#endif /* ACE_LACKS_IOSTREAM_TOTALLY */
}
int
ACE_Test_Output::set_output (const ACE_TCHAR *filename, int append)
{
#if defined (ACE_HAS_PHARLAP) && !defined (ACE_PHARLAP_TESTLOG_TO_FILE)
// For PharLap, just send it all to the host console for now - redirect
// to a file there for saving/analysis.
EtsSelectConsole(ETS_CO_HOST);
ACE_LOG_MSG->msg_ostream (&cout);
#else
ACE_TCHAR temp[MAXPATHLEN + 1] = { 0 };
// Ignore the error value since the directory may already exist.
const ACE_TCHAR *test_dir = 0;
#if defined (ACE_WIN32) || !defined (ACE_USES_WCHAR)
test_dir = ACE_OS::getenv (ACE_TEXT ("ACE_TEST_DIR"));
#else
ACE_TCHAR tempenv[MAXPATHLEN + 1] = { 0 };
char const * const test_dir_n = ACE_OS::getenv ("ACE_TEST_DIR");
if (test_dir_n == 0)
{
test_dir = 0;
}
else
{
ACE_OS::strncpy (tempenv,
ACE_TEXT_CHAR_TO_TCHAR (test_dir_n),
MAXPATHLEN);
test_dir = tempenv;
}
#endif /* ACE_WIN32 */
if (test_dir == 0)
test_dir = ACE_DEFAULT_TEST_DIR;
// This could be done with ACE_OS::sprintf() but it requires different
// format strings for wide-char POSIX vs. narrow-char POSIX and Windows.
// Easier to keep straight like this.
ACE_OS::strncpy (temp, test_dir, MAXPATHLEN);
ACE_OS::strcat (temp, ACE_LOG_DIRECTORY);
ACE_OS::strcat (temp,
ACE::basename (filename, ACE_DIRECTORY_SEPARATOR_CHAR));
ACE_OS::strcat (temp, ACE_LOG_FILE_EXT_NAME);
#if defined (ACE_VXWORKS)
// This is the only way I could figure out to avoid a console
// warning about opening an existing file (w/o O_CREAT), or
// attempting to unlink a non-existant one.
ACE_HANDLE fd = ACE_OS::open (temp,
O_WRONLY|O_CREAT,
S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH);
if (fd != ERROR)
{
ACE_OS::close (fd);
ACE_OS::unlink (temp);
}
#else /* ! ACE_VXWORKS */
// This doesn't seem to work on VxWorks if the directory doesn't
// exist: it creates a plain file instead of a directory. If the
// directory does exist, it causes a weird console error message
// about "cat: input error on standard input: Is a directory". So,
// VxWorks users must create the directory manually.
ACE_OS::mkdir (ACE_LOG_DIRECTORY_FOR_MKDIR);
#endif /* ACE_VXWORKS */
# if !defined (ACE_LACKS_IOSTREAM_TOTALLY)
this->output_file_->open (ACE_TEXT_ALWAYS_CHAR (temp),
ios::out | (append ? ios::app : ios::trunc));
if (!this->output_file_->good ())
return -1;
#else /* when ACE_LACKS_IOSTREAM_TOTALLY */
ACE_TCHAR *fmode = 0;
if (append)
fmode = ACE_TEXT ("a");
else
fmode = ACE_TEXT ("w");
this->output_file_ = ACE_OS::fopen (temp, fmode);
# endif /* ACE_LACKS_IOSTREAM_TOTALLY */
ACE_LOG_MSG->msg_ostream (this->output_file_, 0);
#endif /* ACE_HAS_PHARLAP && !ACE_PHARLAP_TESTLOG_TO_FILE */
ACE_LOG_MSG->clr_flags (ACE_Log_Msg::STDERR | ACE_Log_Msg::LOGGER );
ACE_LOG_MSG->set_flags (ACE_Log_Msg::OSTREAM);
return 0;
}
void
ACE_Test_Output::close (void)
{
if (this->output_file_ &&
(this->output_file_ == ACE_LOG_MSG->msg_ostream ()))
{
#if !defined (ACE_LACKS_IOSTREAM_TOTALLY)
this->output_file_->flush ();
this->output_file_->close ();
#else
ACE_OS::fflush (this->output_file_);
ACE_OS::fclose (this->output_file_);
#endif /* !ACE_LACKS_IOSTREAM_TOTALLY */
ACE_LOG_MSG->msg_ostream (0, 0);
}
// else something else changed the stream and hence should
// have closed the output_file_
}
ACE_Test_Output *
ACE_Test_Output::instance ()
{
if (ACE_Test_Output::instance_ == 0)
{
// Perform Double-Checked Locking Optimization.
ACE_MT (ACE_GUARD_RETURN (ACE_Recursive_Thread_Mutex, ace_mon,
*ACE_Static_Object_Lock::instance (), 0));
if (ACE_Test_Output::instance_ == 0)
{
ACE_NEW_RETURN (ACE_Test_Output::instance_,
ACE_Test_Output,
0);
ACE_REGISTER_FRAMEWORK_COMPONENT(ACE_Test_Output, ACE_Test_Output::instance_)
}
}
return ACE_Test_Output::instance_;
}
const ACE_TCHAR *
ACE_Test_Output::dll_name (void)
{
return ACE_TEXT ("Test_Output");
}
const ACE_TCHAR *
ACE_Test_Output::name (void)
{
return ACE_TEXT ("ACE_Test_Output");
}
void
ACE_Test_Output::close_singleton (void)
{
delete ACE_Test_Output::instance_;
ACE_Test_Output::instance_ = 0;
}
|