1fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Protocol Buffers - Google's data interchange format
2fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Copyright 2009 Google Inc.  All rights reserved.
3afb4b72037e3f13db208590fc782c4bc8e27f862Jeff Davidson// https://developers.google.com/protocol-buffers/
4fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
5fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Redistribution and use in source and binary forms, with or without
6fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// modification, are permitted provided that the following conditions are
7fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// met:
8fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
9fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Redistributions of source code must retain the above copyright
10fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// notice, this list of conditions and the following disclaimer.
11fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Redistributions in binary form must reproduce the above
12fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// copyright notice, this list of conditions and the following disclaimer
13fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// in the documentation and/or other materials provided with the
14fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// distribution.
15fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Neither the name of Google Inc. nor the names of its
16fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// contributors may be used to endorse or promote products derived from
17fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// this software without specific prior written permission.
18fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
19fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
31fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Author: brianolson@google.com (Brian Olson)
32fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//  Based on original Protocol Buffers design by
33fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//  Sanjay Ghemawat, Jeff Dean, and others.
34fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
35fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Test program to verify that GzipOutputStream is compatible with command line
36fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// gzip or java.util.zip.GzipOutputStream
37fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
38fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Reads data on standard input and writes compressed gzip stream to standard
39fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// output.
40fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
41fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include "config.h"
42fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
43fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <stdio.h>
44fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <stdlib.h>
45fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <fcntl.h>
46fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
47fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/io/gzip_stream.h>
48fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/io/zero_copy_stream_impl.h>
49fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
50fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleusing google::protobuf::io::FileOutputStream;
51fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleusing google::protobuf::io::GzipOutputStream;
52fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
53fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleint main(int argc, const char** argv) {
54fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileOutputStream fout(STDOUT_FILENO);
55fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GzipOutputStream out(&fout);
56fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  int readlen;
57fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
58fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  while (true) {
59fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    void* outptr;
60fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    int outlen;
61fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool ok;
62fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    do {
63fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      ok = out.Next(&outptr, &outlen);
64fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (!ok) {
65fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        break;
66fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
67fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } while (outlen <= 0);
68fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    readlen = read(STDIN_FILENO, outptr, outlen);
69fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (readlen <= 0) {
70fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      out.BackUp(outlen);
71fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
72fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
73fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (readlen < outlen) {
74fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      out.BackUp(outlen - readlen);
75fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
76fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
77fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
78fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return 0;
79fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
80