summaryrefslogtreecommitdiffstats
path: root/comm/mailnews/local/test/unit/test_over2GBMailboxes.js
blob: fe7976b0fe7d9842e3322b1473bd8428d392ca7b (plain)
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* Test of accessing over 2 GiB local folder. */

const { PromiseTestUtils } = ChromeUtils.import(
  "resource://testing-common/mailnews/PromiseTestUtils.jsm"
);

// Find hdr for message whose offset is over 2 GiB.
function findHugeMessageHdr(folder) {
  //getMessageHdr() {
  for (let header of folder.msgDatabase.enumerateMessages()) {
    if (header.messageOffset >= 0x80000000) {
      return header;
    }
  }

  do_throw("Over 2 GiB msgkey was not found!");
  return null; // This won't ever happen, but we're keeping the linter happy.
}

let gInboxFile;
let gInbox;
let gSmallMsgFile = do_get_file("../../../data/bugmail10");

add_setup(async function () {
  // Make sure we're using mbox.
  Services.prefs.setCharPref(
    "mail.serverDefaultStoreContractID",
    "@mozilla.org/msgstore/berkeleystore;1"
  );

  localAccountUtils.loadLocalMailAccount();

  gInbox = localAccountUtils.inboxFolder;
  gInboxFile = gInbox.filePath;

  let neededFreeSpace = 0x100000000;
  let freeDiskSpace = gInboxFile.diskSpaceAvailable;
  info("Free disk space = " + mailTestUtils.toMiBString(freeDiskSpace));
  if (freeDiskSpace < neededFreeSpace) {
    throw new Error(
      "This test needs " +
        mailTestUtils.toMiBString(neededFreeSpace) +
        " free space to run. Aborting."
    );
  }
});

// Extend mbox file to over 2 GiB.
add_task(async function extendPast2GiB() {
  let outputStream = Cc["@mozilla.org/network/file-output-stream;1"]
    .createInstance(Ci.nsIFileOutputStream)
    .QueryInterface(Ci.nsISeekableStream);
  // Open in write-only mode, no truncate.
  outputStream.init(gInboxFile, 0x02, -1, 0);
  // seek past 2GB.
  outputStream.seek(0, 0x80000010);
  // Write a "space" character.
  outputStream.write(" ", 1);
  outputStream.close();
});

// Copy another (normal sized) message into the local folder.
// This message should be past the 2GiB position.
add_task(async function appendSmallMessage() {
  // Remember initial mbox file size.
  let initialInboxSize = gInbox.filePath.fileSize;
  info(`Local inbox size (before copyFileMessage()) = ${initialInboxSize}`);

  let copyListener = new PromiseTestUtils.PromiseCopyListener();
  MailServices.copy.copyFileMessage(
    gSmallMsgFile,
    gInbox,
    null /* msgToReplace*/,
    false /* isDraftOrTemplate */,
    0 /* message flags */,
    "" /* keywords */,
    copyListener,
    null /* window */
  );
  await copyListener.promise;

  // Make sure inbox file grew (i.e., we were not writing over data).
  let localInboxSize = gInbox.filePath.fileSize;
  info(
    "Local inbox size (after copyFileMessageInLocalFolder()) = " +
      localInboxSize
  );
  Assert.greater(localInboxSize, initialInboxSize);
});

// Copy the huge message into a subfolder.
add_task(async function copyHugeMessage() {
  let trash =
    localAccountUtils.incomingServer.rootMsgFolder.getChildNamed("Trash");
  let copyListener = new PromiseTestUtils.PromiseCopyListener();
  MailServices.copy.copyMessages(
    gInbox,
    [findHugeMessageHdr(gInbox)],
    trash /* destFolder */,
    false,
    copyListener,
    null,
    false
  );
  await copyListener.promise;
});

// Read out the smaller message beyond the 2 GiB offset and make sure
// it matches what we expect.
add_task(async function verifySmallMessage() {
  let msghdr = findHugeMessageHdr(gInbox);
  let msgURI = msghdr.folder.getUriForMsg(msghdr);
  let msgServ = MailServices.messageServiceFromURI(msgURI);

  let streamListener = new PromiseTestUtils.PromiseStreamListener();
  msgServ.streamMessage(msgURI, streamListener, null, null, false, "", true);
  let got = await streamListener.promise;

  let expected = await IOUtils.readUTF8(gSmallMsgFile.path);
  Assert.equal(got, expected);
});

add_task(async function cleanup() {
  // Free up disk space - if you want to look at the file after running
  // this test, comment out this line.
  gInbox.filePath.remove(false);
});