ikan Uploader :
Directory : /home/ubuntu/node-v16.18.1/src/
Upload File : |
| Current File : //home/ubuntu/node-v16.18.1/src/node_blob.cc |
#include "node_blob.h"
#include "async_wrap-inl.h"
#include "base_object-inl.h"
#include "env-inl.h"
#include "memory_tracker-inl.h"
#include "node_errors.h"
#include "node_external_reference.h"
#include "threadpoolwork-inl.h"
#include "v8.h"
#include <algorithm>
namespace node {
using v8::Array;
using v8::ArrayBuffer;
using v8::ArrayBufferView;
using v8::BackingStore;
using v8::Context;
using v8::EscapableHandleScope;
using v8::Function;
using v8::FunctionCallbackInfo;
using v8::FunctionTemplate;
using v8::HandleScope;
using v8::Local;
using v8::MaybeLocal;
using v8::Number;
using v8::Object;
using v8::String;
using v8::Uint32;
using v8::Undefined;
using v8::Value;
void Blob::Initialize(
Local<Object> target,
Local<Value> unused,
Local<Context> context,
void* priv) {
Environment* env = Environment::GetCurrent(context);
BlobBindingData* const binding_data =
env->AddBindingData<BlobBindingData>(context, target);
if (binding_data == nullptr) return;
env->SetMethod(target, "createBlob", New);
env->SetMethod(target, "storeDataObject", StoreDataObject);
env->SetMethod(target, "getDataObject", GetDataObject);
env->SetMethod(target, "revokeDataObject", RevokeDataObject);
FixedSizeBlobCopyJob::Initialize(env, target);
}
Local<FunctionTemplate> Blob::GetConstructorTemplate(Environment* env) {
Local<FunctionTemplate> tmpl = env->blob_constructor_template();
if (tmpl.IsEmpty()) {
tmpl = FunctionTemplate::New(env->isolate());
tmpl->InstanceTemplate()->SetInternalFieldCount(
BaseObject::kInternalFieldCount);
tmpl->Inherit(BaseObject::GetConstructorTemplate(env));
tmpl->SetClassName(
FIXED_ONE_BYTE_STRING(env->isolate(), "Blob"));
env->SetProtoMethod(tmpl, "toArrayBuffer", ToArrayBuffer);
env->SetProtoMethod(tmpl, "slice", ToSlice);
env->set_blob_constructor_template(tmpl);
}
return tmpl;
}
bool Blob::HasInstance(Environment* env, v8::Local<v8::Value> object) {
return GetConstructorTemplate(env)->HasInstance(object);
}
BaseObjectPtr<Blob> Blob::Create(Environment* env,
const std::vector<BlobEntry>& store,
size_t length) {
HandleScope scope(env->isolate());
Local<Function> ctor;
if (!GetConstructorTemplate(env)->GetFunction(env->context()).ToLocal(&ctor))
return BaseObjectPtr<Blob>();
Local<Object> obj;
if (!ctor->NewInstance(env->context()).ToLocal(&obj))
return BaseObjectPtr<Blob>();
return MakeBaseObject<Blob>(env, obj, store, length);
}
void Blob::New(const FunctionCallbackInfo<Value>& args) {
Environment* env = Environment::GetCurrent(args);
CHECK(args[0]->IsArray()); // sources
CHECK(args[1]->IsUint32()); // length
std::vector<BlobEntry> entries;
size_t length = args[1].As<Uint32>()->Value();
size_t len = 0;
Local<Array> ary = args[0].As<Array>();
for (size_t n = 0; n < ary->Length(); n++) {
Local<Value> entry;
if (!ary->Get(env->context(), n).ToLocal(&entry))
return;
CHECK(entry->IsArrayBufferView() || Blob::HasInstance(env, entry));
if (entry->IsArrayBufferView()) {
Local<ArrayBufferView> view = entry.As<ArrayBufferView>();
CHECK_EQ(view->ByteOffset(), 0);
std::shared_ptr<BackingStore> store = view->Buffer()->GetBackingStore();
size_t byte_length = view->ByteLength();
view->Buffer()->Detach(); // The Blob will own the backing store now.
entries.emplace_back(BlobEntry{std::move(store), byte_length, 0});
len += byte_length;
} else {
Blob* blob;
ASSIGN_OR_RETURN_UNWRAP(&blob, entry);
auto source = blob->entries();
entries.insert(entries.end(), source.begin(), source.end());
len += blob->length();
}
}
CHECK_EQ(length, len);
BaseObjectPtr<Blob> blob = Create(env, entries, length);
if (blob)
args.GetReturnValue().Set(blob->object());
}
void Blob::ToArrayBuffer(const FunctionCallbackInfo<Value>& args) {
Environment* env = Environment::GetCurrent(args);
Blob* blob;
ASSIGN_OR_RETURN_UNWRAP(&blob, args.Holder());
Local<Value> ret;
if (blob->GetArrayBuffer(env).ToLocal(&ret))
args.GetReturnValue().Set(ret);
}
void Blob::ToSlice(const FunctionCallbackInfo<Value>& args) {
Environment* env = Environment::GetCurrent(args);
Blob* blob;
ASSIGN_OR_RETURN_UNWRAP(&blob, args.Holder());
CHECK(args[0]->IsUint32());
CHECK(args[1]->IsUint32());
size_t start = args[0].As<Uint32>()->Value();
size_t end = args[1].As<Uint32>()->Value();
BaseObjectPtr<Blob> slice = blob->Slice(env, start, end);
if (slice)
args.GetReturnValue().Set(slice->object());
}
void Blob::MemoryInfo(MemoryTracker* tracker) const {
tracker->TrackFieldWithSize("store", length_);
}
MaybeLocal<Value> Blob::GetArrayBuffer(Environment* env) {
EscapableHandleScope scope(env->isolate());
size_t len = length();
std::shared_ptr<BackingStore> store =
ArrayBuffer::NewBackingStore(env->isolate(), len);
if (len > 0) {
unsigned char* dest = static_cast<unsigned char*>(store->Data());
size_t total = 0;
for (const auto& entry : entries()) {
unsigned char* src = static_cast<unsigned char*>(entry.store->Data());
src += entry.offset;
memcpy(dest, src, entry.length);
dest += entry.length;
total += entry.length;
CHECK_LE(total, len);
}
}
return scope.Escape(ArrayBuffer::New(env->isolate(), store));
}
BaseObjectPtr<Blob> Blob::Slice(Environment* env, size_t start, size_t end) {
CHECK_LE(start, length());
CHECK_LE(end, length());
CHECK_LE(start, end);
std::vector<BlobEntry> slices;
size_t total = end - start;
size_t remaining = total;
if (total == 0) return Create(env, slices, 0);
for (const auto& entry : entries()) {
if (start + entry.offset > entry.store->ByteLength()) {
start -= entry.length;
continue;
}
size_t offset = entry.offset + start;
size_t len = std::min(remaining, entry.store->ByteLength() - offset);
slices.emplace_back(BlobEntry{entry.store, len, offset});
remaining -= len;
start = 0;
if (remaining == 0)
break;
}
return Create(env, slices, total);
}
Blob::Blob(
Environment* env,
v8::Local<v8::Object> obj,
const std::vector<BlobEntry>& store,
size_t length)
: BaseObject(env, obj),
store_(store),
length_(length) {
MakeWeak();
}
BaseObjectPtr<BaseObject>
Blob::BlobTransferData::Deserialize(
Environment* env,
Local<Context> context,
std::unique_ptr<worker::TransferData> self) {
if (context != env->context()) {
THROW_ERR_MESSAGE_TARGET_CONTEXT_UNAVAILABLE(env);
return {};
}
return Blob::Create(env, store_, length_);
}
BaseObject::TransferMode Blob::GetTransferMode() const {
return BaseObject::TransferMode::kCloneable;
}
std::unique_ptr<worker::TransferData> Blob::CloneForMessaging() const {
return std::make_unique<BlobTransferData>(store_, length_);
}
void Blob::StoreDataObject(const v8::FunctionCallbackInfo<v8::Value>& args) {
Environment* env = Environment::GetCurrent(args);
BlobBindingData* binding_data =
Environment::GetBindingData<BlobBindingData>(args);
CHECK(args[0]->IsString()); // ID key
CHECK(Blob::HasInstance(env, args[1])); // Blob
CHECK(args[2]->IsUint32()); // Length
CHECK(args[3]->IsString()); // Type
Utf8Value key(env->isolate(), args[0]);
Blob* blob;
ASSIGN_OR_RETURN_UNWRAP(&blob, args[1]);
size_t length = args[2].As<Uint32>()->Value();
Utf8Value type(env->isolate(), args[3]);
binding_data->store_data_object(
std::string(*key, key.length()),
BlobBindingData::StoredDataObject(
BaseObjectPtr<Blob>(blob),
length,
std::string(*type, type.length())));
}
void Blob::RevokeDataObject(const v8::FunctionCallbackInfo<v8::Value>& args) {
BlobBindingData* binding_data =
Environment::GetBindingData<BlobBindingData>(args);
Environment* env = Environment::GetCurrent(args);
CHECK(args[0]->IsString()); // ID key
Utf8Value key(env->isolate(), args[0]);
binding_data->revoke_data_object(std::string(*key, key.length()));
}
void Blob::GetDataObject(const v8::FunctionCallbackInfo<v8::Value>& args) {
BlobBindingData* binding_data =
Environment::GetBindingData<BlobBindingData>(args);
Environment* env = Environment::GetCurrent(args);
CHECK(args[0]->IsString());
Utf8Value key(env->isolate(), args[0]);
BlobBindingData::StoredDataObject stored =
binding_data->get_data_object(std::string(*key, key.length()));
if (stored.blob) {
Local<Value> type;
if (!String::NewFromUtf8(
env->isolate(),
stored.type.c_str(),
v8::NewStringType::kNormal,
static_cast<int>(stored.type.length())).ToLocal(&type)) {
return;
}
Local<Value> values[] = {
stored.blob->object(),
Uint32::NewFromUnsigned(env->isolate(), stored.length),
type
};
args.GetReturnValue().Set(
Array::New(
env->isolate(),
values,
arraysize(values)));
}
}
FixedSizeBlobCopyJob::FixedSizeBlobCopyJob(
Environment* env,
Local<Object> object,
Blob* blob,
FixedSizeBlobCopyJob::Mode mode)
: AsyncWrap(env, object, AsyncWrap::PROVIDER_FIXEDSIZEBLOBCOPY),
ThreadPoolWork(env),
mode_(mode) {
if (mode == FixedSizeBlobCopyJob::Mode::SYNC) MakeWeak();
source_ = blob->entries();
length_ = blob->length();
}
void FixedSizeBlobCopyJob::AfterThreadPoolWork(int status) {
Environment* env = AsyncWrap::env();
CHECK_EQ(mode_, Mode::ASYNC);
CHECK(status == 0 || status == UV_ECANCELED);
std::unique_ptr<FixedSizeBlobCopyJob> ptr(this);
HandleScope handle_scope(env->isolate());
Context::Scope context_scope(env->context());
Local<Value> args[2];
if (status == UV_ECANCELED) {
args[0] = Number::New(env->isolate(), status),
args[1] = Undefined(env->isolate());
} else {
args[0] = Undefined(env->isolate());
args[1] = ArrayBuffer::New(env->isolate(), destination_);
}
ptr->MakeCallback(env->ondone_string(), arraysize(args), args);
}
void FixedSizeBlobCopyJob::DoThreadPoolWork() {
unsigned char* dest = static_cast<unsigned char*>(destination_->Data());
if (length_ > 0) {
size_t total = 0;
for (const auto& entry : source_) {
unsigned char* src = static_cast<unsigned char*>(entry.store->Data());
src += entry.offset;
memcpy(dest, src, entry.length);
dest += entry.length;
total += entry.length;
CHECK_LE(total, length_);
}
}
}
void FixedSizeBlobCopyJob::MemoryInfo(MemoryTracker* tracker) const {
tracker->TrackFieldWithSize("source", length_);
tracker->TrackFieldWithSize(
"destination",
destination_ ? destination_->ByteLength() : 0);
}
void FixedSizeBlobCopyJob::Initialize(Environment* env, Local<Object> target) {
v8::Local<v8::FunctionTemplate> job = env->NewFunctionTemplate(New);
job->Inherit(AsyncWrap::GetConstructorTemplate(env));
job->InstanceTemplate()->SetInternalFieldCount(
AsyncWrap::kInternalFieldCount);
env->SetProtoMethod(job, "run", Run);
env->SetConstructorFunction(target, "FixedSizeBlobCopyJob", job);
}
void FixedSizeBlobCopyJob::New(const FunctionCallbackInfo<Value>& args) {
static constexpr size_t kMaxSyncLength = 4096;
static constexpr size_t kMaxEntryCount = 4;
Environment* env = Environment::GetCurrent(args);
CHECK(args.IsConstructCall());
CHECK(args[0]->IsObject());
CHECK(Blob::HasInstance(env, args[0]));
Blob* blob;
ASSIGN_OR_RETURN_UNWRAP(&blob, args[0]);
// This is a fairly arbitrary heuristic. We want to avoid deferring to
// the threadpool if the amount of data being copied is small and there
// aren't that many entries to copy.
FixedSizeBlobCopyJob::Mode mode =
(blob->length() < kMaxSyncLength &&
blob->entries().size() < kMaxEntryCount) ?
FixedSizeBlobCopyJob::Mode::SYNC :
FixedSizeBlobCopyJob::Mode::ASYNC;
new FixedSizeBlobCopyJob(env, args.This(), blob, mode);
}
void FixedSizeBlobCopyJob::Run(const FunctionCallbackInfo<Value>& args) {
Environment* env = Environment::GetCurrent(args);
FixedSizeBlobCopyJob* job;
ASSIGN_OR_RETURN_UNWRAP(&job, args.Holder());
job->destination_ =
ArrayBuffer::NewBackingStore(env->isolate(), job->length_);
if (job->mode() == FixedSizeBlobCopyJob::Mode::ASYNC)
return job->ScheduleWork();
job->DoThreadPoolWork();
args.GetReturnValue().Set(
ArrayBuffer::New(env->isolate(), job->destination_));
}
void FixedSizeBlobCopyJob::RegisterExternalReferences(
ExternalReferenceRegistry* registry) {
registry->Register(New);
registry->Register(Run);
}
void BlobBindingData::StoredDataObject::MemoryInfo(
MemoryTracker* tracker) const {
tracker->TrackField("blob", blob);
tracker->TrackFieldWithSize("type", type.length());
}
BlobBindingData::StoredDataObject::StoredDataObject(
const BaseObjectPtr<Blob>& blob_,
size_t length_,
const std::string& type_)
: blob(blob_),
length(length_),
type(type_) {}
BlobBindingData::BlobBindingData(Environment* env, Local<Object> wrap)
: SnapshotableObject(env, wrap, type_int) {
MakeWeak();
}
void BlobBindingData::MemoryInfo(MemoryTracker* tracker) const {
tracker->TrackField("data_objects", data_objects_);
}
void BlobBindingData::store_data_object(
const std::string& uuid,
const BlobBindingData::StoredDataObject& object) {
data_objects_[uuid] = object;
}
void BlobBindingData::revoke_data_object(const std::string& uuid) {
if (data_objects_.find(uuid) == data_objects_.end()) {
return;
}
data_objects_.erase(uuid);
CHECK_EQ(data_objects_.find(uuid), data_objects_.end());
}
BlobBindingData::StoredDataObject BlobBindingData::get_data_object(
const std::string& uuid) {
auto entry = data_objects_.find(uuid);
if (entry == data_objects_.end())
return BlobBindingData::StoredDataObject {};
return entry->second;
}
void BlobBindingData::Deserialize(
Local<Context> context,
Local<Object> holder,
int index,
InternalFieldInfo* info) {
DCHECK_EQ(index, BaseObject::kEmbedderType);
HandleScope scope(context->GetIsolate());
Environment* env = Environment::GetCurrent(context);
BlobBindingData* binding =
env->AddBindingData<BlobBindingData>(context, holder);
CHECK_NOT_NULL(binding);
}
void BlobBindingData::PrepareForSerialization(
Local<Context> context,
v8::SnapshotCreator* creator) {
// Stored blob objects are not actually persisted.
}
InternalFieldInfo* BlobBindingData::Serialize(int index) {
DCHECK_EQ(index, BaseObject::kEmbedderType);
InternalFieldInfo* info = InternalFieldInfo::New(type());
return info;
}
constexpr FastStringKey BlobBindingData::type_name;
void Blob::RegisterExternalReferences(ExternalReferenceRegistry* registry) {
registry->Register(Blob::New);
registry->Register(Blob::ToArrayBuffer);
registry->Register(Blob::ToSlice);
registry->Register(Blob::StoreDataObject);
registry->Register(Blob::GetDataObject);
registry->Register(Blob::RevokeDataObject);
FixedSizeBlobCopyJob::RegisterExternalReferences(registry);
}
} // namespace node
NODE_MODULE_CONTEXT_AWARE_INTERNAL(blob, node::Blob::Initialize)
NODE_MODULE_EXTERNAL_REFERENCE(blob, node::Blob::RegisterExternalReferences)
Kontol Shell Bypass