ostream.cpp revision 356:1ee8caae33af
1/* 2 * Copyright 1997-2008 Sun Microsystems, Inc. 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, 20 * CA 95054 USA or visit www.sun.com if you need additional information or 21 * have any questions. 22 * 23 */ 24 25# include "incls/_precompiled.incl" 26# include "incls/_ostream.cpp.incl" 27 28extern "C" void jio_print(const char* s); // Declarationtion of jvm method 29 30outputStream::outputStream(int width) { 31 _width = width; 32 _position = 0; 33 _newlines = 0; 34 _precount = 0; 35 _indentation = 0; 36} 37 38outputStream::outputStream(int width, bool has_time_stamps) { 39 _width = width; 40 _position = 0; 41 _newlines = 0; 42 _precount = 0; 43 _indentation = 0; 44 if (has_time_stamps) _stamp.update(); 45} 46 47void outputStream::update_position(const char* s, size_t len) { 48 for (size_t i = 0; i < len; i++) { 49 char ch = s[i]; 50 if (ch == '\n') { 51 _newlines += 1; 52 _precount += _position + 1; 53 _position = 0; 54 } else if (ch == '\t') { 55 int tw = 8 - (_position & 7); 56 _position += tw; 57 _precount -= tw-1; // invariant: _precount + _position == total count 58 } else { 59 _position += 1; 60 } 61 } 62} 63 64// Execute a vsprintf, using the given buffer if necessary. 65// Return a pointer to the formatted string. 66const char* outputStream::do_vsnprintf(char* buffer, size_t buflen, 67 const char* format, va_list ap, 68 bool add_cr, 69 size_t& result_len) { 70 const char* result; 71 if (add_cr) buflen--; 72 if (!strchr(format, '%')) { 73 // constant format string 74 result = format; 75 result_len = strlen(result); 76 if (add_cr && result_len >= buflen) result_len = buflen-1; // truncate 77 } else if (format[0] == '%' && format[1] == 's' && format[2] == '\0') { 78 // trivial copy-through format string 79 result = va_arg(ap, const char*); 80 result_len = strlen(result); 81 if (add_cr && result_len >= buflen) result_len = buflen-1; // truncate 82 } else if (vsnprintf(buffer, buflen, format, ap) >= 0) { 83 result = buffer; 84 result_len = strlen(result); 85 } else { 86 DEBUG_ONLY(warning("increase O_BUFLEN in ostream.hpp -- output truncated");) 87 result = buffer; 88 result_len = buflen - 1; 89 buffer[result_len] = 0; 90 } 91 if (add_cr) { 92 if (result != buffer) { 93 strncpy(buffer, result, buflen); 94 result = buffer; 95 } 96 buffer[result_len++] = '\n'; 97 buffer[result_len] = 0; 98 } 99 return result; 100} 101 102void outputStream::print(const char* format, ...) { 103 char buffer[O_BUFLEN]; 104 va_list ap; 105 va_start(ap, format); 106 size_t len; 107 const char* str = do_vsnprintf(buffer, O_BUFLEN, format, ap, false, len); 108 write(str, len); 109 va_end(ap); 110} 111 112void outputStream::print_cr(const char* format, ...) { 113 char buffer[O_BUFLEN]; 114 va_list ap; 115 va_start(ap, format); 116 size_t len; 117 const char* str = do_vsnprintf(buffer, O_BUFLEN, format, ap, true, len); 118 write(str, len); 119 va_end(ap); 120} 121 122void outputStream::vprint(const char *format, va_list argptr) { 123 char buffer[O_BUFLEN]; 124 size_t len; 125 const char* str = do_vsnprintf(buffer, O_BUFLEN, format, argptr, false, len); 126 write(str, len); 127} 128 129void outputStream::vprint_cr(const char* format, va_list argptr) { 130 char buffer[O_BUFLEN]; 131 size_t len; 132 const char* str = do_vsnprintf(buffer, O_BUFLEN, format, argptr, true, len); 133 write(str, len); 134} 135 136void outputStream::fill_to(int col) { 137 int need_fill = col - position(); 138 sp(need_fill); 139} 140 141void outputStream::move_to(int col, int slop, int min_space) { 142 if (position() >= col + slop) 143 cr(); 144 int need_fill = col - position(); 145 if (need_fill < min_space) 146 need_fill = min_space; 147 sp(need_fill); 148} 149 150void outputStream::put(char ch) { 151 assert(ch != 0, "please fix call site"); 152 char buf[] = { ch, '\0' }; 153 write(buf, 1); 154} 155 156#define SP_USE_TABS false 157 158void outputStream::sp(int count) { 159 if (count < 0) return; 160 if (SP_USE_TABS && count >= 8) { 161 int target = position() + count; 162 while (count >= 8) { 163 this->write("\t", 1); 164 count -= 8; 165 } 166 count = target - position(); 167 } 168 while (count > 0) { 169 int nw = (count > 8) ? 8 : count; 170 this->write(" ", nw); 171 count -= nw; 172 } 173} 174 175void outputStream::cr() { 176 this->write("\n", 1); 177} 178 179void outputStream::stamp() { 180 if (! _stamp.is_updated()) { 181 _stamp.update(); // start at 0 on first call to stamp() 182 } 183 184 // outputStream::stamp() may get called by ostream_abort(), use snprintf 185 // to avoid allocating large stack buffer in print(). 186 char buf[40]; 187 jio_snprintf(buf, sizeof(buf), "%.3f", _stamp.seconds()); 188 print_raw(buf); 189} 190 191void outputStream::stamp(bool guard, 192 const char* prefix, 193 const char* suffix) { 194 if (!guard) { 195 return; 196 } 197 print_raw(prefix); 198 stamp(); 199 print_raw(suffix); 200} 201 202void outputStream::date_stamp(bool guard, 203 const char* prefix, 204 const char* suffix) { 205 if (!guard) { 206 return; 207 } 208 print_raw(prefix); 209 static const char error_time[] = "yyyy-mm-ddThh:mm:ss.mmm+zzzz"; 210 static const int buffer_length = 32; 211 char buffer[buffer_length]; 212 const char* iso8601_result = os::iso8601_time(buffer, buffer_length); 213 if (iso8601_result != NULL) { 214 print_raw(buffer); 215 } else { 216 print_raw(error_time); 217 } 218 print_raw(suffix); 219 return; 220} 221 222void outputStream::indent() { 223 while (_position < _indentation) sp(); 224} 225 226void outputStream::print_jlong(jlong value) { 227 // N.B. Same as INT64_FORMAT 228 print(os::jlong_format_specifier(), value); 229} 230 231void outputStream::print_julong(julong value) { 232 // N.B. Same as UINT64_FORMAT 233 print(os::julong_format_specifier(), value); 234} 235 236stringStream::stringStream(size_t initial_size) : outputStream() { 237 buffer_length = initial_size; 238 buffer = NEW_RESOURCE_ARRAY(char, buffer_length); 239 buffer_pos = 0; 240 buffer_fixed = false; 241} 242 243// useful for output to fixed chunks of memory, such as performance counters 244stringStream::stringStream(char* fixed_buffer, size_t fixed_buffer_size) : outputStream() { 245 buffer_length = fixed_buffer_size; 246 buffer = fixed_buffer; 247 buffer_pos = 0; 248 buffer_fixed = true; 249} 250 251void stringStream::write(const char* s, size_t len) { 252 size_t write_len = len; // number of non-null bytes to write 253 size_t end = buffer_pos + len + 1; // position after write and final '\0' 254 if (end > buffer_length) { 255 if (buffer_fixed) { 256 // if buffer cannot resize, silently truncate 257 end = buffer_length; 258 write_len = end - buffer_pos - 1; // leave room for the final '\0' 259 } else { 260 // For small overruns, double the buffer. For larger ones, 261 // increase to the requested size. 262 if (end < buffer_length * 2) { 263 end = buffer_length * 2; 264 } 265 char* oldbuf = buffer; 266 buffer = NEW_RESOURCE_ARRAY(char, end); 267 strncpy(buffer, oldbuf, buffer_pos); 268 buffer_length = end; 269 } 270 } 271 // invariant: buffer is always null-terminated 272 guarantee(buffer_pos + write_len + 1 <= buffer_length, "stringStream oob"); 273 buffer[buffer_pos + write_len] = 0; 274 strncpy(buffer + buffer_pos, s, write_len); 275 buffer_pos += write_len; 276 277 // Note that the following does not depend on write_len. 278 // This means that position and count get updated 279 // even when overflow occurs. 280 update_position(s, len); 281} 282 283char* stringStream::as_string() { 284 char* copy = NEW_RESOURCE_ARRAY(char, buffer_pos+1); 285 strncpy(copy, buffer, buffer_pos); 286 copy[buffer_pos] = 0; // terminating null 287 return copy; 288} 289 290stringStream::~stringStream() {} 291 292xmlStream* xtty; 293outputStream* tty; 294outputStream* gclog_or_tty; 295extern Mutex* tty_lock; 296 297fileStream::fileStream(const char* file_name) { 298 _file = fopen(file_name, "w"); 299 _need_close = true; 300} 301 302void fileStream::write(const char* s, size_t len) { 303 if (_file != NULL) fwrite(s, 1, len, _file); 304 update_position(s, len); 305} 306 307fileStream::~fileStream() { 308 if (_file != NULL) { 309 if (_need_close) fclose(_file); 310 _file = NULL; 311 } 312} 313 314void fileStream::flush() { 315 fflush(_file); 316} 317 318fdStream::fdStream(const char* file_name) { 319 _fd = open(file_name, O_WRONLY | O_CREAT | O_TRUNC, 0666); 320 _need_close = true; 321} 322 323fdStream::~fdStream() { 324 if (_fd != -1) { 325 if (_need_close) close(_fd); 326 _fd = -1; 327 } 328} 329 330void fdStream::write(const char* s, size_t len) { 331 if (_fd != -1) ::write(_fd, s, (int)len); 332 update_position(s, len); 333} 334 335defaultStream* defaultStream::instance = NULL; 336int defaultStream::_output_fd = 1; 337int defaultStream::_error_fd = 2; 338FILE* defaultStream::_output_stream = stdout; 339FILE* defaultStream::_error_stream = stderr; 340 341#define LOG_MAJOR_VERSION 160 342#define LOG_MINOR_VERSION 1 343 344void defaultStream::init() { 345 _inited = true; 346 if (LogVMOutput || LogCompilation) { 347 init_log(); 348 } 349} 350 351bool defaultStream::has_log_file() { 352 // lazily create log file (at startup, LogVMOutput is false even 353 // if +LogVMOutput is used, because the flags haven't been parsed yet) 354 // For safer printing during fatal error handling, do not init logfile 355 // if a VM error has been reported. 356 if (!_inited && !is_error_reported()) init(); 357 return _log_file != NULL; 358} 359 360static const char* make_log_name(const char* log_name, const char* force_directory, char* buf) { 361 const char* basename = log_name; 362 char file_sep = os::file_separator()[0]; 363 const char* cp; 364 for (cp = log_name; *cp != '\0'; cp++) { 365 if (*cp == '/' || *cp == file_sep) { 366 basename = cp+1; 367 } 368 } 369 const char* nametail = log_name; 370 371 strcpy(buf, ""); 372 if (force_directory != NULL) { 373 strcat(buf, force_directory); 374 strcat(buf, os::file_separator()); 375 nametail = basename; // completely skip directory prefix 376 } 377 378 const char* star = strchr(basename, '*'); 379 int star_pos = (star == NULL) ? -1 : (star - nametail); 380 381 if (star_pos >= 0) { 382 // convert foo*bar.log to foo123bar.log 383 int buf_pos = (int) strlen(buf); 384 strncpy(&buf[buf_pos], nametail, star_pos); 385 sprintf(&buf[buf_pos + star_pos], "%u", os::current_process_id()); 386 nametail += star_pos + 1; // skip prefix and star 387 } 388 389 strcat(buf, nametail); // append rest of name, or all of name 390 return buf; 391} 392 393void defaultStream::init_log() { 394 // %%% Need a MutexLocker? 395 const char* log_name = LogFile != NULL ? LogFile : "hotspot.log"; 396 char buf[O_BUFLEN*2]; 397 const char* try_name = make_log_name(log_name, NULL, buf); 398 fileStream* file = new(ResourceObj::C_HEAP) fileStream(try_name); 399 if (!file->is_open()) { 400 // Try again to open the file. 401 char warnbuf[O_BUFLEN*2]; 402 sprintf(warnbuf, "Warning: Cannot open log file: %s\n", try_name); 403 // Note: This feature is for maintainer use only. No need for L10N. 404 jio_print(warnbuf); 405 try_name = make_log_name("hs_pid*.log", os::get_temp_directory(), buf); 406 sprintf(warnbuf, "Warning: Forcing option -XX:LogFile=%s\n", try_name); 407 jio_print(warnbuf); 408 delete file; 409 file = new(ResourceObj::C_HEAP) fileStream(try_name); 410 } 411 if (file->is_open()) { 412 _log_file = file; 413 xmlStream* xs = new(ResourceObj::C_HEAP) xmlStream(file); 414 _outer_xmlStream = xs; 415 if (this == tty) xtty = xs; 416 // Write XML header. 417 xs->print_cr("<?xml version='1.0' encoding='UTF-8'?>"); 418 // (For now, don't bother to issue a DTD for this private format.) 419 jlong time_ms = os::javaTimeMillis() - tty->time_stamp().milliseconds(); 420 // %%% Should be: jlong time_ms = os::start_time_milliseconds(), if 421 // we ever get round to introduce that method on the os class 422 xs->head("hotspot_log version='%d %d'" 423 " process='%d' time_ms='"INT64_FORMAT"'", 424 LOG_MAJOR_VERSION, LOG_MINOR_VERSION, 425 os::current_process_id(), time_ms); 426 // Write VM version header immediately. 427 xs->head("vm_version"); 428 xs->head("name"); xs->text("%s", VM_Version::vm_name()); xs->cr(); 429 xs->tail("name"); 430 xs->head("release"); xs->text("%s", VM_Version::vm_release()); xs->cr(); 431 xs->tail("release"); 432 xs->head("info"); xs->text("%s", VM_Version::internal_vm_info_string()); xs->cr(); 433 xs->tail("info"); 434 xs->tail("vm_version"); 435 // Record information about the command-line invocation. 436 xs->head("vm_arguments"); // Cf. Arguments::print_on() 437 if (Arguments::num_jvm_flags() > 0) { 438 xs->head("flags"); 439 Arguments::print_jvm_flags_on(xs->text()); 440 xs->tail("flags"); 441 } 442 if (Arguments::num_jvm_args() > 0) { 443 xs->head("args"); 444 Arguments::print_jvm_args_on(xs->text()); 445 xs->tail("args"); 446 } 447 if (Arguments::java_command() != NULL) { 448 xs->head("command"); xs->text()->print_cr("%s", Arguments::java_command()); 449 xs->tail("command"); 450 } 451 if (Arguments::sun_java_launcher() != NULL) { 452 xs->head("launcher"); xs->text()->print_cr("%s", Arguments::sun_java_launcher()); 453 xs->tail("launcher"); 454 } 455 if (Arguments::system_properties() != NULL) { 456 xs->head("properties"); 457 // Print it as a java-style property list. 458 // System properties don't generally contain newlines, so don't bother with unparsing. 459 for (SystemProperty* p = Arguments::system_properties(); p != NULL; p = p->next()) { 460 xs->text()->print_cr("%s=%s", p->key(), p->value()); 461 } 462 xs->tail("properties"); 463 } 464 xs->tail("vm_arguments"); 465 // tty output per se is grouped under the <tty>...</tty> element. 466 xs->head("tty"); 467 // All further non-markup text gets copied to the tty: 468 xs->_text = this; // requires friend declaration! 469 } else { 470 delete(file); 471 // and leave xtty as NULL 472 LogVMOutput = false; 473 DisplayVMOutput = true; 474 LogCompilation = false; 475 } 476} 477 478// finish_log() is called during normal VM shutdown. finish_log_on_error() is 479// called by ostream_abort() after a fatal error. 480// 481void defaultStream::finish_log() { 482 xmlStream* xs = _outer_xmlStream; 483 xs->done("tty"); 484 485 // Other log forks are appended here, at the End of Time: 486 CompileLog::finish_log(xs->out()); // write compile logging, if any, now 487 488 xs->done("hotspot_log"); 489 xs->flush(); 490 491 fileStream* file = _log_file; 492 _log_file = NULL; 493 494 delete _outer_xmlStream; 495 _outer_xmlStream = NULL; 496 497 file->flush(); 498 delete file; 499} 500 501void defaultStream::finish_log_on_error(char *buf, int buflen) { 502 xmlStream* xs = _outer_xmlStream; 503 504 if (xs && xs->out()) { 505 506 xs->done_raw("tty"); 507 508 // Other log forks are appended here, at the End of Time: 509 CompileLog::finish_log_on_error(xs->out(), buf, buflen); // write compile logging, if any, now 510 511 xs->done_raw("hotspot_log"); 512 xs->flush(); 513 514 fileStream* file = _log_file; 515 _log_file = NULL; 516 _outer_xmlStream = NULL; 517 518 if (file) { 519 file->flush(); 520 521 // Can't delete or close the file because delete and fclose aren't 522 // async-safe. We are about to die, so leave it to the kernel. 523 // delete file; 524 } 525 } 526} 527 528intx defaultStream::hold(intx writer_id) { 529 bool has_log = has_log_file(); // check before locking 530 if (// impossible, but who knows? 531 writer_id == NO_WRITER || 532 533 // bootstrap problem 534 tty_lock == NULL || 535 536 // can't grab a lock or call Thread::current() if TLS isn't initialized 537 ThreadLocalStorage::thread() == NULL || 538 539 // developer hook 540 !SerializeVMOutput || 541 542 // VM already unhealthy 543 is_error_reported() || 544 545 // safepoint == global lock (for VM only) 546 (SafepointSynchronize::is_synchronizing() && 547 Thread::current()->is_VM_thread()) 548 ) { 549 // do not attempt to lock unless we know the thread and the VM is healthy 550 return NO_WRITER; 551 } 552 if (_writer == writer_id) { 553 // already held, no need to re-grab the lock 554 return NO_WRITER; 555 } 556 tty_lock->lock_without_safepoint_check(); 557 // got the lock 558 if (writer_id != _last_writer) { 559 if (has_log) { 560 _log_file->bol(); 561 // output a hint where this output is coming from: 562 _log_file->print_cr("<writer thread='"INTX_FORMAT"'/>", writer_id); 563 } 564 _last_writer = writer_id; 565 } 566 _writer = writer_id; 567 return writer_id; 568} 569 570void defaultStream::release(intx holder) { 571 if (holder == NO_WRITER) { 572 // nothing to release: either a recursive lock, or we scribbled (too bad) 573 return; 574 } 575 if (_writer != holder) { 576 return; // already unlocked, perhaps via break_tty_lock_for_safepoint 577 } 578 _writer = NO_WRITER; 579 tty_lock->unlock(); 580} 581 582 583// Yuck: jio_print does not accept char*/len. 584static void call_jio_print(const char* s, size_t len) { 585 char buffer[O_BUFLEN+100]; 586 if (len > sizeof(buffer)-1) { 587 warning("increase O_BUFLEN in ostream.cpp -- output truncated"); 588 len = sizeof(buffer)-1; 589 } 590 strncpy(buffer, s, len); 591 buffer[len] = '\0'; 592 jio_print(buffer); 593} 594 595 596void defaultStream::write(const char* s, size_t len) { 597 intx thread_id = os::current_thread_id(); 598 intx holder = hold(thread_id); 599 600 if (DisplayVMOutput && 601 (_outer_xmlStream == NULL || !_outer_xmlStream->inside_attrs())) { 602 // print to output stream. It can be redirected by a vfprintf hook 603 if (s[len] == '\0') { 604 jio_print(s); 605 } else { 606 call_jio_print(s, len); 607 } 608 } 609 610 // print to log file 611 if (has_log_file()) { 612 int nl0 = _newlines; 613 xmlTextStream::write(s, len); 614 // flush the log file too, if there were any newlines 615 if (nl0 != _newlines){ 616 flush(); 617 } 618 } else { 619 update_position(s, len); 620 } 621 622 release(holder); 623} 624 625intx ttyLocker::hold_tty() { 626 if (defaultStream::instance == NULL) return defaultStream::NO_WRITER; 627 intx thread_id = os::current_thread_id(); 628 return defaultStream::instance->hold(thread_id); 629} 630 631void ttyLocker::release_tty(intx holder) { 632 if (holder == defaultStream::NO_WRITER) return; 633 defaultStream::instance->release(holder); 634} 635 636void ttyLocker::break_tty_lock_for_safepoint(intx holder) { 637 if (defaultStream::instance != NULL && 638 defaultStream::instance->writer() == holder) { 639 if (xtty != NULL) { 640 xtty->print_cr("<!-- safepoint while printing -->"); 641 } 642 defaultStream::instance->release(holder); 643 } 644 // (else there was no lock to break) 645} 646 647void ostream_init() { 648 if (defaultStream::instance == NULL) { 649 defaultStream::instance = new(ResourceObj::C_HEAP) defaultStream(); 650 tty = defaultStream::instance; 651 652 // We want to ensure that time stamps in GC logs consider time 0 653 // the time when the JVM is initialized, not the first time we ask 654 // for a time stamp. So, here, we explicitly update the time stamp 655 // of tty. 656 tty->time_stamp().update_to(1); 657 } 658} 659 660void ostream_init_log() { 661 // For -Xloggc:<file> option - called in runtime/thread.cpp 662 // Note : this must be called AFTER ostream_init() 663 664 gclog_or_tty = tty; // default to tty 665 if (Arguments::gc_log_filename() != NULL) { 666 fileStream * gclog = new(ResourceObj::C_HEAP) 667 fileStream(Arguments::gc_log_filename()); 668 if (gclog->is_open()) { 669 // now we update the time stamp of the GC log to be synced up 670 // with tty. 671 gclog->time_stamp().update_to(tty->time_stamp().ticks()); 672 gclog_or_tty = gclog; 673 } 674 } 675 676 // If we haven't lazily initialized the logfile yet, do it now, 677 // to avoid the possibility of lazy initialization during a VM 678 // crash, which can affect the stability of the fatal error handler. 679 defaultStream::instance->has_log_file(); 680} 681 682// ostream_exit() is called during normal VM exit to finish log files, flush 683// output and free resource. 684void ostream_exit() { 685 static bool ostream_exit_called = false; 686 if (ostream_exit_called) return; 687 ostream_exit_called = true; 688 if (gclog_or_tty != tty) { 689 delete gclog_or_tty; 690 } 691 { 692 // we temporaly disable PrintMallocFree here 693 // as otherwise it'll lead to using of almost deleted 694 // tty or defaultStream::instance in logging facility 695 // of HeapFree(), see 6391258 696 DEBUG_ONLY(FlagSetting fs(PrintMallocFree, false);) 697 if (tty != defaultStream::instance) { 698 delete tty; 699 } 700 if (defaultStream::instance != NULL) { 701 delete defaultStream::instance; 702 } 703 } 704 tty = NULL; 705 xtty = NULL; 706 gclog_or_tty = NULL; 707 defaultStream::instance = NULL; 708} 709 710// ostream_abort() is called by os::abort() when VM is about to die. 711void ostream_abort() { 712 // Here we can't delete gclog_or_tty and tty, just flush their output 713 if (gclog_or_tty) gclog_or_tty->flush(); 714 if (tty) tty->flush(); 715 716 if (defaultStream::instance != NULL) { 717 static char buf[4096]; 718 defaultStream::instance->finish_log_on_error(buf, sizeof(buf)); 719 } 720} 721 722staticBufferStream::staticBufferStream(char* buffer, size_t buflen, 723 outputStream *outer_stream) { 724 _buffer = buffer; 725 _buflen = buflen; 726 _outer_stream = outer_stream; 727} 728 729void staticBufferStream::write(const char* c, size_t len) { 730 _outer_stream->print_raw(c, (int)len); 731} 732 733void staticBufferStream::flush() { 734 _outer_stream->flush(); 735} 736 737void staticBufferStream::print(const char* format, ...) { 738 va_list ap; 739 va_start(ap, format); 740 size_t len; 741 const char* str = do_vsnprintf(_buffer, _buflen, format, ap, false, len); 742 write(str, len); 743 va_end(ap); 744} 745 746void staticBufferStream::print_cr(const char* format, ...) { 747 va_list ap; 748 va_start(ap, format); 749 size_t len; 750 const char* str = do_vsnprintf(_buffer, _buflen, format, ap, true, len); 751 write(str, len); 752 va_end(ap); 753} 754 755void staticBufferStream::vprint(const char *format, va_list argptr) { 756 size_t len; 757 const char* str = do_vsnprintf(_buffer, _buflen, format, argptr, false, len); 758 write(str, len); 759} 760 761void staticBufferStream::vprint_cr(const char* format, va_list argptr) { 762 size_t len; 763 const char* str = do_vsnprintf(_buffer, _buflen, format, argptr, true, len); 764 write(str, len); 765} 766 767bufferedStream::bufferedStream(size_t initial_size, size_t bufmax) : outputStream() { 768 buffer_length = initial_size; 769 buffer = NEW_C_HEAP_ARRAY(char, buffer_length); 770 buffer_pos = 0; 771 buffer_fixed = false; 772 buffer_max = bufmax; 773} 774 775bufferedStream::bufferedStream(char* fixed_buffer, size_t fixed_buffer_size, size_t bufmax) : outputStream() { 776 buffer_length = fixed_buffer_size; 777 buffer = fixed_buffer; 778 buffer_pos = 0; 779 buffer_fixed = true; 780 buffer_max = bufmax; 781} 782 783void bufferedStream::write(const char* s, size_t len) { 784 785 if(buffer_pos + len > buffer_max) { 786 flush(); 787 } 788 789 size_t end = buffer_pos + len; 790 if (end >= buffer_length) { 791 if (buffer_fixed) { 792 // if buffer cannot resize, silently truncate 793 len = buffer_length - buffer_pos - 1; 794 } else { 795 // For small overruns, double the buffer. For larger ones, 796 // increase to the requested size. 797 if (end < buffer_length * 2) { 798 end = buffer_length * 2; 799 } 800 buffer = REALLOC_C_HEAP_ARRAY(char, buffer, end); 801 buffer_length = end; 802 } 803 } 804 memcpy(buffer + buffer_pos, s, len); 805 buffer_pos += len; 806 update_position(s, len); 807} 808 809char* bufferedStream::as_string() { 810 char* copy = NEW_RESOURCE_ARRAY(char, buffer_pos+1); 811 strncpy(copy, buffer, buffer_pos); 812 copy[buffer_pos] = 0; // terminating null 813 return copy; 814} 815 816bufferedStream::~bufferedStream() { 817 if (!buffer_fixed) { 818 FREE_C_HEAP_ARRAY(char, buffer); 819 } 820} 821 822#ifndef PRODUCT 823 824#if defined(SOLARIS) || defined(LINUX) 825#include <sys/types.h> 826#include <sys/socket.h> 827#include <netinet/in.h> 828#include <arpa/inet.h> 829#endif 830 831// Network access 832networkStream::networkStream() : bufferedStream(1024*10, 1024*10) { 833 834 _socket = -1; 835 836 hpi::initialize_socket_library(); 837 838 int result = hpi::socket(AF_INET, SOCK_STREAM, 0); 839 if (result <= 0) { 840 assert(false, "Socket could not be created!"); 841 } else { 842 _socket = result; 843 } 844} 845 846int networkStream::read(char *buf, size_t len) { 847 return hpi::recv(_socket, buf, (int)len, 0); 848} 849 850void networkStream::flush() { 851 if (size() != 0) { 852 int result = hpi::raw_send(_socket, (char *)base(), (int)size(), 0); 853 assert(result != -1, "connection error"); 854 assert(result == (int)size(), "didn't send enough data"); 855 } 856 reset(); 857} 858 859networkStream::~networkStream() { 860 close(); 861} 862 863void networkStream::close() { 864 if (_socket != -1) { 865 flush(); 866 hpi::socket_close(_socket); 867 _socket = -1; 868 } 869} 870 871bool networkStream::connect(const char *ip, short port) { 872 873 struct sockaddr_in server; 874 server.sin_family = AF_INET; 875 server.sin_port = htons(port); 876 877 server.sin_addr.s_addr = inet_addr(ip); 878 if (server.sin_addr.s_addr == (uint32_t)-1) { 879#ifdef _WINDOWS 880 struct hostent* host = hpi::get_host_by_name((char*)ip); 881#else 882 struct hostent* host = gethostbyname(ip); 883#endif 884 if (host != NULL) { 885 memcpy(&server.sin_addr, host->h_addr_list[0], host->h_length); 886 } else { 887 return false; 888 } 889 } 890 891 892 int result = hpi::connect(_socket, (struct sockaddr*)&server, sizeof(struct sockaddr_in)); 893 return (result >= 0); 894} 895 896#endif 897