net_buf.h

/* -*- Mode: C++; c-basic-offset: 2; tab-width: 2; indent-tabs-mode: nil -*-
 * 
 * Quadra, an action puzzle game
 * Copyright (C) 1998-2000  Ludus Design
 * 
 * This program 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 2.1 of the License, or (at your option) any later version.
 * 
 * This program 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 this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#ifndef _HEADER_NET_BUF
#define _HEADER_NET_BUF

#include <stdint.h>
#include <string.h>

#include "types.h"
#include "net.h"

class Net_buf {
public:
	uint8_t *point;
	Net_connection *from;
	uint32_t from_addr;
	uint8_t buf[NETBUF_SIZE];
	void write_dword(uint32_t v) {
    *point++ = (v >> 24) & 0xff;
    *point++ = (v >> 16) & 0xff;
    *point++ = (v >> 8)  & 0xff;
    *point++ = v & 0xff;
	}
	void write_word(uint16_t v) {
    *point++ = (v >> 8)  & 0xff;
    *point++ = v & 0xff;
	}
	void write_byte(uint8_t v) {
		*(uint8_t *) point = v;
		point += sizeof(uint8_t);
	}
	void write_bool(bool b) {
		write_byte(b? 1:0);
	}
	void write_mem(const void *v, int num) {
		memcpy(point, v, num);
		point += num;
	}
	void write_string(const char *v) {
		write_mem(v, strlen(v)+1); // write a string with its '0'
	}
	uint32_t read_dword() {
		if(len() <= NETBUF_SIZE-sizeof(uint32_t)) {
      uint32_t ret;
      ret = *point << 24; point++;
      ret |= *point << 16; point++;
      ret |= *point << 8; point++;
      ret |= *point; point++;
      return ret;
		}
		else
			return 0;
	}
	uint16_t read_word() {
		if(len() <= NETBUF_SIZE-sizeof(uint16_t)) {
      uint32_t ret;
      ret = *point << 8; point++;
      ret |= *point; point++;
			return ret;
		}
		else
			return 0;
	}
	uint8_t read_byte() {
		if(len() <= NETBUF_SIZE-sizeof(uint8_t)) {
			uint8_t ret = *(uint8_t *) point;
			point += sizeof(uint8_t);
			return ret;
		}
		else
			return 0;
	}
	bool read_bool() {
		if(read_byte())
			return true;
		else
			return false;
	}
	void read_mem(void *v, int num) {
		if(len() <= NETBUF_SIZE-num) {
			memcpy(v, point, num);
			point += num;
		}
		else
			memset(v, 0, num);
	}
	bool read_string(char *v, int size) { // read a string with its '0'
		do {
			char c=(char)read_byte();
			if(c>0 && c<' ')
				c=' ';
			*v=c;
			if(!*v)
				return true;
			v++;
			size--;
		} while(size);
		if(v[-1]) {
			v[-1]=0;
			return false;
		}
		return true;
	}
	Net_buf() {
		reset();
		from=NULL;
		memset(buf, 0, sizeof(buf));
	}
	void reset() {
		point = buf;
	}
	const unsigned int len() const {
		return point - buf;
	}
};

#endif