14f94c520f8d699a5973956a1716272146be17128Zonr Chang/* 24f94c520f8d699a5973956a1716272146be17128Zonr Chang * Copyright 2012, The Android Open Source Project 34f94c520f8d699a5973956a1716272146be17128Zonr Chang * 44f94c520f8d699a5973956a1716272146be17128Zonr Chang * Licensed under the Apache License, Version 2.0 (the "License"); 54f94c520f8d699a5973956a1716272146be17128Zonr Chang * you may not use this file except in compliance with the License. 64f94c520f8d699a5973956a1716272146be17128Zonr Chang * You may obtain a copy of the License at 74f94c520f8d699a5973956a1716272146be17128Zonr Chang * 84f94c520f8d699a5973956a1716272146be17128Zonr Chang * http://www.apache.org/licenses/LICENSE-2.0 94f94c520f8d699a5973956a1716272146be17128Zonr Chang * 104f94c520f8d699a5973956a1716272146be17128Zonr Chang * Unless required by applicable law or agreed to in writing, software 114f94c520f8d699a5973956a1716272146be17128Zonr Chang * distributed under the License is distributed on an "AS IS" BASIS, 124f94c520f8d699a5973956a1716272146be17128Zonr Chang * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 134f94c520f8d699a5973956a1716272146be17128Zonr Chang * See the License for the specific language governing permissions and 144f94c520f8d699a5973956a1716272146be17128Zonr Chang * limitations under the License. 154f94c520f8d699a5973956a1716272146be17128Zonr Chang */ 164f94c520f8d699a5973956a1716272146be17128Zonr Chang 17c72c4ddfcd79c74f70713da91a69569451b5c19eZonr Chang#include "bcc/Support/OutputFile.h" 184f94c520f8d699a5973956a1716272146be17128Zonr Chang 194f94c520f8d699a5973956a1716272146be17128Zonr Chang#include <cstdlib> 204f94c520f8d699a5973956a1716272146be17128Zonr Chang 214f94c520f8d699a5973956a1716272146be17128Zonr Chang#include <llvm/Support/raw_ostream.h> 224f94c520f8d699a5973956a1716272146be17128Zonr Chang 23ef73a242762bcd8113b9b65ceccbe7d909b5acbcZonr Chang#include "bcc/Support/Log.h" 244f94c520f8d699a5973956a1716272146be17128Zonr Chang 254f94c520f8d699a5973956a1716272146be17128Zonr Changusing namespace bcc; 264f94c520f8d699a5973956a1716272146be17128Zonr Chang 274f94c520f8d699a5973956a1716272146be17128Zonr ChangOutputFile::OutputFile(const std::string &pFilename, unsigned pFlags) 284f94c520f8d699a5973956a1716272146be17128Zonr Chang : super(pFilename, pFlags) { } 294f94c520f8d699a5973956a1716272146be17128Zonr Chang 304f94c520f8d699a5973956a1716272146be17128Zonr Changssize_t OutputFile::write(const void *pBuf, size_t count) { 314f94c520f8d699a5973956a1716272146be17128Zonr Chang if ((mFD < 0) || hasError()) { 324f94c520f8d699a5973956a1716272146be17128Zonr Chang return -1; 334f94c520f8d699a5973956a1716272146be17128Zonr Chang } 344f94c520f8d699a5973956a1716272146be17128Zonr Chang 35900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if ((count <= 0) || (pBuf == nullptr)) { 364f94c520f8d699a5973956a1716272146be17128Zonr Chang // Keep safe and issue a warning. 37d14994d0c77dbd688d6771f5283554aa0ad55808Shih-wei Liao ALOGW("OutputFile::write: count = %zu, buffer = %p", count, pBuf); 384f94c520f8d699a5973956a1716272146be17128Zonr Chang return 0; 394f94c520f8d699a5973956a1716272146be17128Zonr Chang } 404f94c520f8d699a5973956a1716272146be17128Zonr Chang 414f94c520f8d699a5973956a1716272146be17128Zonr Chang while (count > 0) { 424f94c520f8d699a5973956a1716272146be17128Zonr Chang ssize_t write_size = ::write(mFD, pBuf, count); 434f94c520f8d699a5973956a1716272146be17128Zonr Chang 444f94c520f8d699a5973956a1716272146be17128Zonr Chang if (write_size > 0) { 454f94c520f8d699a5973956a1716272146be17128Zonr Chang return write_size; 464f94c520f8d699a5973956a1716272146be17128Zonr Chang } else if ((errno == EAGAIN) || (errno == EINTR)) { 474f94c520f8d699a5973956a1716272146be17128Zonr Chang // If the errno is EAGAIN or EINTR, then we try to write again. 484f94c520f8d699a5973956a1716272146be17128Zonr Chang // 494f94c520f8d699a5973956a1716272146be17128Zonr Chang // Fall-through 504f94c520f8d699a5973956a1716272146be17128Zonr Chang } else { 514f94c520f8d699a5973956a1716272146be17128Zonr Chang detectError(); 524f94c520f8d699a5973956a1716272146be17128Zonr Chang return -1; 534f94c520f8d699a5973956a1716272146be17128Zonr Chang } 544f94c520f8d699a5973956a1716272146be17128Zonr Chang } 554f94c520f8d699a5973956a1716272146be17128Zonr Chang // unreachable 564f94c520f8d699a5973956a1716272146be17128Zonr Chang return 0; 574f94c520f8d699a5973956a1716272146be17128Zonr Chang} 584f94c520f8d699a5973956a1716272146be17128Zonr Chang 594f94c520f8d699a5973956a1716272146be17128Zonr Changvoid OutputFile::truncate() { 604f94c520f8d699a5973956a1716272146be17128Zonr Chang if (mFD < 0) { 614f94c520f8d699a5973956a1716272146be17128Zonr Chang return; 624f94c520f8d699a5973956a1716272146be17128Zonr Chang } 634f94c520f8d699a5973956a1716272146be17128Zonr Chang 644f94c520f8d699a5973956a1716272146be17128Zonr Chang do { 654f94c520f8d699a5973956a1716272146be17128Zonr Chang if (::ftruncate(mFD, 0) == 0) { 664f94c520f8d699a5973956a1716272146be17128Zonr Chang return; 674f94c520f8d699a5973956a1716272146be17128Zonr Chang } 684f94c520f8d699a5973956a1716272146be17128Zonr Chang } while (errno == EINTR); 694f94c520f8d699a5973956a1716272146be17128Zonr Chang detectError(); 704f94c520f8d699a5973956a1716272146be17128Zonr Chang 714f94c520f8d699a5973956a1716272146be17128Zonr Chang return; 724f94c520f8d699a5973956a1716272146be17128Zonr Chang} 734f94c520f8d699a5973956a1716272146be17128Zonr Chang 744f94c520f8d699a5973956a1716272146be17128Zonr Changllvm::raw_fd_ostream *OutputFile::dup() { 754f94c520f8d699a5973956a1716272146be17128Zonr Chang int newfd; 764f94c520f8d699a5973956a1716272146be17128Zonr Chang 774f94c520f8d699a5973956a1716272146be17128Zonr Chang do { 784f94c520f8d699a5973956a1716272146be17128Zonr Chang newfd = ::dup(mFD); 794f94c520f8d699a5973956a1716272146be17128Zonr Chang if (newfd < 0) { 804f94c520f8d699a5973956a1716272146be17128Zonr Chang if (errno != EINTR) { 814f94c520f8d699a5973956a1716272146be17128Zonr Chang detectError(); 82900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes return nullptr; 834f94c520f8d699a5973956a1716272146be17128Zonr Chang } 844f94c520f8d699a5973956a1716272146be17128Zonr Chang // EINTR 854f94c520f8d699a5973956a1716272146be17128Zonr Chang continue; 864f94c520f8d699a5973956a1716272146be17128Zonr Chang } 874f94c520f8d699a5973956a1716272146be17128Zonr Chang // dup() returns ok. 884f94c520f8d699a5973956a1716272146be17128Zonr Chang break; 894f94c520f8d699a5973956a1716272146be17128Zonr Chang } while (true); 904f94c520f8d699a5973956a1716272146be17128Zonr Chang 914f94c520f8d699a5973956a1716272146be17128Zonr Chang llvm::raw_fd_ostream *result = 924f94c520f8d699a5973956a1716272146be17128Zonr Chang new (std::nothrow) llvm::raw_fd_ostream(newfd, /* shouldClose */true); 934f94c520f8d699a5973956a1716272146be17128Zonr Chang 94900c6c1f08f7c572125d7d39abe0f0f9eafbfa14Chris Wailes if (result == nullptr) { 95d0993af5b1337f8186dd9aedea2e138a919e02e9Stephen Hines mError = std::make_error_code(std::errc::not_enough_memory); 964f94c520f8d699a5973956a1716272146be17128Zonr Chang } 974f94c520f8d699a5973956a1716272146be17128Zonr Chang 984f94c520f8d699a5973956a1716272146be17128Zonr Chang return result; 994f94c520f8d699a5973956a1716272146be17128Zonr Chang} 100