Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(160)

Unified Diff: tools/pnacl-llc/SRPCStreamer.cpp

Issue 939073008: Rebased PNaCl localmods in LLVM to 223109 (Closed)
Patch Set: undo localmod Created 5 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « tools/pnacl-llc/SRPCStreamer.h ('k') | tools/pnacl-llc/ThreadedFunctionQueue.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: tools/pnacl-llc/SRPCStreamer.cpp
diff --git a/tools/pnacl-llc/SRPCStreamer.cpp b/tools/pnacl-llc/SRPCStreamer.cpp
new file mode 100644
index 0000000000000000000000000000000000000000..577dc174fbdf554a19442e9083c1ca712c779e96
--- /dev/null
+++ b/tools/pnacl-llc/SRPCStreamer.cpp
@@ -0,0 +1,148 @@
+//===-- SRPCStreamer.cpp - Stream bitcode over SRPC ----------------------===//
+//
+// The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+#if defined(PNACL_BROWSER_TRANSLATOR)
+#define DEBUG_TYPE "bitcode-stream"
+#include "SRPCStreamer.h"
+#include "llvm/Support/Debug.h"
+#include "llvm/Support/raw_ostream.h"
+#include <errno.h>
+
+using llvm::dbgs;
+
+const size_t QueueStreamer::queuesize_limit_;
+
+size_t QueueStreamer::GetBytes(unsigned char *buf, size_t len) {
+ size_t total_copied = 0;
+ pthread_mutex_lock(&Mutex);
+ while (!Done && queueSize() < len - total_copied) {
+ size_t size = queueSize();
+ DEBUG(dbgs() << "QueueStreamer::GetBytes len " << len << " size " <<
+ size << " << waiting\n");
+ queueGet(buf + total_copied, size);
+ total_copied += size;
+ pthread_cond_signal(&Cond);
+ pthread_cond_wait(&Cond, &Mutex);
+ }
+ // If this is the last partial chunk, adjust len such that the amount we
+ // fetch will be just the remaining bytes.
+ if (Done && queueSize() < len - total_copied) {
+ len = queueSize() + total_copied;
+ }
+ queueGet(buf + total_copied, len - total_copied);
+ pthread_cond_signal(&Cond);
+ pthread_mutex_unlock(&Mutex);
+ return len;
+}
+
+size_t QueueStreamer::PutBytes(unsigned char *buf, size_t len) {
+ size_t total_copied = 0;
+ pthread_mutex_lock(&Mutex);
+ while (capacityRemaining() < len - total_copied) {
+ if (Bytes.size() * 2 > queuesize_limit_) {
+ size_t space = capacityRemaining();
+ queuePut(buf + total_copied, space);
+ total_copied += space;
+ pthread_cond_signal(&Cond);
+ pthread_cond_wait(&Cond, &Mutex);
+ } else {
+ queueResize();
+ }
+ }
+ queuePut(buf + total_copied, len - total_copied);
+ pthread_cond_signal(&Cond);
+ pthread_mutex_unlock(&Mutex);
+ return len;
+}
+
+void QueueStreamer::SetDone() {
+ // Still need the lock to avoid signaling between the check and
+ // the wait in GetBytes.
+ pthread_mutex_lock(&Mutex);
+ Done = true;
+ pthread_cond_signal(&Cond);
+ pthread_mutex_unlock(&Mutex);
+}
+
+// Double the size of the queue. Called with Mutex to protect Cons/Prod/Bytes.
+void QueueStreamer::queueResize() {
+ int leftover = Bytes.size() - Cons;
+ DEBUG(dbgs() << "resizing to " << Bytes.size() * 2 << " " << leftover << " "
+ << Prod << " " << Cons << "\n");
+ Bytes.resize(Bytes.size() * 2);
+ if (Cons > Prod) {
+ // There are unread bytes left between Cons and the previous end of the
+ // buffer. Move them to the new end of the buffer.
+ memmove(&Bytes[Bytes.size() - leftover], &Bytes[Cons], leftover);
+ Cons = Bytes.size() - leftover;
+ }
+}
+
+// Called with Mutex held to protect Cons, Prod, and Bytes
+void QueueStreamer::queuePut(unsigned char *buf, size_t len) {
+ size_t EndSpace = std::min(len, Bytes.size() - Prod);
+ DEBUG(dbgs() << "put, len " << len << " Endspace " << EndSpace << " p " <<
+ Prod << " c " << Cons << "\n");
+ // Copy up to the end of the buffer
+ memcpy(&Bytes[Prod], buf, EndSpace);
+ // Wrap around if necessary
+ memcpy(&Bytes[0], buf + EndSpace, len - EndSpace);
+ Prod = (Prod + len) % Bytes.size();
+}
+
+// Called with Mutex held to protect Cons, Prod, and Bytes
+void QueueStreamer::queueGet(unsigned char *buf, size_t len) {
+ assert(len <= queueSize());
+ size_t EndSpace = std::min(len, Bytes.size() - Cons);
+ DEBUG(dbgs() << "get, len " << len << " Endspace " << EndSpace << " p " <<
+ Prod << " c " << Cons << "\n");
+ // Copy up to the end of the buffer
+ memcpy(buf, &Bytes[Cons], EndSpace);
+ // Wrap around if necessary
+ memcpy(buf + EndSpace, &Bytes[0], len - EndSpace);
+ Cons = (Cons + len) % Bytes.size();
+}
+
+llvm::DataStreamer *SRPCStreamer::init(void *(*Callback)(void *), void *arg,
+ std::string *ErrMsg) {
+ int err = pthread_create(&CompileThread, NULL, Callback, arg);
+ if (err) {
+ if (ErrMsg) *ErrMsg = std::string(strerror(errno));
+ return NULL;
+ }
+ return &Q;
+}
+
+size_t SRPCStreamer::gotChunk(unsigned char *bytes, size_t len) {
+ if (__sync_fetch_and_add(&Error, 0)) return 0; // Atomic read.
+ return Q.PutBytes(bytes, len);
+}
+
+int SRPCStreamer::streamEnd(std::string *ErrMsg) {
+ Q.SetDone();
+ int err = pthread_join(CompileThread, NULL);
+ __sync_synchronize();
+ if (Error) {
+ if (ErrMsg)
+ *ErrMsg = std::string("PNaCl Translator Error: " + ErrorMessage);
+ return 1;
+ } else if (err) {
+ if (ErrMsg) *ErrMsg = std::string(strerror(errno));
+ return err;
+ }
+ return 0;
+}
+
+void SRPCStreamer::setFatalError(const std::string& message) {
+ __sync_fetch_and_add(&Error, 1);
+ ErrorMessage = message;
+ __sync_synchronize();
+ pthread_exit(NULL);
+}
+
+#endif // __native_client__
« no previous file with comments | « tools/pnacl-llc/SRPCStreamer.h ('k') | tools/pnacl-llc/ThreadedFunctionQueue.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698