org/sonews/util/io/ArticleReader.java
author chris <chris@marvin>
Fri, 26 Jun 2009 16:48:50 +0200
changeset 1 6fceb66e1ad7
child 3 2fdc9cc89502
permissions -rw-r--r--
Hooray... sonews/0.5.0 final

HG: Enter commit message. Lines beginning with 'HG:' are removed.
HG: Remove all lines to abort the collapse operation.
chris@1
     1
/*
chris@1
     2
 *   SONEWS News Server
chris@1
     3
 *   see AUTHORS for the list of contributors
chris@1
     4
 *
chris@1
     5
 *   This program is free software: you can redistribute it and/or modify
chris@1
     6
 *   it under the terms of the GNU General Public License as published by
chris@1
     7
 *   the Free Software Foundation, either version 3 of the License, or
chris@1
     8
 *   (at your option) any later version.
chris@1
     9
 *
chris@1
    10
 *   This program is distributed in the hope that it will be useful,
chris@1
    11
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
chris@1
    12
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
chris@1
    13
 *   GNU General Public License for more details.
chris@1
    14
 *
chris@1
    15
 *   You should have received a copy of the GNU General Public License
chris@1
    16
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
chris@1
    17
 */
chris@1
    18
chris@1
    19
package org.sonews.util.io;
chris@1
    20
chris@1
    21
import java.io.BufferedInputStream;
chris@1
    22
import java.io.BufferedOutputStream;
chris@1
    23
import java.io.ByteArrayOutputStream;
chris@1
    24
import java.io.IOException;
chris@1
    25
import java.io.InputStream;
chris@1
    26
import java.io.UnsupportedEncodingException;
chris@1
    27
import java.net.Socket;
chris@1
    28
import java.net.UnknownHostException;
chris@1
    29
import org.sonews.util.Log;
chris@1
    30
chris@1
    31
/**
chris@1
    32
 * Reads an news article from a NNTP server.
chris@1
    33
 * @author Christian Lins
chris@1
    34
 * @since sonews/0.5.0
chris@1
    35
 */
chris@1
    36
public class ArticleReader 
chris@1
    37
{
chris@1
    38
chris@1
    39
  private BufferedOutputStream out;
chris@1
    40
  private BufferedInputStream  in;
chris@1
    41
  private String               messageID;
chris@1
    42
  
chris@1
    43
  public ArticleReader(String host, int port, String messageID)
chris@1
    44
    throws IOException, UnknownHostException
chris@1
    45
  {
chris@1
    46
    this.messageID = messageID;
chris@1
    47
chris@1
    48
    // Connect to NNTP server
chris@1
    49
    Socket socket = new Socket(host, port);
chris@1
    50
    this.out = new BufferedOutputStream(socket.getOutputStream());
chris@1
    51
    this.in  = new BufferedInputStream(socket.getInputStream());
chris@1
    52
    String line = readln(this.in);
chris@1
    53
    if(!line.startsWith("200 "))
chris@1
    54
    {
chris@1
    55
      throw new IOException("Invalid hello from server: " + line);
chris@1
    56
    }
chris@1
    57
  }
chris@1
    58
  
chris@1
    59
  private boolean eofArticle(byte[] buf)
chris@1
    60
  {
chris@1
    61
    if(buf.length < 4)
chris@1
    62
    {
chris@1
    63
      return false;
chris@1
    64
    }
chris@1
    65
    
chris@1
    66
    int l = buf.length - 1;
chris@1
    67
    return buf[l-3] == 10 // '*\n'
chris@1
    68
        && buf[l-2] == '.'                   // '.'
chris@1
    69
        && buf[l-1] == 13 && buf[l] == 10;  // '\r\n'
chris@1
    70
  }
chris@1
    71
  
chris@1
    72
  public byte[] getArticleData()
chris@1
    73
    throws IOException, UnsupportedEncodingException
chris@1
    74
  {
chris@1
    75
    try
chris@1
    76
    {
chris@1
    77
      this.out.write(("ARTICLE " + this.messageID + "\r\n").getBytes("UTF-8"));
chris@1
    78
      this.out.flush();
chris@1
    79
chris@1
    80
      String line = readln(this.in);
chris@1
    81
      if(line.startsWith("220 "))
chris@1
    82
      {
chris@1
    83
        ByteArrayOutputStream buf = new ByteArrayOutputStream();
chris@1
    84
        
chris@1
    85
        while(!eofArticle(buf.toByteArray()))
chris@1
    86
        {
chris@1
    87
          for(int b = in.read(); b != 10; b = in.read())
chris@1
    88
          {
chris@1
    89
            buf.write(b);
chris@1
    90
          }
chris@1
    91
chris@1
    92
          buf.write(10);
chris@1
    93
        }
chris@1
    94
        
chris@1
    95
        return buf.toByteArray();
chris@1
    96
      }
chris@1
    97
      else
chris@1
    98
      {
chris@1
    99
        Log.msg("ArticleReader: " + line, false);
chris@1
   100
        return null;
chris@1
   101
      }
chris@1
   102
    }
chris@1
   103
    catch(IOException ex)
chris@1
   104
    {
chris@1
   105
      throw ex;
chris@1
   106
    }
chris@1
   107
    finally
chris@1
   108
    {
chris@1
   109
      this.out.write("QUIT\r\n".getBytes("UTF-8"));
chris@1
   110
      this.out.flush();
chris@1
   111
      this.out.close();
chris@1
   112
    }
chris@1
   113
  }
chris@1
   114
  
chris@1
   115
  private String readln(InputStream in)
chris@1
   116
    throws IOException
chris@1
   117
  {
chris@1
   118
    ByteArrayOutputStream buf = new ByteArrayOutputStream();
chris@1
   119
    for(int b = in.read(); b != 10 /* \n */; b = in.read())
chris@1
   120
    {
chris@1
   121
      buf.write(b);
chris@1
   122
    }
chris@1
   123
    
chris@1
   124
    return new String(buf.toByteArray());
chris@1
   125
  }
chris@1
   126
chris@1
   127
}