1 files changed, 68 insertions, 0 deletions
diff --git a/extensions/common/crx_file.cc b/extensions/common/crx_file.cc
new file mode 100644
index 0000000..4f50962
--- /dev/null
+++ b/extensions/common/crx_file.cc
@@ -0,0 +1,68 @@
+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "extensions/common/crx_file.h"
+
+namespace extensions {
+
+namespace {
+
+// The current version of the crx format.
+static const uint32 kCurrentVersion = 2;
+
+// The maximum size the crx parser will tolerate for a public key.
+static const uint32 kMaxPublicKeySize = 1 << 16;
+
+// The maximum size the crx parser will tolerate for a signature.
+static const uint32 kMaxSignatureSize = 1 << 16;
+
+}  // namespace
+
+// The magic string embedded in the header.
+const char kCrxFileHeaderMagic[] = "Cr24";
+
+scoped_ptr<CrxFile> CrxFile::Parse(const CrxFile::Header& header,
+                                   CrxFile::Error* error) {
+  if (HeaderIsValid(header, error))
+    return scoped_ptr<CrxFile>(new CrxFile(header));
+  return scoped_ptr<CrxFile>();
+}
+
+scoped_ptr<CrxFile> CrxFile::Create(const uint32 key_size,
+                                    const uint32 signature_size,
+                                    CrxFile::Error* error) {
+  CrxFile::Header header;
+  memcpy(&header.magic, kCrxFileHeaderMagic, kCrxFileHeaderMagicSize);
+  header.version = kCurrentVersion;
+  header.key_size = key_size;
+  header.signature_size = signature_size;
+  if (HeaderIsValid(header, error))
+    return scoped_ptr<CrxFile>(new CrxFile(header));
+  return scoped_ptr<CrxFile>();
+}
+
+CrxFile::CrxFile(const Header& header) : header_(header) {
+}
+
+bool CrxFile::HeaderIsValid(const CrxFile::Header& header,
+                            CrxFile::Error* error) {
+  bool valid = false;
+  if (strncmp(kCrxFileHeaderMagic, header.magic, sizeof(header.magic)))
+    *error = kWrongMagic;
+  else if (header.version != kCurrentVersion)
+    *error = kInvalidVersion;
+  else if (header.key_size > kMaxPublicKeySize)
+    *error = kInvalidKeyTooLarge;
+  else if (header.key_size == 0)
+    *error = kInvalidKeyTooSmall;
+  else if (header.signature_size > kMaxSignatureSize)
+    *error = kInvalidSignatureTooLarge;
+  else if (header.signature_size == 0)
+    *error = kInvalidSignatureTooSmall;
+  else
+    valid = true;
+  return valid;
+}
+
+}  // namespace extensions