/*
 * Copyright (c) 1995 Regents of the University of California.
 * All rights reserved.
 *
 * This code is was contributed by Anastasio Andrea Scalisi
 * (scalisi@mailer.cefriel.it), of CEFRIEL, Milan, Italy.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *	This product includes software developed by the Computer Systems
 *	Engineering Group at Lawrence Berkeley Laboratory.
 * 4. Neither the name of the University nor of the Laboratory may be used
 *    to endorse or promote products derived from this software without
 *    specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#ifndef lint
static char rcsid[] =
    "@(#) $Header: /work/projects/tove/cvs/src/testing/vic-2.8-cqcam/vat_net-atm.cc,v 1.3 1998/12/07 11:23:38 uimonen Exp $ (LBL)";
#endif

#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <sys/uio.h>
#include "net.h"
#include <sys/socket.h>
#include <net/if.h>
#include <atm.h>
#include <iostream.h>


struct quality
{
  int pbat; /* peak bandwidth target */
  int pbam; /* peak bandwidth mean   */
  int mbat; /* mean bandwidth target */
  int mbam; /* mean bandwidth mean   */
  int mbut; /* burst lenght target   */ 
  int mbum; /* burst lenght mean     */
};

typedef const char * ATMnumber;
typedef const char * SAPnumber;


class ToveATM : public Network {
 public:
  ~ToveATM();
  virtual int command(int argc, const char*const* argv);
  void send(const msghdr& mh); 
protected:  
  int open(ATMnumber phoneno_, SAPnumber sap_, quality& qos_);
  int listen(ATMnumber phoneno_, SAPnumber sap_, quality& qos_);
  int close();

  virtual void dosend(u_char* buf, int len, int fd);
  virtual int dorecv(u_char* buf, int len, u_int32_t& from, int fd);
  int atmconnect(ATMnumber phoneno_, SAPnumber sap_, quality& qos_);
  int atmlisten(ATMnumber phoneno_, SAPnumber sap_, quality& qos_);
};


static class ATMNetworkMatcher : public Matcher {
public:
	ATMNetworkMatcher() : Matcher("network") {}
	TclObject* match(const char* id) {
		if (strcasecmp(id, "atm") == 0)
			return (new ToveATM);
		else
			return (0);
	}
} nm_atm;


int ToveATM :: command(int argc, const char*const* argv)
{
	Tcl& tcl = Tcl::instance();
	if (argc == 2) {
		if (strcmp(argv[1], "close") == 0) {
			close();
			return (TCL_OK);
		}
	} else if (argc == 10) {
		/*
		 * $net open dst sap pbat pbam mbat mbam mbut mbum
		 */
		if (strcmp(argv[1], "open") == 0) {
			quality qos;
			ATMnumber dstNumber = argv[2];
			SAPnumber sap = argv[3];
			qos.pbat = atoi(argv[4]);
			qos.pbam = atoi(argv[5]);
			qos.mbat = atoi(argv[6]);
			qos.mbam = atoi(argv[7]);
			qos.mbut = atoi(argv[8]);
			qos.mbum = atoi(argv[9]);
			
			if (open(dstNumber, sap, qos) < 0)
				tcl.result("0");
			else
				tcl.result("1");
			return (TCL_OK);
		}
		/*
		 * $net listen dst sap pbat pbam mbat mbam mbut mbum
		 */
		else if (strcmp(argv[1], "listen") == 0) {
			quality qos;
			ATMnumber dstNumber = argv[2];
			SAPnumber sap = argv[3];
			qos.pbat = atoi(argv[4]);
			qos.pbam = atoi(argv[5]);
			qos.mbat = atoi(argv[6]);
			qos.mbam = atoi(argv[7]);
			qos.mbut = atoi(argv[8]);
			qos.mbum = atoi(argv[9]);
			
			if (listen(dstNumber, sap, qos) < 0)
				tcl.result("0");
			else
				tcl.result("1");
			return (TCL_OK);
		}
        }
	return (Network::command(argc, argv));
}


int ToveATM :: open(ATMnumber phoneno_, SAPnumber sap_, quality& qos_)
{
  ssock_ = atmconnect(phoneno_, sap_, qos_);
  rsock_ = ssock_;
  return (ssock_);
}


int ToveATM :: listen(ATMnumber phoneno_, SAPnumber sap_, quality& qos_)
{
  ssock_ = atmlisten(phoneno_, sap_, qos_);
  rsock_ = ssock_;
  return (ssock_);
}


ToveATM :: ~ToveATM()
{
	close();
}


int ToveATM :: close()
{
	if (ssock_ >= 0) {
		::close(ssock_);
		ssock_ = -1;
	}
	return (0);
}


void ToveATM :: send(const msghdr& mh)
{
   /* Fore API has no scatter/gather hook.  Copy the packet. */
   int cc = cpmsg(mh);
   Network::send(wrkbuf_, cc);
}


void ToveATM :: dosend(u_char* buf, int len, int fd)
{
  int cc = ::send(fd, (char*)buf, len, 0);
  if (cc < 0) 
    {
      perror("vat: atm_send");
    }
}


int ToveATM :: dorecv(u_char* buf, int len, u_int32_t& from, int fd)
{
  from = 0;
  int cc = ::recv(fd, (char*)buf, len, 0);
  if (cc < 0) 
    {
      perror("vat: atm_recv");
      return (-1);
    }

  return (cc);
}


int ToveATM :: atmconnect(ATMnumber phoneno_, SAPnumber sap_, quality& qos_)
{
    struct sockaddr_atmsvc addr;
    struct atm_qos qos;
    struct atm_blli blli;

    int descriptor = socket(PF_ATMSVC, SOCK_DGRAM, 0);

    if (descriptor < 0)
    {
        cerr << "toveATM: Error in atmconnect" << endl;
        perror("socket");
        exit(1);
    }

    memset(&qos, 0, sizeof(qos));
    qos.aal = ATM_AAL5;
    qos.txtp.traffic_class = qos.rxtp.traffic_class = ATM_UBR;
    qos.txtp.max_sdu = qos.rxtp.max_sdu = 160;
    if (setsockopt(descriptor, SOL_ATM, SO_ATMQOS, &qos, sizeof(qos)) < 0)
    {
        cerr << "toveATM: Error in atmconnect" << endl;
        perror("setsockopt");
        exit(1);
    }

    memset(&addr, 0, sizeof(addr));
    addr.sas_family = AF_ATMSVC;

    if (text2atm(phoneno_, (struct sockaddr *)&addr, sizeof(addr), T2A_SVC) < 0)
    {
        cerr << "toveATM: Error in atmconnect" << endl;
        perror("text2atm");
        exit(1);
    }

    /* Define the SAP in the BLLI */
    memset(&blli, 0, sizeof(struct atm_blli));
    blli.l2_proto = ATM_L2_NONE;
    blli.l3_proto = ATM_L3_USER;
    blli.l3.user = (unsigned char) atoi(sap_);
    blli.next = 0;
//    addr.sas_addr.blli = &blli;
//    addr.sas_addr.bhli.hl_type = ATM_HL_NONE;

    cerr << "Calling addr " << phoneno_ << " SAP " << sap_ << endl;
    
    if (::connect(descriptor, (struct sockaddr *) &addr, sizeof(addr)) < 0)
    {
        cerr << "toveATM: Error in atmconnect" << endl;
        perror("connect");
        exit(1);
    }

    return descriptor;
}


int ToveATM :: atmlisten(ATMnumber phoneno_, SAPnumber sap_, quality& qos_)
{
    struct sockaddr_atmsvc addr;
    struct atm_qos qos;
    struct atm_blli blli;

    int descriptor = socket(PF_ATMSVC, SOCK_DGRAM, 0);
    if (descriptor < 0)
    {
        cerr << "toveATM: Error in atmlisten" << endl;
        perror("socket");
        exit(1);
    }

    memset(&qos, 0, sizeof(qos));
    qos.aal = ATM_AAL5;
    qos.txtp.traffic_class = ATM_ANYCLASS;
    qos.rxtp.traffic_class = qos.txtp.traffic_class;
    if (setsockopt(descriptor, SOL_ATM, SO_ATMQOS, &qos, sizeof(qos)) < 0)
    {
        cerr << "toveATM: Error in atmlisten" << endl;
        perror("setsockopt");
        exit(1);
    }

    memset(&addr, 0, sizeof(addr));
    addr.sas_family = AF_ATMSVC;

    /* Define the SAP in the BLLI */
    memset(&blli, 0, sizeof(blli));
    blli.l2_proto = ATM_L2_NONE;
    blli.l3_proto = ATM_L3_USER;
    blli.l3.user = (unsigned char) atoi(sap_);
    blli.next = 0;
//    addr.sas_addr.blli = &blli;
//    addr.sas_addr.bhli.hl_type = ATM_HL_NONE;

    if (::bind(descriptor, (struct sockaddr *) &addr, sizeof(addr)) < 0)
    {
        cerr << "toveATM: Error in atmlisten" << endl;
        perror("bind");
        exit(1);
    }
    
    if (::listen(descriptor, 1) < 0)
    {
        cerr << "toveATM: Error in atmlisten" << endl;
        perror("listen");
        exit(1);
    }
    
    int realDescriptor = 0;
    int addrlen;
    if ((realDescriptor =
        ::accept(descriptor, (struct sockaddr *) &addr, &addrlen)) < 0)
    {
        cerr << "toveATM: Error in atmlisten" << endl;
        perror("accept");
        exit(1);
    }
    
    if (::close(descriptor) < 0)
    {
        cerr << "toveATM: Error in atmlisten" << endl;
        perror("close (lsoc)");
        exit(1);
    }

    return realDescriptor;
}