//==========================================================================
//
//      ./agent/current/tests/snmpping.c
//
//
//==========================================================================
// ####ECOSGPLCOPYRIGHTBEGIN####                                            
// -------------------------------------------                              
// This file is part of eCos, the Embedded Configurable Operating System.   
// Copyright (C) 1998, 1999, 2000, 2001, 2002 Free Software Foundation, Inc.
//
// eCos is free software; you can redistribute it and/or modify it under    
// the terms of the GNU General Public License as published by the Free     
// Software Foundation; either version 2 or (at your option) any later      
// version.                                                                 
//
// eCos 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 General Public License    
// for more details.                                                        
//
// You should have received a copy of the GNU General Public License        
// along with eCos; if not, write to the Free Software Foundation, Inc.,    
// 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.            
//
// As a special exception, if other files instantiate templates or use      
// macros or inline functions from this file, or you compile this file      
// and link it with other works to produce a work based on this file,       
// this file does not by itself cause the resulting work to be covered by   
// the GNU General Public License. However the source code for this file    
// must still be made available in accordance with section (3) of the GNU   
// General Public License v2.                                               
//
// This exception does not invalidate any other reasons why a work based    
// on this file might be covered by the GNU General Public License.         
// -------------------------------------------                              
// ####ECOSGPLCOPYRIGHTEND####                                              
//####UCDSNMPCOPYRIGHTBEGIN####
//
// -------------------------------------------
//
// Portions of this software may have been derived from the UCD-SNMP
// project,  <http://ucd-snmp.ucdavis.edu/>  from the University of
// California at Davis, which was originally based on the Carnegie Mellon
// University SNMP implementation.  Portions of this software are therefore
// covered by the appropriate copyright disclaimers included herein.
//
// The release used was version 4.1.2 of May 2000.  "ucd-snmp-4.1.2"
// -------------------------------------------
//
//####UCDSNMPCOPYRIGHTEND####
//==========================================================================
//#####DESCRIPTIONBEGIN####
//
// Author(s):    hmt
// Contributors: hmt
// Date:         2000-05-30
// Purpose:      Port of UCD-SNMP distribution to eCos.
// Description:  
//              
//
//####DESCRIPTIONEND####
//
//==========================================================================
/********************************************************************
       Copyright 1989, 1991, 1992 by Carnegie Mellon University

			  Derivative Work -
Copyright 1996, 1998, 1999, 2000 The Regents of the University of California

			 All Rights Reserved

Permission to use, copy, modify and distribute this software and its
documentation for any purpose and without fee is hereby granted,
provided that the above copyright notice appears in all copies and
that both that copyright notice and this permission notice appear in
supporting documentation, and that the name of CMU and The Regents of
the University of California not be used in advertising or publicity
pertaining to distribution of the software without specific written
permission.

CMU AND THE REGENTS OF THE UNIVERSITY OF CALIFORNIA DISCLAIM ALL
WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL CMU OR
THE REGENTS OF THE UNIVERSITY OF CALIFORNIA BE LIABLE FOR ANY SPECIAL,
INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
FROM THE LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*********************************************************************/


// -------------------------------------------------------------------------
// Configuration of the test... now from CDL

// Do we test the interfaces in promiscuous mode?
//#define CYGSEM_SNMPAGENT_TESTS_PROMISCUOUS

// Do we make the test run forever?
//#define CYGNUM_SNMPAGENT_TESTS_ITERATIONS 999999

// Do we initialize SNMP v3 MIBs and authentication database?
//#define CYGSEM_SNMPAGENT_TESTS_SNMPv3

// ------------------------------------------------------------------------


// PING test code

#include <network.h>
#include <stdio.h>

#include <pkgconf/system.h>
#include <pkgconf/net.h>
#include <pkgconf/snmpagent.h>

#include <unistd.h>

#ifdef  CYGSEM_SNMPAGENT_TESTS_SNMPv3
#include <ucd-snmp/config.h>
#include <ucd-snmp/asn1.h>
#include <ucd-snmp/snmp_api.h>
#include <ucd-snmp/snmp_vars.h>

#include <ucd-snmp/snmpv3.h>
#include <ucd-snmp/usmUser.h>
#include <ucd-snmp/usmStats.h>
#include <ucd-snmp/snmpEngine.h>
#endif // CYGSEM_SNMPAGENT_TESTS_SNMPv3

#include <cyg/infra/testcase.h>

#ifdef CYGBLD_DEVS_ETH_DEVICE_H    // Get the device config if it exists
#include CYGBLD_DEVS_ETH_DEVICE_H  // May provide CYGTST_DEVS_ETH_TEST_NET_REALTIME
#endif

#ifdef CYGPKG_NET_TESTS_USE_RT_TEST_HARNESS // do we use the rt test?
# ifdef CYGTST_DEVS_ETH_TEST_NET_REALTIME // Get the test ancilla if it exists
#  include CYGTST_DEVS_ETH_TEST_NET_REALTIME
# endif
#endif

// Fill in the blanks if necessary
#ifndef TNR_OFF
# define TNR_OFF()
#endif
#ifndef TNR_ON
# define TNR_ON()
#endif
#ifndef TNR_INIT
# define TNR_INIT()
#endif
#ifndef TNR_PRINT_ACTIVITY
# define TNR_PRINT_ACTIVITY()
#endif

#define STACK_SIZE (CYGNUM_HAL_STACK_SIZE_TYPICAL + 0x1000)
static char stack[STACK_SIZE];
static cyg_thread thread_data;
static cyg_handle_t thread_handle;

#define NUM_PINGS 16
#define MAX_PACKET 4096
#define MIN_PACKET   64
#define MAX_SEND   4000

#define PACKET_ADD  ((MAX_SEND - MIN_PACKET)/NUM_PINGS)
#define nPACKET_ADD  1 

static unsigned char pkt1[MAX_PACKET], pkt2[MAX_PACKET];

#define UNIQUEID 0x1234

// Compute INET checksum
int
inet_cksum(u_short *addr, int len)
{
    register int nleft = len;
    register u_short *w = addr;
    register u_short answer;
    register u_int sum = 0;
    u_short odd_byte = 0;

    /*
     *  Our algorithm is simple, using a 32 bit accumulator (sum),
     *  we add sequential 16 bit words to it, and at the end, fold
     *  back all the carry bits from the top 16 bits into the lower
     *  16 bits.
     */
    while( nleft > 1 )  {
        sum += *w++;
        nleft -= 2;
    }

    /* mop up an odd byte, if necessary */
    if( nleft == 1 ) {
        *(u_char *)(&odd_byte) = *(u_char *)w;
        sum += odd_byte;
    }

    /*
     * add back carry outs from top 16 bits to low 16 bits
     */
    sum = (sum >> 16) + (sum & 0x0000ffff); /* add hi 16 to low 16 */
    sum += (sum >> 16);                     /* add carry */
    answer = ~sum;                          /* truncate to 16 bits */
    return (answer);
}

static int
show_icmp(unsigned char *pkt, int len, 
          struct sockaddr_in *from, struct sockaddr_in *to)
{
    cyg_tick_count_t *tp, tv;
    struct ip *ip;
    struct icmp *icmp;
    tv = cyg_current_time();
    ip = (struct ip *)pkt;
    if ((len < sizeof(*ip)) || ip->ip_v != IPVERSION) {
        diag_printf("%s: Short packet or not IP! - Len: %d, Version: %d\n", 
                    inet_ntoa(from->sin_addr), len, ip->ip_v);
        return 0;
    }
    icmp = (struct icmp *)(pkt + sizeof(*ip));
    len -= (sizeof(*ip) + 8);
    if ( 0 >= len ) {
        diag_printf("%s: Completely bogus short packet%s\n",
                    inet_ntoa(from->sin_addr), 0 == len ? "" : " [no ICMP header]");
        return 0;
    }
    tp = (cyg_tick_count_t *)&icmp->icmp_data;
    if (icmp->icmp_type != ICMP_ECHOREPLY) {
        diag_printf("%s: Invalid ICMP - type: %d, len (databytes): %d\n", 
                    inet_ntoa(from->sin_addr), icmp->icmp_type, len);
        return 0;
    }
    if (icmp->icmp_id != UNIQUEID) {
        diag_printf("%s: ICMP received for wrong id - sent: %x, recvd: %x\n", 
                    inet_ntoa(from->sin_addr), UNIQUEID, icmp->icmp_id);
    }
    diag_printf("%d bytes from %s: ", len, inet_ntoa(from->sin_addr));
    diag_printf("icmp_seq=%d", icmp->icmp_seq);
    diag_printf(", time=%dms\n", (int)(tv - *tp)*10);
    return (from->sin_addr.s_addr == to->sin_addr.s_addr);
}

static void
ping_host(int s, struct sockaddr_in *host)
{
    struct icmp *icmp = (struct icmp *)pkt1;
    int icmp_len = MIN_PACKET;
    int seq, ok_recv, bogus_recv;
    cyg_tick_count_t *tp;
    long *dp;
    struct sockaddr_in from;
    int i, len, fromlen;

    ok_recv = 0;
    bogus_recv = 0;
    TNR_OFF();
    diag_printf("PING server %s\n", inet_ntoa(host->sin_addr));
    for (seq = 0;  seq < NUM_PINGS;  seq++, icmp_len += PACKET_ADD ) {
        TNR_ON();
        // Build ICMP packet
        icmp->icmp_type = ICMP_ECHO;
        icmp->icmp_code = 0;
        icmp->icmp_cksum = 0;
        icmp->icmp_seq = seq;
        icmp->icmp_id = 0x1234;
        // Set up ping data
        tp = (cyg_tick_count_t *)&icmp->icmp_data;
        *tp++ = cyg_current_time();
        dp = (long *)tp;
        for (i = sizeof(*tp);  i < icmp_len;  i += sizeof(*dp)) {
            *dp++ = i;
        }
        // Add checksum
        icmp->icmp_cksum = inet_cksum( (u_short *)icmp, icmp_len+8);
        // Send it off
        if (sendto(s, icmp, icmp_len+8, 0, (struct sockaddr *)host, sizeof(*host)) < 0) {
            TNR_OFF();
            perror("sendto");
            continue;
        }
        // Wait for a response
        fromlen = sizeof(from);
        len = recvfrom(s, pkt2, sizeof(pkt2), 0, (struct sockaddr *)&from, &fromlen);
        TNR_OFF();
        if (len < 0) {
            perror("recvfrom");
            icmp_len = MIN_PACKET - PACKET_ADD; // just in case - long routes
        } else {
            if (show_icmp(pkt2, len, &from, host)) {
                ok_recv++;
            } else {
                bogus_recv++;
            }
        }
    }
    TNR_OFF();
    diag_printf("Sent %d packets, received %d OK, %d bad\n", NUM_PINGS, ok_recv, bogus_recv);
}

static void
ping_test(struct bootp *bp)
{
    struct protoent *p;
    struct timeval tv;
    struct sockaddr_in host;
    int s;

    if ((p = getprotobyname("icmp")) == (struct protoent *)0) {
        perror("getprotobyname");
        return;
    }
    s = socket(AF_INET, SOCK_RAW, p->p_proto);
    if (s < 0) {
        perror("socket");
        return;
    }
    tv.tv_sec = 1;
    tv.tv_usec = 0;
    setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
    // Set up host address
    host.sin_family = AF_INET;
    host.sin_len = sizeof(host);
    host.sin_addr = bp->bp_siaddr;
    host.sin_port = 0;
    ping_host(s, &host);
    // Now try a bogus host
    // (also, map 76 <-> 191 so that if a pair runs they ping each other)
    host.sin_addr = bp->bp_yiaddr; // *my* address.
//    host.sin_addr.s_addr = htonl(ntohl(host.sin_addr.s_addr) ^ 0xf3);
    host.sin_addr.s_addr = htonl(ntohl(host.sin_addr.s_addr) ^ 2);
    ping_host(s, &host);
    close(s);
}

static void
ping_test_loopback( int lo )
{
    struct protoent *p;
    struct timeval tv;
    struct sockaddr_in host;
    int s;

    if ((p = getprotobyname("icmp")) == (struct protoent *)0) {
        perror("getprotobyname");
        return;
    }
    s = socket(AF_INET, SOCK_RAW, p->p_proto);
    if (s < 0) {
        perror("socket");
        return;
    }
    tv.tv_sec = 1;
    tv.tv_usec = 0;
    setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
    // Set up host address
    host.sin_family = AF_INET;
    host.sin_len = sizeof(host);
    host.sin_addr.s_addr = htonl(INADDR_LOOPBACK + (0x100 * lo));
    host.sin_port = 0;
    ping_host(s, &host);
    // Now try a bogus host
    host.sin_addr.s_addr = htonl(ntohl(host.sin_addr.s_addr) + 32);
    ping_host(s, &host);
    close(s);
}

#ifdef CYGSEM_SNMPAGENT_TESTS_PROMISCUOUS
static void 
interface_promisc(const char *intf)
{
  struct ifreq ifr;
  int s;

  s = socket(AF_INET, SOCK_DGRAM, 0);
  if (s < 0) {
    perror("socket");
    return;
  }

  strcpy(ifr.ifr_name, intf);
  ifr.ifr_flags = IFF_UP | IFF_BROADCAST | IFF_RUNNING | IFF_PROMISC;
  if (ioctl(s, SIOCSIFFLAGS, &ifr)) {
    perror("SIOCSIFFLAGS");
  }
  close(s);
}
#endif // CYGSEM_SNMPAGENT_TESTS_PROMISCUOUS

void snmp_do_reinit( void )
{
    diag_printf( "SNMP re-init function\n" );
#ifdef  CYGSEM_SNMPAGENT_TESTS_SNMPv3
    // Initialisation for USM is now invoked from mib_module_inits.h
    //init_usmUser();             /* MIBs to support SNMPv3             */
    //init_usmStats();
    //init_snmpEngine();
    usm_parse_create_usmUser(NULL, "root MD5 md5passwd DES DESpasswd");
#endif //  CYGSEM_SNMPAGENT_TESTS_SNMPv3
}

int create_snmpd_conf (void) {
#ifdef CYGSEM_SNMPAGENT_TESTS_SNMPv3
#ifdef CYGPKG_SNMPLIB_FILESYSTEM_SUPPORT 
#ifdef CYGPKG_FS_RAM
  int c;
  FILE *fd;

  diag_printf ("\nStarting creation of snmpd.conf\n");

  /* Mount RAM-FS */
  if (mount ("", "/", "ramfs") != 0) {
    diag_printf ("File system mount failed; errno=%d \n", errno);
    return -1;
  }
  
  if (mkdir ("/etc", 0) != 0) {
    diag_printf ("mkdir (etc) failed;  errno=%d\n", errno);
    return -1;
  } 

  if (chdir ("/etc") != 0) {
    diag_printf ("... Change-dir (etc) failed; errno=%d\n", errno);
    return -1;
  } else { 
    diag_printf ("chdir-etc done\n");
  }

  if (mkdir ("snmp", 0) != 0) {
    diag_printf ("mkdir failed (snmp);  errno=%d\n", errno);
    return -1;
  } else {
    diag_printf ("mkdir-snmp done\n");
  }

  if (chdir ("snmp") != 0) {
    diag_printf ("... Change-dir (snmp) failed; errno=%d\n", errno);
    return -1;
  } else { 
    diag_printf ("... Change-dir (snmp) done \n");
  }

  /* Open File & Write to it  */
  if ((fd = fopen( "snmpd.conf", "w" )) == NULL) {
    diag_printf ("fopen failed\n");
    return -1;
  }

  fprintf (fd, "#        sec.name     source       community\n");
  fprintf (fd, "com2sec   public     default       crux\n");
  fprintf (fd, "com2sec   root       default       crux\n");
  fprintf (fd, "#                 sec.model   sec.name\n");
  fprintf (fd, "group     public    v1        public\n");
  fprintf (fd, "group     public    v2c       public\n");
  fprintf (fd, "group     public    usm       root\n");
  fprintf (fd, "view      all  included  .1\n");
  fprintf (fd, "access    public    \"\"   any  noauth    exact     all  none none\n");
  fprintf (fd, "\n\n");

  if (fclose (fd)) {
    diag_printf ("fclose failed\n");
    return -1;
  }

  /* Read Back */
  fd = fopen( "/etc/snmp/snmpd.conf", "r" );
  if (fd == NULL) { 
    diag_printf ("fopen failed\n");
    return -1;
  }

  while ((c=fgetc (fd)) != EOF) {
    diag_printf ("%c", c);
  }

  if (fclose (fd))  {
    diag_printf ("fclose failed\n");
    return -1;
  }
#endif
#endif
#endif
  return 0;
}

void
net_test(cyg_addrword_t p)
{
    int i = CYGNUM_SNMPAGENT_TESTS_ITERATIONS;
    int ieth0_up = 0, ieth1_up = 0;
    int j;

    diag_printf("Start PING test\n");
    TNR_INIT();
    init_all_network_interfaces();
#ifdef CYGHWR_NET_DRIVER_ETH0
    ieth0_up = eth0_up;
#endif
#ifdef CYGHWR_NET_DRIVER_ETH1
    ieth1_up = eth1_up;
#endif
#ifdef CYGSEM_SNMPAGENT_TESTS_PROMISCUOUS
#ifdef CYGHWR_NET_DRIVER_ETH0
        if (eth0_up)
            interface_promisc("eth0");
#endif
#ifdef CYGHWR_NET_DRIVER_ETH1
        if (eth1_up)
            interface_promisc("eth1");
#endif
#endif // CYGSEM_SNMPAGENT_TESTS_PROMISCUOUS
    {
        extern void cyg_net_snmp_init(void);
        extern void (*snmpd_reinit_function)( void );

        snmpd_reinit_function = snmp_do_reinit;

        if (create_snmpd_conf ()) {
          CYG_TEST_FAIL_EXIT("create_snmpd_conf() error\n");
        }
        cyg_net_snmp_init();
    }
    do {
        TNR_ON();
#ifdef CYGHWR_NET_DRIVER_ETH0
        if (eth0_up) {
            ping_test(&eth0_bootp_data);
            cyg_thread_delay(500);
        }
#endif
#if NLOOP > 0
        for ( j = 0; j < NLOOP; j++ ) {
            ping_test_loopback( j );
            cyg_thread_delay(500);
        }
#endif
#ifdef CYGHWR_NET_DRIVER_ETH1
        if (eth1_up) {
            ping_test(&eth1_bootp_data);
            cyg_thread_delay(500);
        }
#endif
        TNR_OFF();
        TNR_PRINT_ACTIVITY();

        // If an interface has gone down eg. due to DHCP timing out,
        // re-initialize everything:
        if (0
#ifdef CYGHWR_NET_DRIVER_ETH0
            || ieth0_up != eth0_up
#endif
#ifdef CYGHWR_NET_DRIVER_ETH1
            || ieth1_up != eth1_up 
#endif
            ) {
            diag_printf( "Re-initializing the world: eth0 %d/%d eth1 %d/%d!\n",
                         ieth0_up, 0
#ifdef CYGHWR_NET_DRIVER_ETH0
                         | eth0_up
#endif
                         , ieth1_up, 0
#ifdef CYGHWR_NET_DRIVER_ETH1
                         | eth1_up
#endif
                );
            init_all_network_interfaces();
            for ( j = 0; j < CYGPKG_NET_NLOOP; j++ )
                init_loopback_interface( j );

            diag_printf( "Re-initialized the world: eth0 %d/%d eth1 %d/%d!\n",
                         ieth0_up, 0
#ifdef CYGHWR_NET_DRIVER_ETH0
                         | eth0_up
#endif
                         , ieth1_up, 0
#ifdef CYGHWR_NET_DRIVER_ETH1
                         | eth1_up
#endif
                );
        }
    } while ( i-- > 0 );
    CYG_TEST_PASS_FINISH( "Done pinging while SNMP looks on" );
}

void
cyg_start(void)
{
    // Create a main thread, so we can run the scheduler and have time 'pass'
    cyg_thread_create(10,                // Priority - just a number
                      net_test,          // entry
                      0,                 // entry parameter
                      "Network test",    // Name
                      &stack[0],         // Stack
                      STACK_SIZE,        // Size
                      &thread_handle,    // Handle
                      &thread_data       // Thread data structure
            );
    cyg_thread_resume(thread_handle);  // Start it
    cyg_scheduler_start();
}

// EOF snmpping.c
