261 lines
		
	
	
	
		
			9.3 KiB
			
		
	
	
	
		
			C
		
	
	
	
	
	
		
		
			
		
	
	
			261 lines
		
	
	
	
		
			9.3 KiB
			
		
	
	
	
		
			C
		
	
	
	
	
	
|   | /******************************************************************************
 | ||
|  |  * ring.h | ||
|  |  * | ||
|  |  * Shared producer-consumer ring macros. | ||
|  |  * | ||
|  |  * Tim Deegan and Andrew Warfield November 2004. | ||
|  |  */ | ||
|  | 
 | ||
|  | #ifndef __XEN_PUBLIC_IO_RING_H__
 | ||
|  | #define __XEN_PUBLIC_IO_RING_H__
 | ||
|  | 
 | ||
|  | typedef unsigned int RING_IDX; | ||
|  | 
 | ||
|  | /* Round a 32-bit unsigned constant down to the nearest power of two. */ | ||
|  | #define __RD2(_x)  (((_x) & 0x00000002) ? 0x2		       : ((_x) & 0x1))
 | ||
|  | #define __RD4(_x)  (((_x) & 0x0000000c) ? __RD2((_x)>>2)<<2    : __RD2(_x))
 | ||
|  | #define __RD8(_x)  (((_x) & 0x000000f0) ? __RD4((_x)>>4)<<4    : __RD4(_x))
 | ||
|  | #define __RD16(_x) (((_x) & 0x0000ff00) ? __RD8((_x)>>8)<<8    : __RD8(_x))
 | ||
|  | #define __RD32(_x) (((_x) & 0xffff0000) ? __RD16((_x)>>16)<<16 : __RD16(_x))
 | ||
|  | 
 | ||
|  | /*
 | ||
|  |  * Calculate size of a shared ring, given the total available space for the | ||
|  |  * ring and indexes (_sz), and the name tag of the request/response structure. | ||
|  |  * A ring contains as many entries as will fit, rounded down to the nearest | ||
|  |  * power of two (so we can mask with (size-1) to loop around). | ||
|  |  */ | ||
|  | #define __RING_SIZE(_s, _sz) \
 | ||
|  |     (__RD32(((_sz) - (long)&(_s)->ring + (long)(_s)) / sizeof((_s)->ring[0]))) | ||
|  | 
 | ||
|  | /*
 | ||
|  |  * Macros to make the correct C datatypes for a new kind of ring. | ||
|  |  * | ||
|  |  * To make a new ring datatype, you need to have two message structures, | ||
|  |  * let's say struct request, and struct response already defined. | ||
|  |  * | ||
|  |  * In a header where you want the ring datatype declared, you then do: | ||
|  |  * | ||
|  |  *     DEFINE_RING_TYPES(mytag, struct request, struct response); | ||
|  |  * | ||
|  |  * These expand out to give you a set of types, as you can see below. | ||
|  |  * The most important of these are: | ||
|  |  * | ||
|  |  *     struct mytag_sring      - The shared ring. | ||
|  |  *     struct mytag_front_ring - The 'front' half of the ring. | ||
|  |  *     struct mytag_back_ring  - The 'back' half of the ring. | ||
|  |  * | ||
|  |  * To initialize a ring in your code you need to know the location and size | ||
|  |  * of the shared memory area (PAGE_SIZE, for instance). To initialise | ||
|  |  * the front half: | ||
|  |  * | ||
|  |  *     struct mytag_front_ring front_ring; | ||
|  |  *     SHARED_RING_INIT((struct mytag_sring *)shared_page); | ||
|  |  *     FRONT_RING_INIT(&front_ring, (struct mytag_sring *)shared_page, | ||
|  |  *		       PAGE_SIZE); | ||
|  |  * | ||
|  |  * Initializing the back follows similarly (note that only the front | ||
|  |  * initializes the shared ring): | ||
|  |  * | ||
|  |  *     struct mytag_back_ring back_ring; | ||
|  |  *     BACK_RING_INIT(&back_ring, (struct mytag_sring *)shared_page, | ||
|  |  *		      PAGE_SIZE); | ||
|  |  */ | ||
|  | 
 | ||
|  | #define DEFINE_RING_TYPES(__name, __req_t, __rsp_t)			\
 | ||
|  | 									\ | ||
|  | /* Shared ring entry */							\ | ||
|  | union __name##_sring_entry {						\ | ||
|  |     __req_t req;							\ | ||
|  |     __rsp_t rsp;							\ | ||
|  | };									\ | ||
|  | 									\ | ||
|  | /* Shared ring page */							\ | ||
|  | struct __name##_sring {							\ | ||
|  |     RING_IDX req_prod, req_event;					\ | ||
|  |     RING_IDX rsp_prod, rsp_event;					\ | ||
|  |     uint8_t  pad[48];							\ | ||
|  |     union __name##_sring_entry ring[1]; /* variable-length */		\ | ||
|  | };									\ | ||
|  | 									\ | ||
|  | /* "Front" end's private variables */					\ | ||
|  | struct __name##_front_ring {						\ | ||
|  |     RING_IDX req_prod_pvt;						\ | ||
|  |     RING_IDX rsp_cons;							\ | ||
|  |     unsigned int nr_ents;						\ | ||
|  |     struct __name##_sring *sring;					\ | ||
|  | };									\ | ||
|  | 									\ | ||
|  | /* "Back" end's private variables */					\ | ||
|  | struct __name##_back_ring {						\ | ||
|  |     RING_IDX rsp_prod_pvt;						\ | ||
|  |     RING_IDX req_cons;							\ | ||
|  |     unsigned int nr_ents;						\ | ||
|  |     struct __name##_sring *sring;					\ | ||
|  | }; | ||
|  | 
 | ||
|  | /*
 | ||
|  |  * Macros for manipulating rings. | ||
|  |  * | ||
|  |  * FRONT_RING_whatever works on the "front end" of a ring: here | ||
|  |  * requests are pushed on to the ring and responses taken off it. | ||
|  |  * | ||
|  |  * BACK_RING_whatever works on the "back end" of a ring: here | ||
|  |  * requests are taken off the ring and responses put on. | ||
|  |  * | ||
|  |  * N.B. these macros do NO INTERLOCKS OR FLOW CONTROL. | ||
|  |  * This is OK in 1-for-1 request-response situations where the | ||
|  |  * requestor (front end) never has more than RING_SIZE()-1 | ||
|  |  * outstanding requests. | ||
|  |  */ | ||
|  | 
 | ||
|  | /* Initialising empty rings */ | ||
|  | #define SHARED_RING_INIT(_s) do {					\
 | ||
|  |     (_s)->req_prod  = (_s)->rsp_prod  = 0;				\ | ||
|  |     (_s)->req_event = (_s)->rsp_event = 1;				\ | ||
|  |     memset((_s)->pad, 0, sizeof((_s)->pad));				\ | ||
|  | } while(0) | ||
|  | 
 | ||
|  | #define FRONT_RING_INIT(_r, _s, __size) do {				\
 | ||
|  |     (_r)->req_prod_pvt = 0;						\ | ||
|  |     (_r)->rsp_cons = 0;							\ | ||
|  |     (_r)->nr_ents = __RING_SIZE(_s, __size);				\ | ||
|  |     (_r)->sring = (_s);							\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | #define BACK_RING_INIT(_r, _s, __size) do {				\
 | ||
|  |     (_r)->rsp_prod_pvt = 0;						\ | ||
|  |     (_r)->req_cons = 0;							\ | ||
|  |     (_r)->nr_ents = __RING_SIZE(_s, __size);				\ | ||
|  |     (_r)->sring = (_s);							\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | /* Initialize to existing shared indexes -- for recovery */ | ||
|  | #define FRONT_RING_ATTACH(_r, _s, __size) do {				\
 | ||
|  |     (_r)->sring = (_s);							\ | ||
|  |     (_r)->req_prod_pvt = (_s)->req_prod;				\ | ||
|  |     (_r)->rsp_cons = (_s)->rsp_prod;					\ | ||
|  |     (_r)->nr_ents = __RING_SIZE(_s, __size);				\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | #define BACK_RING_ATTACH(_r, _s, __size) do {				\
 | ||
|  |     (_r)->sring = (_s);							\ | ||
|  |     (_r)->rsp_prod_pvt = (_s)->rsp_prod;				\ | ||
|  |     (_r)->req_cons = (_s)->req_prod;					\ | ||
|  |     (_r)->nr_ents = __RING_SIZE(_s, __size);				\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | /* How big is this ring? */ | ||
|  | #define RING_SIZE(_r)							\
 | ||
|  |     ((_r)->nr_ents) | ||
|  | 
 | ||
|  | /* Number of free requests (for use on front side only). */ | ||
|  | #define RING_FREE_REQUESTS(_r)						\
 | ||
|  |     (RING_SIZE(_r) - ((_r)->req_prod_pvt - (_r)->rsp_cons)) | ||
|  | 
 | ||
|  | /* Test if there is an empty slot available on the front ring.
 | ||
|  |  * (This is only meaningful from the front. ) | ||
|  |  */ | ||
|  | #define RING_FULL(_r)							\
 | ||
|  |     (RING_FREE_REQUESTS(_r) == 0) | ||
|  | 
 | ||
|  | /* Test if there are outstanding messages to be processed on a ring. */ | ||
|  | #define RING_HAS_UNCONSUMED_RESPONSES(_r)				\
 | ||
|  |     ((_r)->sring->rsp_prod - (_r)->rsp_cons) | ||
|  | 
 | ||
|  | #define RING_HAS_UNCONSUMED_REQUESTS(_r)				\
 | ||
|  |     ({									\ | ||
|  | 	unsigned int req = (_r)->sring->req_prod - (_r)->req_cons;	\ | ||
|  | 	unsigned int rsp = RING_SIZE(_r) -				\ | ||
|  | 			   ((_r)->req_cons - (_r)->rsp_prod_pvt);	\ | ||
|  | 	req < rsp ? req : rsp;						\ | ||
|  |     }) | ||
|  | 
 | ||
|  | /* Direct access to individual ring elements, by index. */ | ||
|  | #define RING_GET_REQUEST(_r, _idx)					\
 | ||
|  |     (&((_r)->sring->ring[((_idx) & (RING_SIZE(_r) - 1))].req)) | ||
|  | 
 | ||
|  | #define RING_GET_RESPONSE(_r, _idx)					\
 | ||
|  |     (&((_r)->sring->ring[((_idx) & (RING_SIZE(_r) - 1))].rsp)) | ||
|  | 
 | ||
|  | /* Loop termination condition: Would the specified index overflow the ring? */ | ||
|  | #define RING_REQUEST_CONS_OVERFLOW(_r, _cons)				\
 | ||
|  |     (((_cons) - (_r)->rsp_prod_pvt) >= RING_SIZE(_r)) | ||
|  | 
 | ||
|  | #define RING_PUSH_REQUESTS(_r) do {					\
 | ||
|  |     wmb(); /* back sees requests /before/ updated producer index */	\ | ||
|  |     (_r)->sring->req_prod = (_r)->req_prod_pvt;				\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | #define RING_PUSH_RESPONSES(_r) do {					\
 | ||
|  |     wmb(); /* front sees responses /before/ updated producer index */	\ | ||
|  |     (_r)->sring->rsp_prod = (_r)->rsp_prod_pvt;				\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | /*
 | ||
|  |  * Notification hold-off (req_event and rsp_event): | ||
|  |  * | ||
|  |  * When queueing requests or responses on a shared ring, it may not always be | ||
|  |  * necessary to notify the remote end. For example, if requests are in flight | ||
|  |  * in a backend, the front may be able to queue further requests without | ||
|  |  * notifying the back (if the back checks for new requests when it queues | ||
|  |  * responses). | ||
|  |  * | ||
|  |  * When enqueuing requests or responses: | ||
|  |  * | ||
|  |  *  Use RING_PUSH_{REQUESTS,RESPONSES}_AND_CHECK_NOTIFY(). The second argument | ||
|  |  *  is a boolean return value. True indicates that the receiver requires an | ||
|  |  *  asynchronous notification. | ||
|  |  * | ||
|  |  * After dequeuing requests or responses (before sleeping the connection): | ||
|  |  * | ||
|  |  *  Use RING_FINAL_CHECK_FOR_REQUESTS() or RING_FINAL_CHECK_FOR_RESPONSES(). | ||
|  |  *  The second argument is a boolean return value. True indicates that there | ||
|  |  *  are pending messages on the ring (i.e., the connection should not be put | ||
|  |  *  to sleep). | ||
|  |  * | ||
|  |  *  These macros will set the req_event/rsp_event field to trigger a | ||
|  |  *  notification on the very next message that is enqueued. If you want to | ||
|  |  *  create batches of work (i.e., only receive a notification after several | ||
|  |  *  messages have been enqueued) then you will need to create a customised | ||
|  |  *  version of the FINAL_CHECK macro in your own code, which sets the event | ||
|  |  *  field appropriately. | ||
|  |  */ | ||
|  | 
 | ||
|  | #define RING_PUSH_REQUESTS_AND_CHECK_NOTIFY(_r, _notify) do {		\
 | ||
|  |     RING_IDX __old = (_r)->sring->req_prod;				\ | ||
|  |     RING_IDX __new = (_r)->req_prod_pvt;				\ | ||
|  |     wmb(); /* back sees requests /before/ updated producer index */	\ | ||
|  |     (_r)->sring->req_prod = __new;					\ | ||
|  |     mb(); /* back sees new requests /before/ we check req_event */	\ | ||
|  |     (_notify) = ((RING_IDX)(__new - (_r)->sring->req_event) <		\ | ||
|  | 		 (RING_IDX)(__new - __old));				\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | #define RING_PUSH_RESPONSES_AND_CHECK_NOTIFY(_r, _notify) do {		\
 | ||
|  |     RING_IDX __old = (_r)->sring->rsp_prod;				\ | ||
|  |     RING_IDX __new = (_r)->rsp_prod_pvt;				\ | ||
|  |     wmb(); /* front sees responses /before/ updated producer index */	\ | ||
|  |     (_r)->sring->rsp_prod = __new;					\ | ||
|  |     mb(); /* front sees new responses /before/ we check rsp_event */	\ | ||
|  |     (_notify) = ((RING_IDX)(__new - (_r)->sring->rsp_event) <		\ | ||
|  | 		 (RING_IDX)(__new - __old));				\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | #define RING_FINAL_CHECK_FOR_REQUESTS(_r, _work_to_do) do {		\
 | ||
|  |     (_work_to_do) = RING_HAS_UNCONSUMED_REQUESTS(_r);			\ | ||
|  |     if (_work_to_do) break;						\ | ||
|  |     (_r)->sring->req_event = (_r)->req_cons + 1;			\ | ||
|  |     mb();								\ | ||
|  |     (_work_to_do) = RING_HAS_UNCONSUMED_REQUESTS(_r);			\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | #define RING_FINAL_CHECK_FOR_RESPONSES(_r, _work_to_do) do {		\
 | ||
|  |     (_work_to_do) = RING_HAS_UNCONSUMED_RESPONSES(_r);			\ | ||
|  |     if (_work_to_do) break;						\ | ||
|  |     (_r)->sring->rsp_event = (_r)->rsp_cons + 1;			\ | ||
|  |     mb();								\ | ||
|  |     (_work_to_do) = RING_HAS_UNCONSUMED_RESPONSES(_r);			\ | ||
|  | } while (0) | ||
|  | 
 | ||
|  | #endif /* __XEN_PUBLIC_IO_RING_H__ */
 |