File: file_sink.cpp

package info (click to toggle)
firefox 144.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 4,637,504 kB
  • sloc: cpp: 7,576,692; javascript: 6,430,831; ansic: 3,748,119; python: 1,398,978; xml: 628,810; asm: 438,679; java: 186,194; sh: 63,212; makefile: 19,159; objc: 13,086; perl: 12,986; yacc: 4,583; cs: 3,846; pascal: 3,448; lex: 1,720; ruby: 1,003; exp: 762; php: 436; lisp: 258; awk: 247; sql: 66; sed: 53; csh: 10
file content (33 lines) | stat: -rw-r--r-- 1,088 bytes parent folder | download | duplicates (3) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "file_sink.h"
#include <iostream>
#include <string>
#include <windows.h>

// Open the download receiver
bool FileSink::open(std::wstring& filename) {
  // Note: this only succeeds if the filename does not exist.
  fileHandle.own(CreateFileW(filename.c_str(), GENERIC_WRITE, 0, nullptr,
                             CREATE_NEW, FILE_ATTRIBUTE_NORMAL, nullptr));
  if (fileHandle.get() == INVALID_HANDLE_VALUE) {
    return false;
  } else {
    return true;
  }
}

// Send data to the download receiver
bool FileSink::accept(char* buf, int bytesToWrite) {
  while (bytesToWrite > 0) {
    DWORD bytesWritten = 0;
    if (!WriteFile(fileHandle, buf, bytesToWrite, &bytesWritten, nullptr)) {
      std::cout << "Failed to write! " << GetLastError() << std::endl;
      return false;  // Some kind of error happened
    }
    bytesToWrite -= bytesWritten;
  }
  return true;
}