File: pdfencrypt

package info (click to toggle)
origami-pdf 2.1.0-1~exp2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,444 kB
  • sloc: ruby: 17,883; makefile: 8
file content (113 lines) | stat: -rwxr-xr-x 3,033 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
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
#!/usr/bin/env ruby

=begin

= Info
    Encrypts a PDF document.

= License
    Copyright (C) 2016  Guillaume Delugré.

    Origami is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Origami is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with Origami.  If not, see <http://www.gnu.org/licenses/>.

=end

begin
    require 'origami'
rescue LoadError
    $: << File.join(__dir__, '../lib')
    require 'origami'
end
include Origami

require 'optparse'

class OptParser
    BANNER = <<USAGE
Usage: #{$0} [<PDF-file>] [-p <password>] [-c <cipher>] [-s <key-size>] [--hardened] [-o <output-file>]
Encrypts a PDF document. Supports RC4 40 to 128 bits, AES128, AES256.
Bug reports or feature requests at: http://github.com/gdelugre/origami

Options:
USAGE

    def self.parser(options)
        OptionParser.new do |opts|
            opts.banner = BANNER

            opts.on("-o", "--output FILE", "Output PDF file (stdout by default)") do |o|
                options[:output] = o
            end

            opts.on("-p", "--password PASSWORD", "Password of the document") do |p|
                options[:password] = p
            end

            opts.on("-c", "--cipher CIPHER", "Cipher used to encrypt the document (Default: AES)") do |c|
                options[:cipher] = c
            end

            opts.on("-s", "--key-size KEYSIZE", "Key size in bits (Default: 256)") do |s|
                options[:key_size] = s.to_i
            end

            opts.on("--hardened", "Use stronger key validation scheme (only AES-256)") do
                options[:hardened] = true
            end

            opts.on_tail("-h", "--help", "Show this message") do
                puts opts
                exit
            end
        end
    end

    def self.parse(args)
        options =
        {
            output: STDOUT,
            password: '',
            cipher: 'aes',
            key_size: 256,
            hardened: false
        }

        self.parser(options).parse!(args)

        options
    end
end

begin
    @options = OptParser.parse(ARGV)

    target = (ARGV.empty?) ? STDIN : ARGV.shift
    params =
    {
        verbosity: Parser::VERBOSE_QUIET,
    }

    pdf = PDF.read(target, params)
    pdf.encrypt(
        user_passwd:    @options[:password],
        owner_passwd:   @options[:password],
        cipher:         @options[:cipher],
        key_size:       @options[:key_size],
        hardened:       @options[:hardened]
    )
    pdf.save(@options[:output], noindent: true)

rescue
    abort "#{$!.class}: #{$!.message}"
end