panda3d/panda/src/downloader/downloader.h

// Filename: downloader.h
// Created by:  mike (09Jan97)
//
////////////////////////////////////////////////////////////////////
//
#ifndef DOWNLOADER_H
#define DOWNLOADER_H
//
////////////////////////////////////////////////////////////////////
// Includes
////////////////////////////////////////////////////////////////////
#include <pandabase.h>
#include <notify.h>
#include <filename.h>
#include <buffer.h>
#include <pointerTo.h>
#include <clockObject.h>

#if defined(WIN32_VC)
  #include <winsock2.h>
#else
  #include <netinet/in.h>  // Irix seems to require this one for resolv.h.
  #include <sys/types.h>
  #include <sys/socket.h>
  #include <resolv.h>
#endif

////////////////////////////////////////////////////////////////////
//       Class : Downloader
// Description :
////////////////////////////////////////////////////////////////////
class EXPCL_PANDAEXPRESS Downloader {
PUBLISHED:
  enum DownloadCode {
    DL_eof = 5,
    DL_network_no_data = 4,

    DL_ok = 3,
    DL_write = 2,
    DL_success = 1,

    // General download errors
    DL_error = -1,
    DL_error_abort = -2,

    // General network errors
    DL_error_network_dead = -30,
    DL_error_network_unreachable = -31,
    DL_error_network_disconnected = -32,
    DL_error_network_timeout = -33,
    DL_error_network_no_data = -34,

    // Local network errors
    DL_error_network_disconnected_locally = -40,
    DL_error_network_buffer_overflow = -41,
    DL_error_network_disk_quota_exceeded = -42,

    // Remote host network errors
    DL_error_network_remote_host_disconnected = -50,
    DL_error_network_remote_host_down = -51,
    DL_error_network_remote_host_unreachable = -52,
    DL_error_network_remote_host_not_found = -53,
    DL_error_network_remote_host_no_response = -54,

    // General local errors
    DL_error_write = -60,

    // HTTP errors
    DL_error_http_server_timeout = -70,
    DL_error_http_gateway_timeout = -71,
    DL_error_http_service_unavailable = -72,
  };

  Downloader(void);
  virtual ~Downloader(void);

  int connect_to_server(const string &name, uint port=80);
  void disconnect_from_server(void);

  int initiate(const string &file_name, Filename file_dest);
  int initiate(const string &file_name, Filename file_dest,
		int first_byte, int last_byte, int total_bytes,
		bool partial_content = true);
  int run(void);

  INLINE void set_frequency(float frequency);
  INLINE float get_frequency(void) const;
  INLINE void set_byte_rate(float bytes);
  INLINE float get_byte_rate(void) const;
  INLINE void set_disk_write_frequency(int frequency);
  INLINE int get_disk_write_frequency(void) const;
  INLINE int get_bytes_written(void) const;
  INLINE float get_bytes_per_second(void) const;

private:
  class DownloadStatus {
  public:
    DownloadStatus(char *buffer, int first_byte, int last_byte,
			int total_bytes, bool partial_content);
    void reset(void);

  public:
    bool _first_line_complete;
    bool _header_is_complete;
    bool _header_is_valid;
    char *_start;
    char *_next_in;
    int _bytes_in_buffer;
    int _total_bytes_written;
    int _first_byte;
    int _last_byte;
    int _total_bytes;
    bool _partial_content;
    char *_buffer;
  };

  INLINE void recompute_buffer(void);

  int connect_to_server(void);
  int safe_send(int socket, const char *data, int length, long timeout);
  int fast_receive(int socket, DownloadStatus *status, int rec_size);
  int parse_http_response(const string &resp);
  int parse_header(DownloadStatus *status);
  int write_to_disk(DownloadStatus *status);

  void cleanup(void);
  char *handle_socket_error(void) const;

  int get_network_error(void) const;

private:
  bool _connected;
  int _socket;
  string _server_name;
  struct sockaddr_in _sin;
  bool _TCP_stack_initialized;

  bool _initiated;
  bool _ever_initiated;
  PT(Buffer) _buffer;
  int _disk_write_frequency;
  float _frequency;
  float _byte_rate; 
  ulong _receive_size;
  int _disk_buffer_size;
  ofstream _dest_stream;
  bool _recompute_buffer;

  DownloadStatus *_current_status;
  bool _got_any_data;

  double _tlast;
  double _tfirst;
  ClockObject _clock;
};

#include "downloader.I"

#endif