aboutsummaryrefslogtreecommitdiffstats
path: root/Projects/USBtoSerial/Lib/RingBuff.c
blob: 2109f8cc0bdfa59a480f9795c87b08c8aff61731 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/*
             LUFA Library
     Copyright (C) Dean Camera, 2010.
              
  dean [at] fourwalledcubicle [dot] com
      www.fourwalledcubicle.com
*/

/*
  Copyright 2010  Dean Camera (dean [at] fourwalledcubicle [dot] com)

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

  The author disclaim all warranties with regard to this
  software, including all implied warranties of merchantability
  and fitness.  In no event shall the author be liable for any
  special, indirect or consequential damages or any damages
  whatsoever resulting from 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.
*/

#include "RingBuff.h"

void Buffer_Initialize(RingBuff_t* const Buffer)
{
	BUFF_ATOMIC_BLOCK
	{
		Buffer->InPtr    = (RingBuff_Data_t*)&Buffer->Buffer;
		Buffer->OutPtr   = (RingBuff_Data_t*)&Buffer->Buffer;
		Buffer->Elements = 0;
	}
}

void Buffer_StoreElement(RingBuff_t* const Buffer, RingBuff_Data_t Data)
{
	BUFF_ATOMIC_BLOCK
	{
	#if defined(BUFF_DROPOLD)
		if (Buffer->Elements == BUFF_LENGTH)
		{
			Buffer->OutPtr++;
	
			if (Buffer->OutPtr == &Buffer->Buffer[BUFF_LENGTH])
			  Buffer->OutPtr = (RingBuff_Data_t*)&Buffer->Buffer;
		}
		else
		{
			Buffer->Elements++;
		}
	#elif defined(BUFF_DROPNEW)
		if (Buffer->Elements == BUFF_LENGTH)
		  return;
	
		  Buffer->Elements++;
	#elif defined(BUFF_NODROPCHECK)
		  Buffer->Elements++;
	#endif
	
		*(Buffer->InPtr) = Data;
		Buffer->InPtr++;
	
		if (Buffer->InPtr == &Buffer->Buffer[BUFF_LENGTH])
		  Buffer->InPtr = (RingBuff_Data_t*)&Buffer->Buffer;
	}
}

RingBuff_Data_t Buffer_GetElement(RingBuff_t* const Buffer)
{
	RingBuff_Data_t BuffData;
	
	BUFF_ATOMIC_BLOCK
	{
#if defined(BUFF_EMPTYRETURNSZERO)
		if (!(Buffer->Elements))
		  return 0;
#elif !defined(BUFF_NOEMPTYCHECK)
	#error No empty buffer check behavior specified.
#endif

		BuffData = *(Buffer->OutPtr);
	
		Buffer->OutPtr++;
		Buffer->Elements--;
	
		if (Buffer->OutPtr == &Buffer->Buffer[BUFF_LENGTH])
		  Buffer->OutPtr = (RingBuff_Data_t*)&Buffer->Buffer;
	}
	
	return BuffData;
}

#if defined(BUFF_USEPEEK)
RingBuff_Data_t Buffer_PeekElement(const RingBuff_t* const Buffer)
{
	RingBuff_Data_t BuffData;

	BUFF_ATOMIC_BLOCK
	{
#if defined(BUFF_EMPTYRETURNSZERO)
		if (!(Buffer->Elements))
		  return 0;
#elif !defined(BUFF_NOEMPTYCHECK)
	#error No empty buffer check behavior specified.
#endif

		BuffData = *(Buffer->OutPtr);
	}

	return BuffData;
}
#endif
ET, SOCK_STREAM, 0); memset(&addr, 0, sizeof(addr)); addr.sin_family = AF_INET; addr.sin_addr.s_addr = htonl(INADDR_ANY); addr.sin_port = htons(VTPM_MIG_PORT); if (bind(sock_descr, (struct sockaddr *)&addr, sizeof(addr)) == -1 ) { vtpmlogerror(VTPM_LOG_VTPM, "Failed to bind to port %d.\n", VTPM_MIG_PORT); return 1; } listen(sock_descr, 10); for(;;) { // ============ clear client info and wait for connection ========== memset(&client_addr, 0, sizeof(client_addr)); client_length = sizeof(client_addr); vtpmloginfo(VTPM_LOG_VTPM, "Waiting for incoming migrations...\n"); client_sock=accept(sock_descr, &client_addr, &client_length); if (client_sock == -1) { vtpmlogerror(VTPM_LOG_VTPM, "Incoming connectionn failed.\n"); goto abort_command; } else { vtpmloginfo(VTPM_LOG_VTPM, "Incoming connection accepted.\n"); } // =================== Read incoming command ====================== size_read = read( client_sock, cmd_header, VTPM_COMMAND_HEADER_SIZE); if (size_read > 0) { vtpmloginfo(VTPM_LOG_VTPM_DEEP, "RECV: 0x"); for (i=0; i<size_read; i++) vtpmloginfomore(VTPM_LOG_VTPM_DEEP, "%x ", cmd_header[i]); } else { vtpmlogerror(VTPM_LOG_VTPM, "Error reading from socket.\n"); build_error_msg(&result_buf, TPM_IOERROR); goto abort_command_with_error; } if (size_read < (int) VTPM_COMMAND_HEADER_SIZE) { vtpmlogerror(VTPM_LOG_VTPM, "Command from socket shorter than std header.\n"); build_error_msg(&result_buf, TPM_BAD_PARAMETER); goto abort_command_with_error; } // Unpack response from client BSG_UnpackList(cmd_header, 3, BSG_TPM_TAG, &tag, BSG_TYPE_UINT32, &in_param_size, BSG_TPM_COMMAND_CODE, &ord ); // If response has parameters, read them. // Note that out_param_size is in the client's context adj_param_size = in_param_size - VTPM_COMMAND_HEADER_SIZE; if (adj_param_size > 0) { buffer_init( &in_param_buf, adj_param_size, NULL); size_read = read(client_sock, in_param_buf.bytes, adj_param_size); if (size_read > 0) { for (i=0; i< size_read; i++) vtpmloginfomore(VTPM_LOG_VTPM_DEEP, "%x ", in_param_buf.bytes[i]); } else { vtpmlogerror(VTPM_LOG_VTPM, "Error reading from socket.\n"); build_error_msg(&result_buf, TPM_IOERROR); goto abort_command_with_error; } vtpmloginfomore(VTPM_LOG_VTPM, "\n"); if (size_read < (int)adj_param_size) { vtpmloginfomore(VTPM_LOG_VTPM, "\n"); vtpmlogerror(VTPM_LOG_VTPM, "Command read(%d) is shorter than header indicates(%d).\n", size_read, adj_param_size); build_error_msg(&result_buf, TPM_BAD_PARAMETER); goto abort_command_with_error; } } else { vtpmloginfomore(VTPM_LOG_VTPM, "\n"); } /* Handle Command */ switch (ord) { case VTPM_MORD_MIG_STEP2: handle_vtpm_mig_step2(&in_param_buf, &result_buf); break; case VTPM_MORD_MIG_STEP3: handle_vtpm_mig_step3(&in_param_buf, &result_buf); break; default: build_error_msg(&result_buf, TPM_BAD_PARAMETER); goto abort_command_with_error; } abort_command_with_error: /* Write Response */ size_write = write(client_sock, result_buf.bytes, buffer_len(&result_buf)); if (size_write > 0) { vtpmloginfo(VTPM_LOG_VTPM_DEEP, "SENT: 0x"); for (i=0; i< buffer_len(&result_buf); i++) { vtpmloginfomore(VTPM_LOG_VTPM_DEEP, "%x ", result_buf.bytes[i]); } vtpmloginfomore(VTPM_LOG_VTPM_DEEP, "\n"); } else { vtpmlogerror(VTPM_LOG_VTPM, "Error writing response to client.\n"); goto abort_command; } if (size_write != (int) buffer_len(&result_buf) ) vtpmlogerror(VTPM_LOG_VTPM, "Could not send entire response to client(%d/%d)\n", size_write, buffer_len(&result_buf)); abort_command: close(client_sock); buffer_free(&in_param_buf); buffer_free(&result_buf); } // For (;;) return 0; }