/** @Name: COUNTER.c @Purpose: @User: Fill in purpose Message Types: @User: Document messages (requests and replies) here Op Short Descr Request Reply ----------------------------------------------------------- 0 Hello [0] instance# instance# [1] command = 0 command = 0 [2] flags [3] seq [4] *len Notation: - this: struct rp_instance * (ptr to plugin instance) Post-Conditions: COUNTER_load - Plugin class (COUNTER_class) structure is initialized - COUNTER_get_class() is registered with PCU COUNTER_create_instance - Plugin instance memory is allocated and initialized (base class and its extension): this->rpclass = &COUNTER_class ... other members of base class ... COUNTER_bind_instance COUNTER_handle_packet COUNTER_handle_message COUNTER_unbind_instance COUNTER_free_instance - Plugin instance memory is free COUNTER_unload - All instances are free - Class is not registered with PCU */ #include "stdinc.h" #include "COUNTER.h" MOD_MISC("COUNTER") // Static plugin class structure. // Each kernel module has one class structure that is initialized by // the plugin and a pointer to it is returned. The class structure can then // be used to create plugin instances. Note that this is a structure and // not a pointer. static struct rp_class COUNTER_class; // @User: Add class-wide global variables here void COUNTER_init_class() { // initialize plugin class // // Initialize class struct: // classid - user defined // itype - RP_INTERFACE_TYPE_PKT - for read-only plugins. The handle packet // signature is // RP_INTERFACE_TYPE_HLIST for plugins that will modify pkts or want // to source or sink data. The signature is // // create_instance = function pointer to your create instance method. COUNTER_class.classid = COUNTER_ID; COUNTER_class.itype = RP_INTERFACE_TYPE_HLIST; COUNTER_class.create_instance = COUNTER_create_instance; return; } // --| create and initialize an instance of COUNTER |-- struct rp_instance * COUNTER_create_instance( struct rp_class *myclass, // points to class structure u_int32_t instanceid) // new instance identifier { struct COUNTER_instance *myinst; // allocate memory for local instance struct MSR_PLUGIN_MALLOC(myinst,struct COUNTER_instance *, sizeof(struct COUNTER_instance), M_MSR, M_WAITOK); if (myinst == NULL) return NULL; // fill in instance pointers to local methods myinst->rootinstance.rpclass = &COUNTER_class; myinst->rootinstance.handle_packet = COUNTER_handle_packet; myinst->rootinstance.free_instance = COUNTER_free_instance; myinst->rootinstance.bind_instance = COUNTER_bind_instance; myinst->rootinstance.unbind_instance = COUNTER_unbind_instance; myinst->rootinstance.handle_msg = COUNTER_handle_msg; myinst->rootinstance.instanceid = instanceid; // @User: Initialize instance variables here myinst->pkt_count = 0; // initialization code return (struct rp_instance *)myinst; } void COUNTER_handle_packet( struct rp_instance *this, // points to instance structure void *bufferList) // points to list of pkt buffers { struct COUNTER_instance *inst = (struct COUNTER_instance *)this; MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_INFO, "COUNTER_handle_packet: Instance ID = %d got pkt %d\n", this->instanceid, inst->pkt_count)); // @User: Add code to handle packets here inst->pkt_count++; } /* @Purpose: Handle control messages. The request buffer 'buf' is 44 Bytes (11 ints) long. The same array will be used for the reply msg (i.e., request buffer is reused for the reply buffer and is overwritten). Request Buffer (1 word = 1 int): Word 0 Instance number of receiver Word 1 Command (Operation Code) Word 2 Arg 0 (if any) ... Word 10 Arg 8 (if any) Note 1: By convention, command 0 is the 'hello' command which returns the plugin id and the instance number. All other commands are at the discretion of the user. Note 2: The word preceding buf[0] contains the message header and contains the values of 'flags', 'seq' and '*len' which are passed to this routine. @User: Append request codes and reply args to the list below. Request Codes: Reply (#: args): 0: Hello 2: COUNTER_ID, id *: Hello 2: COUNTER_ID, id */ int COUNTER_handle_msg( struct rp_instance *this, // points to instance structure void *buf, // points to request/reply buffer u_int8_t flags, // flags u_int8_t seq, // sequence number u_int8_t *len) // IN: request is in buf[0] thru buf[*len/4-1] // OUT: reply is in buf[0] thru buf[*len/4-1] { struct COUNTER_instance *inst = (struct COUNTER_instance *)this; u_int32_t *vals = (u_int32_t *)buf; u_int32_t id = (u_int32_t)ntohl(*vals); u_int32_t cmnd = (u_int32_t)ntohl(*(vals + 1)); struct msr_bufhdr_t *hdr; *vals = (u_int32_t)htonl(COUNTER_ID); *(vals + 1) = (u_int32_t)htonl(id); switch (cmnd) { case 0: // Hello *len = 2 * sizeof(u_int32_t); break; default: // return pkt count *vals = (u_int32_t)htonl(inst->pkt_count); // pkt count *len = 1 * sizeof(u_int32_t); // #bytes in buf break; } return 0; } // --| Free instance structure |-- void COUNTER_free_instance( struct rp_instance *this) // points to instance structure { if (this) { MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_INFO, "COUNTER_free_instance: Freeing instance id %d ((class id %d)\n", this->instanceid, this->rpclass->classid)); MSR_PLUGIN_FREE(this, M_MSR); } else { MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_WARNING, "COUNTER_free_instance: Passing a NULL this pointer\n")); } } // --| Bind plugin instance to filter |-- void COUNTER_bind_instance( struct rp_instance *this) // points to instance structure { struct COUNTER_instance *inst = (struct COUNTER_instance *)this; MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_INFO, "COUNTER_bind_instance: Binding instance id %d (class id %d, type %d) to filter id %d\n", this->instanceid, this->rpclass->classid, this->rpclass->itype, this->bound_fid)); // @User: Define what occurs when binding your plugin here } // --| Unbind plugin instance from filter void COUNTER_unbind_instance( struct rp_instance *this) // points to instance structure { struct COUNTER_instance *inst = (struct COUNTER_instance *)this; MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_INFO, "COUNTER_unbind_instance: Unbinding instance id %d (class id %d, Type %d) to filter id %d\n", this->instanceid, this->rpclass->classid, this->rpclass->itype, this->bound_fid)); // @User: Define what occurs when unbinding your plugin here inst->pkt_count = 0; } /** ---------------------------------------------------------------- @User: YOU SHOULD NOT NEED TO MAKE ANY CHANGES TO THE REST OF THIS FILE ---------------------------------------------------------------- */ struct rp_class * COUNTER_get_class() { // return class structure return &COUNTER_class; } // External kernel module entry point. // // This function must match the name of the .o file. // It is called each time the module is loaded or unloaded. // The stat information is not needed here, so we will // leave it lkm_nofunc(). // int COUNTER ( struct lkm_table *lkmtp, // points to loadable kernel module table int cmd, // command: load or unload int ver, // version struct kernel_plugin_fct_struct *fctPtr) // points to table of kernel functions { // Do NOT put anything before the kernel_plugin_fcts is set, especially if // it uses any of the MSR_ macros!!! if (kernel_plugin_fcts == NULL) { kernel_plugin_fcts = fctPtr; kernel_plugin_variables = fctPtr->pluginVariables; } MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_VERBOSE, "COUNTER: Entry function -- initialized plugin_fcts pointers\n")); MSR_PLUGIN_DISPATCH(lkmtp, cmd, ver, COUNTER_load, COUNTER_unload, PLUGIN_LKM_NOFUNC_FCT); } // This function is called each time the module is loaded int COUNTER_load( struct lkm_table *lkmtp, // points to loadable kernel module table int cmd) // command (should be load) { int err; MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_VERBOSE, "COUNTER_load: Loading COUNTER ...\n")); if (PLUGIN_LKM_EXISTS_FCT(lkmtp)) { // avoid loading twice MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_INFO, "COUNTER_load: plugin already exists! Returing EEXIST\n")); return (EEXIST); } COUNTER_init_class(); err = PLUGIN_PCU_REGISTER_CLASS_FCT(COUNTER_get_class()); if (err != RP_OK) { MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_ERROR, "COUNTER_load: Error (err = %d) encountered registering class.\n", err)); return -1; } return 0; } // This function is called each time the module is unloaded. // Remove all existing instances and then remove class. int COUNTER_unload( struct lkm_table *lkmtp, // points to loadable kernel module table int cmd) // command (should beunload) { struct rp_class *rpclass; u_int32_t cid; rpclass = COUNTER_get_class(); cid = rpclass->classid; MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_VERBOSE, "COUNTER_unload: my rpclass id = %d, Struct Address = 0x%08x\n", cid, (u_int32_t)rpclass)); if (PLUGIN_PCU_FREE_ALL_INSTANCES_FCT(rpclass) != RP_OK) { MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_ERROR, "COUNTER_unload: Error freeing all instances for class %d\n", cid)); return -1; } if (PLUGIN_PCU_DEREGISTER_CLASS_FCT(rpclass) != RP_OK) { MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_ERROR, "COUNTER_unload: Error deregistering class %d\n", cid)); } MSR_DEBUG((MSR_DEBUG_PLUGIN | MSR_DEBUG_LEVEL_VERBOSE, "COUNTER_unload: Unloaded class %d\n", cid)); return 0; }