开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 网络编程 > 查看源码
test_klhttpd.c
资源名称：klhttpd.rar　[点击查看]
上传用户：jnmzc84
上传日期：2022-08-08
资源大小：35k
文件大小：2k
源码类别：

网络编程

开发平台：

Visual C++

test_klhttpd.c：源码内容
  1. /**
  2. */
  3. #include "klhttp.h"
  4. #include "klhttp-internal.h"
  5. #include <stdio.h>
  6. #ifdef WIN32
  7. #include <winsock2.h>
  8. #endif
  10. /**
  11.   get file size
  12. */
  13. int get_file_size( FILE *fp )
  14. {
  15. int file_size ;
  16. fseek( fp, 0, SEEK_END );
  17. file_size = ftell( fp );
  18. fseek( fp, 0, SEEK_SET );
  19. return file_size;
  20. }
  22. /**
  23.   resource file load.
  24. */
  25. int load_file( const char *uri, struct evbuffer *buf )
  26. {
  27. char file[260];
  28. FILE *fp;
  29. int file_size;
  30. char *data;
  32. if( strcmp( uri, "/" ) == 0 )
  33. {
  34. strcpy( file, "./www/index.html" );
  35. }
  36. else
  37. {
  38. sprintf( file, "./www%s", uri );
  39. }
  41. fp = fopen( file, "rb" );
  42. if( fp == 0 )
  43. {
  44. return -1;
  45. }
  47. file_size = get_file_size( fp );
  48. data = (char*) malloc( file_size );
  49. fread( data, file_size, 1, fp );
  50. evbuffer_add( buf, data, file_size );
  51. free( data );
  53. fclose( fp );
  55. return file_size;
  56. }
  58. int file_exist( const char *uri )
  59. {
  60. char file[256];
  61. FILE *fp;
  62. int file_size;
  64. if( strcmp( uri, "/" ) == 0 )
  65. {
  66. strcpy( file, "./www/index.html" );
  67. }
  68. else
  69. {
  70. sprintf( file, "./www%s", uri );
  71. }
  73. fp = fopen( file, "rb" );
  74. if( fp == 0 )
  75. {
  76. return -1;
  77. }
  79. file_size = get_file_size( fp );
  80. fclose( fp );
  81. return file_size;
  82. }
  84. void handle_request( struct http_connection *conn, const struct http_request *request, void *arg )
  85. {
  86. http_handle_request( conn->outbuf, request, file_exist, load_file );
  87. }
  89. void info_log( const char *msg )
  90. {
  91. printf( "%sn", msg );
  92. }
  94. volatile int is_done = 0;  
  96. /**
  97.   handle the ctrl+c event.
  98. */
  99. #ifdef WIN32
  100. BOOL WINAPI Handler( DWORD ctrl_event )
  101. {
  102. switch( ctrl_event )
  103. {
  104. case CTRL_CLOSE_EVENT:
  105. case CTRL_C_EVENT:
  106. {
  107. /* shutdown the server */
  108. is_done = 1;
  109. }
  111. return TRUE;
  112. }
  114. return FALSE;
  115. }
  116. #endif
  119. int main()
  120. {
  121. struct http_server *httpd;
  122. struct timeval timeout = { 2, 0 };
  123. #ifdef WIN32
  124. {
  125. WSADATA wd;
  126. WSAStartup( MAKEWORD( 2, 0 ), &wd );
  127. }
  128. #endif
  130. #ifdef WIN32
  131. if( !SetConsoleCtrlHandler( Handler, TRUE ) )
  132. {
  133. fprintf( stderr, "error setting event handler.n" );
  134. return -1;
  135. }
  136. #endif
  138. http_set_info_log( info_log );
  139. httpd = http_start( "0.0.0.0", 8080, 1024 );
  140. http_set_rcb( httpd, handle_request, 0 );
  142. printf( "server startupn" );
  143. while( is_done == 0 )
  144. {
  145. http_poll( httpd, &timeout );
  146. }
  148. http_free( httpd );
  150. #ifdef WIN32
  151. WSACleanup();
  152. #endif
  153. printf( "server shutdownn" );
  154. return 0;
  155. }