Method PullData

cpp/iedriver/HookProcessor.cpp:284–323

}

int HookProcessor::PullData(std::vector<char>* data) {
  LOG(TRACE) << "Entering HookProcessor::PullData";
  std::vector<char> buffer(NAMED_PIPE_BUFFER_SIZE);

   // Wait for the client to connect; if it succeeds, 
   // the function returns a nonzero value. If the function
   // returns zero, GetLastError returns ERROR_PIPE_CONNECTED.
  LOG(DEBUG) << "Waiting for connection from browser via named pipe";
  bool is_connected = true;
  if (!::ConnectNamedPipe(this->pipe_handle_, NULL)) {
    DWORD error = ::GetLastError();
    if (error != ERROR_PIPE_CONNECTED) {
      is_connected = false;
    }
  }
  if (is_connected) {
    LOG(DEBUG) << "Connection from browser established via named pipe";
    unsigned long bytes_read = 0;
    BOOL is_read_successful = ::ReadFile(this->pipe_handle_,
                                         &buffer[0],
                                         NAMED_PIPE_BUFFER_SIZE,
                                         &bytes_read,
                                         NULL);
    while (!is_read_successful && ERROR_MORE_DATA == ::GetLastError()) {
      data->insert(data->end(), buffer.begin(), buffer.begin() + bytes_read);
      is_read_successful = ::ReadFile(this->pipe_handle_,
                                      &buffer[0],
                                      NAMED_PIPE_BUFFER_SIZE,
                                      &bytes_read,
                                      NULL);
    }
    if (is_read_successful) {
      data->insert(data->end(), buffer.begin(), buffer.begin() + bytes_read);
    }
    ::DisconnectNamedPipe(this->pipe_handle_);
  } else {
    LOG(WARN) << "No connection received from browser via named pipe";
  }
  return static_cast<int>(data->size());
}

void HookProcessor::ResetFlag() {
  flag = false;