smcp_status_t smcp_outbound_append_content(const char* value, coap_size_t len) { smcp_status_t ret = SMCP_STATUS_FAILURE; smcp_t const self = smcp_get_current_instance(); coap_size_t max_len = smcp_outbound_get_space_remaining(); char* dest; if (SMCP_CSTR_LEN == len) { len = (coap_size_t)strlen(value); } require_action(max_len>len, bail, ret = SMCP_STATUS_MESSAGE_TOO_BIG); dest = smcp_outbound_get_content_ptr(&max_len); require(dest,bail); dest += self->outbound.content_len; memcpy(dest, value, len); self->outbound.content_len += len; ret = SMCP_STATUS_OK; bail: return ret; }
smcp_status_t plugtest_test_handler(smcp_node_t node) { smcp_status_t ret = SMCP_STATUS_NOT_ALLOWED; char* content = NULL; coap_size_t max_len = 0; smcp_method_t method = smcp_inbound_get_code(); if(method==COAP_METHOD_GET) { ret = smcp_outbound_begin_response(COAP_RESULT_205_CONTENT); } else if(method==COAP_METHOD_POST) { ret = smcp_outbound_begin_response(COAP_RESULT_201_CREATED); } else if(method==COAP_METHOD_PUT) { ret = smcp_outbound_begin_response(COAP_RESULT_204_CHANGED); } else if(method==COAP_METHOD_DELETE) { ret = smcp_outbound_begin_response(COAP_RESULT_202_DELETED); } if(ret) goto bail; smcp_outbound_add_option_uint(COAP_OPTION_CONTENT_TYPE, COAP_CONTENT_TYPE_TEXT_PLAIN); content = smcp_outbound_get_content_ptr(&max_len); if(!content) { ret = SMCP_STATUS_FAILURE; goto bail; } smcp_inbound_get_path(content, SMCP_GET_PATH_LEADING_SLASH|SMCP_GET_PATH_INCLUDE_QUERY); strlcat(content,"\nPlugtest!\nMethod = ",max_len); strlcat(content,coap_code_to_cstr(method),max_len); strlcat(content,"\n",max_len); { const uint8_t* value; coap_size_t value_len; coap_option_key_t key; while((key=smcp_inbound_next_option(&value, &value_len))!=COAP_OPTION_INVALID) { strlcat(content,coap_option_key_to_cstr(key,1),max_len); strlcat(content,": ",max_len); if(coap_option_value_is_string(key)) { int argh = strlen(content)+value_len; strlcat(content,(char*)value,MIN(max_len,argh+1)); content[argh] = 0; } else { strlcat(content,"<binary>",max_len); } strlcat(content,"\n",max_len); } } smcp_outbound_set_content_len((coap_size_t)strlen(content)); ret = smcp_outbound_send(); bail: return ret; }
smcp_status_t cgi_node_async_resend_response(void* context) { smcp_status_t ret = 0; cgi_node_request_t request = context; coap_size_t block_len = (1<<((request->block2&0x7)+4)); coap_size_t max_len; syslog(LOG_INFO, "cgi-node: Resending async response. . ."); ret = smcp_outbound_begin_async_response(COAP_RESULT_205_CONTENT,&request->async_response); require_noerr(ret,bail); // ret = smcp_outbound_add_option_uint(COAP_OPTION_CONTENT_TYPE, COAP_CONTENT_TYPE_TEXT_PLAIN); // require_noerr(ret,bail); if(request->block1!=BLOCK_OPTION_UNSPECIFIED) { ret = smcp_outbound_add_option_uint(COAP_OPTION_BLOCK1,request->block1); require_noerr(ret,bail); } if(request->state == CGI_NODE_STATE_ACTIVE_BLOCK2_WAIT_ACK) { if(request->stdout_buffer_len>block_len || request->fd_cmd_stdout>=0 || ((request->block2>>4)!=0)) { if(request->stdout_buffer_len>block_len || request->fd_cmd_stdout>=0) request->block2 |= (1<<3); else request->block2 &= ~(1<<3); ret = smcp_outbound_add_option_uint(COAP_OPTION_BLOCK2,request->block2); require_noerr(ret,bail); } char *content = smcp_outbound_get_content_ptr(&max_len); require_noerr(ret,bail); memcpy(content,request->stdout_buffer,MIN(request->stdout_buffer_len,block_len)); ret = smcp_outbound_set_content_len(MIN(request->stdout_buffer_len,block_len)); require_noerr(ret,bail); } ret = smcp_outbound_send(); require_noerr(ret,bail); bail: return ret; }
smcp_status_t plugtest_obs_handler( struct plugtest_server_s *self ) { smcp_status_t ret = SMCP_STATUS_NOT_ALLOWED; smcp_method_t method = smcp_inbound_get_code(); if(method==COAP_METHOD_GET) { char* content = NULL; coap_size_t max_len = 0; ret = smcp_outbound_begin_response(COAP_RESULT_205_CONTENT); require_noerr(ret, bail); ret = smcp_outbound_add_option_uint(COAP_OPTION_CONTENT_TYPE, COAP_CONTENT_TYPE_TEXT_PLAIN); require_noerr(ret, bail); if(!smcp_timer_is_scheduled(smcp_get_current_instance(), &self->obs_timer)) plugtest_obs_timer_callback(smcp_get_current_instance(),self); ret = smcp_observable_update(&self->observable,PLUGTEST_OBS_KEY); check_noerr(ret); ret = smcp_outbound_add_option_uint(COAP_OPTION_MAX_AGE,10); require_noerr(ret, bail); content = smcp_outbound_get_content_ptr(&max_len); require_action(content!=NULL, bail, ret = SMCP_STATUS_FAILURE); require_action(max_len>11, bail, ret = SMCP_STATUS_MESSAGE_TOO_BIG); ret = smcp_outbound_set_content_len( (coap_size_t)strlen(uint32_to_dec_cstr(content, (uint32_t)time(NULL))) ); require_noerr(ret, bail); ret = smcp_outbound_send(); require_noerr(ret, bail); } bail: return ret; }
smcp_status_t smcp_outbound_set_content_formatted(const char* fmt, ...) { smcp_status_t ret = SMCP_STATUS_FAILURE; va_list args; char* content = smcp_outbound_get_content_ptr(NULL); coap_size_t len = smcp_outbound_get_space_remaining(); require(content!=NULL, bail); content += smcp_get_current_instance()->outbound.content_len; va_start(args,fmt); len = (coap_size_t)vsnprintf(content,len,fmt,args); require(len!=0,bail); len += smcp_get_current_instance()->outbound.content_len; ret = smcp_outbound_set_content_len(len); bail: va_end(args); return ret; }
smcp_status_t resend_test_request(void* context) { test_data_s* test_data = context; smcp_status_t status = 0; status = smcp_outbound_begin(smcp_get_current_instance(),test_data->outbound_code, test_data->outbound_tt); require_noerr(status,bail); status = smcp_outbound_set_uri(test_data->url, 0); require_noerr(status,bail); if (test_data->outbound_content_type != COAP_CONTENT_TYPE_UNKNOWN) { status = smcp_outbound_add_option_uint(COAP_OPTION_CONTENT_TYPE, test_data->outbound_content_type); require_noerr(status,bail); } if (test_data->extra & EXT_BLOCK_01) { struct coap_block_info_s block1_info; uint32_t resource_length = 200; uint32_t block_stop; coap_decode_block(&block1_info, test_data->block1_option + (1<<4)); block_stop = block1_info.block_offset+block1_info.block_size; if (block1_info.block_offset < resource_length) { if (block_stop >= resource_length) { test_data->block1_option &= ~(1<<3); } else { test_data->block1_option |= (1<<3); } if (test_data->has_block1_option) { test_data->block1_option += (1<<4); } smcp_outbound_add_option_uint(COAP_OPTION_BLOCK1, test_data->block1_option); if (block1_info.block_m) { coap_size_t max_len = 0; uint32_t i; uint32_t block_stop; char* content = NULL; content = smcp_outbound_get_content_ptr(&max_len); if (!content) { status = SMCP_STATUS_FAILURE; goto bail; } block_stop = block1_info.block_offset+block1_info.block_size; for (i = block1_info.block_offset; i < block_stop; i++) { if (!((i + 1) % 64)) { content[i-block1_info.block_offset] = '\n'; } else { content[i-block1_info.block_offset] = '0'+(i%10); } } status = smcp_outbound_set_content_len(MIN((coap_code_t)(block_stop-block1_info.block_offset),(coap_code_t)(resource_length-block1_info.block_offset))); require_noerr(status,bail); } } } if(test_data->extra & EXT_BLOCK_02) { smcp_outbound_add_option_uint(COAP_OPTION_BLOCK2, 1); // 32 byte block size. } status = smcp_outbound_send(); if(status) { check_noerr(status); fprintf(stderr, "smcp_outbound_send() returned error %d(%s).\n", status, smcp_status_to_cstr(status)); goto bail; } else { test_data->outbound_attempts++; } bail: return status; }
smcp_status_t smcp_variable_node_request_handler( smcp_variable_node_t node ) { // TODO: Make this function use less stack space! smcp_method_t method = smcp_inbound_get_code(); smcp_status_t ret = SMCP_STATUS_NOT_FOUND; SMCP_NON_RECURSIVE coap_content_type_t content_type; SMCP_NON_RECURSIVE char* content_ptr; SMCP_NON_RECURSIVE size_t content_len; SMCP_NON_RECURSIVE char buffer[SMCP_VARIABLE_MAX_VALUE_LENGTH+1]; SMCP_NON_RECURSIVE coap_content_type_t reply_content_type; uint8_t key_index = BAD_KEY_INDEX; size_t value_len; bool needs_prefix = true; char* prefix_name = ""; content_type = smcp_inbound_get_content_type(); content_ptr = (char*)smcp_inbound_get_content_ptr(); content_len = smcp_inbound_get_content_len(); reply_content_type = SMCP_CONTENT_TYPE_APPLICATION_FORM_URLENCODED; require(node, bail); // Look up the key index. if(smcp_inbound_peek_option(NULL,&value_len)==COAP_OPTION_URI_PATH) { if(!value_len) { needs_prefix = false; smcp_inbound_next_option(NULL,NULL); } else for(key_index=0;key_index<BAD_KEY_INDEX;key_index++) { ret = node->func(node,SMCP_VAR_GET_KEY,key_index,buffer); require_action(ret==0,bail,ret=SMCP_STATUS_NOT_FOUND); if(smcp_inbound_option_strequal(COAP_OPTION_URI_PATH, buffer)) { smcp_inbound_next_option(NULL,NULL); break; } } } { coap_option_key_t key; const uint8_t* value; while((key=smcp_inbound_next_option(&value, &value_len))!=COAP_OPTION_INVALID) { require_action(key!=COAP_OPTION_URI_PATH,bail,ret=SMCP_STATUS_NOT_FOUND); if(key==COAP_OPTION_URI_QUERY) { if( method == COAP_METHOD_POST && value_len>=2 && strhasprefix_const((const char*)value,"v=") ) { DEBUG_PRINTF("variable-node: value is in the query."); content_type = COAP_CONTENT_TYPE_TEXT_PLAIN; content_ptr = (char*)value+2; content_len = value_len-2; } // } else if(key==COAP_OPTION_ETAG) { // } else if(key==COAP_OPTION_IF_MATCH) { // } else if(key==COAP_OPTION_IF_NONE_MATCH) { } else if(key==COAP_OPTION_ACCEPT) { reply_content_type = 0; if(value_len==1) reply_content_type = value[0]; else { // Unsupported type. } } else if(COAP_OPTION_IS_CRITICAL(key)) { ret=SMCP_STATUS_BAD_OPTION; assert_printf("Unrecognized option %d, \"%s\"", key, coap_option_key_to_cstr(key, false) ); goto bail; } } } // TODO: Implement me! if(method == COAP_METHOD_PUT) method = COAP_METHOD_POST; if(method == COAP_METHOD_POST) { require_action(!smcp_inbound_is_dupe(),bail,ret=0); require_action( key_index!=BAD_KEY_INDEX, bail, ret=SMCP_STATUS_NOT_ALLOWED ); if(content_type==SMCP_CONTENT_TYPE_APPLICATION_FORM_URLENCODED) { char* key = NULL; char* value = NULL; content_len = 0; while( url_form_next_value( (char**)&content_ptr, &key, &value ) && key && value ) { if(strequal_const(key, "v")) { content_ptr = value; content_len = strlen(value); break; } } } // Make sure our content is zero terminated. ((char*)content_ptr)[content_len] = 0; ret = node->func(node,SMCP_VAR_SET_VALUE,key_index,(char*)content_ptr); require_noerr(ret,bail); ret = smcp_outbound_begin_response(COAP_RESULT_204_CHANGED); require_noerr(ret,bail); ret = smcp_outbound_send(); require_noerr(ret,bail); } else if(method == COAP_METHOD_GET) { if(key_index==BAD_KEY_INDEX) { char* content_end_ptr; ret = smcp_outbound_begin_response(COAP_RESULT_205_CONTENT); require_noerr(ret,bail); smcp_outbound_add_option_uint(COAP_OPTION_CONTENT_TYPE, COAP_CONTENT_TYPE_APPLICATION_LINK_FORMAT); ret = smcp_observable_update(&node->observable, SMCP_OBSERVABLE_BROADCAST_KEY); check_string(ret==0,smcp_status_to_cstr(ret)); content_ptr = smcp_outbound_get_content_ptr(&content_len); content_end_ptr = content_ptr+content_len; for(key_index=0;key_index<BAD_KEY_INDEX;key_index++) { ret = node->func(node,SMCP_VAR_GET_KEY,key_index,buffer); if(ret) break; if(content_ptr+2>=content_end_ptr) { // No more room for content. break; } *content_ptr++ = '<'; if(needs_prefix) { content_ptr += url_encode_cstr(content_ptr, prefix_name, (content_end_ptr-content_ptr)-1); content_ptr = stpncpy(content_ptr,"/",MIN(1,(content_end_ptr-content_ptr)-1)); } content_ptr += url_encode_cstr(content_ptr, buffer, (content_end_ptr-content_ptr)-1); *content_ptr++ = '>'; ret = node->func(node,SMCP_VAR_GET_VALUE,key_index,buffer); if(content_ptr+4>=content_end_ptr) { // No more room for content. break; } if(!ret) { strcpy(content_ptr,";v="); content_ptr += 3; content_ptr += quoted_cstr(content_ptr, buffer, (content_end_ptr-content_ptr)-1); } ret = node->func(node,SMCP_VAR_GET_LF_TITLE,key_index,buffer); if(content_ptr+8>=content_end_ptr) { // No more room for content. break; } if(!ret) { strcpy(content_ptr,";title="); content_ptr += 7; content_ptr += quoted_cstr(content_ptr, buffer, (content_end_ptr-content_ptr)-1); } // Observation flag if(0==node->func(node,SMCP_VAR_GET_OBSERVABLE,key_index,NULL)) { content_ptr = stpncpy(content_ptr,";obs",MIN(4,(content_end_ptr-content_ptr)-1)); } *content_ptr++ = ','; } ret = smcp_outbound_set_content_len(content_len-(content_end_ptr-content_ptr)); require_noerr(ret,bail); ret = smcp_outbound_send(); } else { size_t replyContentLength = 0; char *replyContent; ret = smcp_outbound_begin_response(COAP_RESULT_205_CONTENT); require_noerr(ret,bail); if(0==node->func(node,SMCP_VAR_GET_OBSERVABLE,key_index,buffer)) { ret = smcp_observable_update(&node->observable, key_index); check_string(ret==0,smcp_status_to_cstr(ret)); } if(reply_content_type == SMCP_CONTENT_TYPE_APPLICATION_FORM_URLENCODED) { uint32_t etag; if(0==node->func(node,SMCP_VAR_GET_MAX_AGE,key_index,buffer)) { #if HAVE_STRTOL uint32_t max_age = strtol(buffer,NULL,0)&0xFFFFFF; #else uint32_t max_age = atoi(buffer)&0xFFFFFF; #endif smcp_outbound_add_option_uint(COAP_OPTION_MAX_AGE, max_age); } ret = node->func(node,SMCP_VAR_GET_VALUE,key_index,buffer); require_noerr(ret,bail); fasthash_start(0); fasthash_feed((const uint8_t*)buffer,strlen(buffer)); etag = fasthash_finish_uint32(); smcp_outbound_add_option_uint(COAP_OPTION_CONTENT_TYPE, SMCP_CONTENT_TYPE_APPLICATION_FORM_URLENCODED); smcp_outbound_add_option_uint(COAP_OPTION_ETAG, etag); replyContent = smcp_outbound_get_content_ptr(&replyContentLength); *replyContent++ = 'v'; *replyContent++ = '='; replyContentLength -= 2; replyContentLength = url_encode_cstr( replyContent, buffer, replyContentLength ); ret = smcp_outbound_set_content_len(replyContentLength+2); } else { ret = node->func(node,SMCP_VAR_GET_VALUE,key_index,buffer); require_noerr(ret,bail); ret = smcp_outbound_append_content(buffer, SMCP_CSTR_LEN); } require_noerr(ret,bail); ret = smcp_outbound_send(); } } else { ret = smcp_default_request_handler( (void*)node ); check_string(ret == SMCP_STATUS_OK, smcp_status_to_cstr(ret)); } bail: check_string(ret == SMCP_STATUS_OK, smcp_status_to_cstr(ret)); return ret; }
smcp_status_t plugtest_large_handler( smcp_node_t node ) { smcp_status_t ret = SMCP_STATUS_NOT_ALLOWED; char* content = NULL; coap_size_t max_len = 0; smcp_method_t method = smcp_inbound_get_code(); uint32_t block_option = 0x03; uint32_t block_start = 0; uint32_t block_stop = 0; uint32_t resource_length = 2000; if(method==COAP_METHOD_GET) { ret = 0; } require_noerr(ret,bail); { const uint8_t* value; coap_size_t value_len; coap_option_key_t key; while((key=smcp_inbound_next_option(&value, &value_len))!=COAP_OPTION_INVALID) { if(key == COAP_OPTION_BLOCK2) { uint8_t i; block_option = 0; for(i = 0; i < value_len; i++) block_option = (block_option << 8) + value[i]; } } } ret = smcp_outbound_begin_response(COAP_RESULT_205_CONTENT); require_noerr(ret,bail); ret = smcp_outbound_add_option_uint(COAP_OPTION_CONTENT_TYPE, COAP_CONTENT_TYPE_TEXT_PLAIN); require_noerr(ret,bail); ret = smcp_outbound_add_option_uint(COAP_OPTION_MAX_AGE,60*60); require_noerr(ret,bail); max_len = smcp_outbound_get_space_remaining()-2; // Here we are making sure our data will fit, // and adjusting our block-option size accordingly. do { struct coap_block_info_s block_info; coap_decode_block(&block_info, block_option); block_start = block_info.block_offset; block_stop = block_info.block_offset + block_info.block_size; if(max_len<(block_stop-block_start) && block_option!=0 && !block_info.block_offset) { block_option--; block_stop = 0; continue; } } while(0==block_stop); require_action(block_start<resource_length,bail,ret=SMCP_STATUS_INVALID_ARGUMENT); if(block_stop>=resource_length) block_option &= ~(1<<3); else block_option |= (1<<3); ret = smcp_outbound_add_option_uint(COAP_OPTION_BLOCK2,block_option); require_noerr(ret,bail); content = smcp_outbound_get_content_ptr(&max_len); require_action(NULL!=content, bail, ret = SMCP_STATUS_FAILURE); require_action(max_len>(block_stop-block_start), bail, ret = SMCP_STATUS_MESSAGE_TOO_BIG); { uint32_t i; for(i=block_start;i<block_stop;i++) { if(!((i+1)%64)) content[i-block_start] = '\n'; else content[i-block_start] = '0'+(i%10); } } ret = smcp_outbound_set_content_len((coap_size_t)MIN(block_stop-block_start,resource_length-block_start)); if(ret) goto bail; ret = smcp_outbound_send(); bail: return ret; }
smcp_status_t smcp_outbound_send() { smcp_status_t ret = SMCP_STATUS_FAILURE; smcp_t const self = smcp_get_current_instance(); coap_size_t header_len = (coap_size_t)(smcp_outbound_get_content_ptr(NULL)-(char*)self->outbound.packet); // Remove the start-of-payload marker if we have no payload. if (!smcp_get_current_instance()->outbound.content_len) { header_len--; } #if DEBUG { DEBUG_PRINTF("Outbound packet size: %d, %d remaining",header_len+smcp_get_current_instance()->outbound.content_len, smcp_outbound_get_space_remaining()); assert(header_len+smcp_get_current_instance()->outbound.content_len<=self->outbound.max_packet_len); assert(coap_verify_packet((char*)self->outbound.packet,header_len+smcp_get_current_instance()->outbound.content_len)); } #endif // DEBUG if (self->current_transaction) { self->current_transaction->sent_code = self->outbound.packet->code; self->current_transaction->sockaddr_remote = *smcp_plat_get_remote_sockaddr(); self->current_transaction->multicast = SMCP_IS_ADDR_MULTICAST(&self->current_transaction->sockaddr_remote.smcp_addr); } #if defined(SMCP_DEBUG_OUTBOUND_DROP_PERCENT) if(SMCP_DEBUG_OUTBOUND_DROP_PERCENT*SMCP_RANDOM_MAX>SMCP_FUNC_RANDOM_UINT32()) { DEBUG_PRINTF("Dropping outbound packet for debugging!"); if(smcp_get_current_instance()->is_responding) smcp_get_current_instance()->did_respond = true; smcp_get_current_instance()->is_responding = false; ret = SMCP_STATUS_OK; goto bail; } #endif if ((self->outbound.packet->tt == COAP_TRANS_TYPE_ACK) && smcp_session_type_is_reliable(smcp_plat_get_session_type()) ) { // If the session type is reliable, we don't bother // sending acks. } else { ret = smcp_plat_outbound_finish( self, (const uint8_t*)self->outbound.packet, header_len + self->outbound.content_len, 0 // FLAGS ); } require(ret == SMCP_STATUS_OK, bail); if (self->is_responding) { self->did_respond = true; } self->is_responding = false; ret = SMCP_STATUS_OK; bail: return ret; }