From 64dc5d656629b78de7a8f90a15d164eac8f73ba2 Mon Sep 17 00:00:00 2001
From: Quim Rovira <quim@rovira.cat>
Date: Sun, 14 Aug 2016 12:57:31 +0200
Subject: [PATCH] Add --repeat and --verbose options to bro-perl

---
 Makefile.PL  |   1 +
 bin/bro-perl | 103 ++++++++++++++++++++++++++++++++++-----------------
 2 files changed, 71 insertions(+), 33 deletions(-)

diff --git a/Makefile.PL b/Makefile.PL
index 79ea79a..bfbb161 100644
--- a/Makefile.PL
+++ b/Makefile.PL
@@ -13,6 +13,7 @@ WriteMakefile(
 	PREREQ_PM        => {
 		'File::Slurp'  => '9999.19',
 		'Getopt::Long' => '2.45',
+		'Time:HiRes'   => '1.9728'
 	},
 	BUILD_REQUIRES   => {},
 	INC              => '-Ibrotli',
diff --git a/bin/bro-perl b/bin/bro-perl
index 89a4e00..6aac129 100755
--- a/bin/bro-perl
+++ b/bin/bro-perl
@@ -5,8 +5,11 @@ use warnings;
 use strict;
 use 5.014;
 
+use bytes;
+
 use File::Slurp;
 use Getopt::Long;
+use Time::HiRes qw/ gettimeofday tv_interval /;
 
 use IO::Compress::Brotli;
 use IO::Uncompress::Brotli;
@@ -14,11 +17,12 @@ use IO::Uncompress::Brotli;
 GetOptions(
     'c|custom-dictionary=s' => \(my $DICTIONARY),
     'd|decompress'          => \(my $DECOMPRESS),
+    'f|force'               => \(my $FORCE),
     'h|help'                => \(my $HELP),
     'i|input=s'             => \(my $INPUT),
     'o|output=s'            => \(my $OUTPUT),
     'q|quality=i'           => \(my $QUALITY = 11),
-    'r|repeat=i'            => \(my $REPEAT),
+    'r|repeat=i'            => \(my $REPEAT = 1),
     's|stream=i'            => \(my $STREAM),
     'v|verbose'             => \(my $VERBOSE),
     'w|window=i'            => \(my $WINDOW = 22),
@@ -30,46 +34,79 @@ if( $HELP ) {
     exit 1;
 }
 
-my ($ifh, $ofh) = (\*STDIN, \*STDOUT);
-
-if( $INPUT ) {
-    open $ifh, "<", $INPUT
-        or die "Cannot open input file $INPUT.\n";
+if( $REPEAT > 1 && !($INPUT && $OUTPUT) ) {
+    say "You can only run a benchmark on files specifying --input and --output";
+    exit 1;
 }
-binmode $ifh;
 
-if( $OUTPUT ) {
-    open $ofh, ">", $OUTPUT
-        or die "Cannot open output file $OUTPUT.\n";
-}
-binmode $ofh;
+my $t0 = [gettimeofday];
+my $total_size = 0;
+my ($encoded, $decoded);
 
-if( $DECOMPRESS ) {
-    if( $STREAM ) {
-        my $brotli = IO::Uncompress::Brotli->create();
-        while( read $ifh, (my $buf), $STREAM ) {
-            print $ofh $brotli->decompress($buf);
-        }
+for ( 1..$REPEAT ) {
+    my $ifh;
+    if( $INPUT ) {
+        open $ifh, "<", $INPUT
+            or die "Cannot open input file $INPUT.\n";
     }
-    else {
-        my $encoded = read_file( $ifh );
-        my $decoded = unbro( $encoded );
-        write_file( $ofh, $decoded );
+    $ifh //= \*STDIN;
+    binmode $ifh;
+
+    my $ofh;
+    if( $OUTPUT ) {
+        die "Output file exists"
+            if( -e $OUTPUT && $REPEAT == 1 && !$FORCE );
+        open $ofh, ">", $OUTPUT
+            or die "Cannot open output file $OUTPUT.\n";
     }
-}
-else {
-    if( $STREAM ) {
-        my $brotli = IO::Compress::Brotli->create();
-        $brotli->quality( $QUALITY );
-        $brotli->window( $WINDOW );
-        while( read $ifh, (my $buf), $STREAM ) {
-            print $ofh $brotli->compress($buf);
+    $ofh //= \*STDOUT;
+    binmode $ofh;
+
+    if( $DECOMPRESS ) {
+        if( $STREAM ) {
+            my $bro = IO::Uncompress::Brotli->create();
+            while( read $ifh, (my $buf), $STREAM ) {
+                $decoded = $bro->decompress($buf);
+                $total_size += bytes::length( $decoded );
+                print $ofh $decoded;
+            }
+        }
+        else {
+            $encoded = read_file( $ifh );
+            $decoded = unbro( $encoded );
+            $total_size += bytes::length( $decoded );
+            write_file( $ofh, $decoded );
         }
     }
     else {
-        my $decoded = read_file( $ifh );
-        my $encoded = bro( $decoded, $QUALITY, $WINDOW );
-        write_file( $ofh, $encoded );
+        if( $STREAM ) {
+            my $bro = IO::Compress::Brotli->create();
+            $bro->quality( $QUALITY );
+            $bro->window( $WINDOW );
+            while( read $ifh, (my $buf), $STREAM ) {
+                $encoded = $bro->compress($buf);
+                $total_size += bytes::length( $encoded );
+                print $ofh $encoded;
+            }
+            $encoded = $bro->finish();
+            $total_size += bytes::length( $encoded );
+            print $ofh $encoded;
+        }
+        else {
+            my $decoded = read_file( $ifh );
+            my $encoded = bro( $decoded, $QUALITY, $WINDOW );
+            $total_size += bytes::length( $encoded );
+            write_file( $ofh, $encoded );
+        }
     }
 }
 
+if( $VERBOSE ) {
+    my $elapsed = tv_interval ( $t0 );
+    say "Ran $REPEAT iterations in a total of $elapsed seconds";
+    say sprintf(
+        "Brotli %s speed: %.6f MB/s",
+        ( $DECOMPRESS ? "decompression" : "compression" ),
+        $total_size / 1024 / 1024 / $elapsed
+    );
+}
-- 
2.30.2