plink_client.c 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203
  1. /*
  2. * Copyright (c) 2021 Alibaba Group. All rights reserved.
  3. * License-Identifier: Apache-2.0
  4. *
  5. * Licensed under the Apache License, Version 2.0 (the "License"); you may
  6. * not use this file except in compliance with the License.
  7. * You may obtain a copy of the License at
  8. *
  9. * http://www.apache.org/licenses/LICENSE-2.0
  10. *
  11. * Unless required by applicable law or agreed to in writing, software
  12. * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  13. * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. * See the License for the specific language governing permissions and
  15. * limitations under the License.
  16. *
  17. */
  18. #include <stdio.h>
  19. #include <stdlib.h>
  20. #include <unistd.h>
  21. #include <fcntl.h>
  22. #include <sys/mman.h>
  23. #include <pthread.h>
  24. #include <memory.h>
  25. #include "process_linker_types.h"
  26. #include "video_mem.h"
  27. #ifndef NULL
  28. #define NULL ((void *)0)
  29. #endif
  30. #define NUM_OF_BUFFERS 5
  31. #define errExit(msg) do { perror(msg); exit(EXIT_FAILURE); \
  32. } while (0)
  33. int main(int argc, char **argv) {
  34. PlinkStatus sts = PLINK_STATUS_OK;
  35. PlinkPacket sendpkt, recvpkt;
  36. PlinkMsg msg;
  37. PlinkHandle plink = NULL;
  38. VmemParams params;
  39. void *vmem = NULL;
  40. FILE *fp = NULL;
  41. int exitcode = 0;
  42. int frames = argc > 1 ? atoi(argv[1]) : 1000;
  43. char *plinkname = argc > 2 ? argv[2] : "/tmp/plink.test";
  44. char *dumpname = argc > 3 ? argv[3] : NULL;
  45. if (dumpname != NULL)
  46. {
  47. fp = fopen(dumpname, "wb");
  48. if (fp == NULL)
  49. errExit("fopen");
  50. }
  51. if (VMEM_create(&vmem) != VMEM_STATUS_OK)
  52. errExit("Failed to create VMEM.");
  53. if (PLINK_create(&plink, plinkname, PLINK_MODE_CLIENT) != PLINK_STATUS_OK)
  54. errExit("Failed to create PLINK.");
  55. if (PLINK_connect(plink, NULL) != PLINK_STATUS_OK)
  56. errExit("Failed to connect to server.");
  57. int frmcnt = 0;
  58. do {
  59. sts = PLINK_recv(plink, 0, &recvpkt);
  60. memset(&params, 0, sizeof(params));
  61. if (recvpkt.fd != PLINK_INVALID_FD)
  62. {
  63. params.fd = recvpkt.fd;
  64. if (VMEM_import(vmem, &params) != VMEM_STATUS_OK)
  65. errExit("Failed to import fd.");
  66. if (VMEM_mmap(vmem, &params) != VMEM_STATUS_OK)
  67. errExit("Failed to mmap buffer.");
  68. }
  69. for (int i = 0; i < recvpkt.num; i++)
  70. {
  71. PlinkDescHdr *hdr = (PlinkDescHdr *)(recvpkt.list[i]);
  72. if (hdr->type == PLINK_TYPE_2D_YUV)
  73. {
  74. PlinkYuvInfo *pic = (PlinkYuvInfo *)(recvpkt.list[i]);
  75. printf("[CLIENT] Received YUV frame %d 0x%010llx: fd %d, %dx%d, stride = luma %d, chroma %d\n",
  76. pic->header.id, pic->bus_address_y, recvpkt.fd,
  77. pic->pic_width, pic->pic_height,
  78. pic->stride_y, pic->stride_u);
  79. // Save YUV data to file
  80. if (fp != NULL && params.vir_address != NULL)
  81. {
  82. void *buffer = params.vir_address;
  83. for (int i = 0; i < pic->pic_height * 3 / 2; i++)
  84. {
  85. fwrite(buffer, pic->pic_width, 1, fp);
  86. buffer += pic->stride_y;
  87. }
  88. }
  89. // return the buffer to source
  90. msg.header.type = PLINK_TYPE_MESSAGE;
  91. msg.header.size = DATA_SIZE(PlinkMsg);
  92. msg.msg = hdr->id;
  93. sendpkt.list[0] = &msg;
  94. sendpkt.num = 1;
  95. sendpkt.fd = PLINK_INVALID_FD;
  96. if (PLINK_send(plink, 0, &sendpkt) == PLINK_STATUS_ERROR)
  97. errExit("Failed to send data.");
  98. }
  99. if (hdr->type == PLINK_TYPE_2D_RGB)
  100. {
  101. PlinkRGBInfo *pic = (PlinkRGBInfo *)(recvpkt.list[i]);
  102. printf("[CLIENT] Received RGB picture %d 0x%010llx: fd %d, %dx%d, stride %d/%d/%d/%d\n",
  103. pic->header.id, pic->bus_address_r, recvpkt.fd,
  104. pic->img_width, pic->img_height,
  105. pic->stride_r, pic->stride_g, pic->stride_b, pic->stride_a);
  106. // Save RGB data to file
  107. if (fp != NULL && params.vir_address != NULL && pic->format == PLINK_COLOR_Format24BitBGR888Planar)
  108. {
  109. void *buffer = params.vir_address;
  110. for (int i = 0; i < pic->img_height * 3; i++)
  111. {
  112. fwrite(buffer, pic->img_width, 1, fp);
  113. buffer += pic->stride_r;
  114. }
  115. }
  116. // return the buffer to source
  117. msg.header.type = PLINK_TYPE_MESSAGE;
  118. msg.header.size = DATA_SIZE(PlinkMsg);
  119. msg.msg = hdr->id;
  120. sendpkt.list[0] = &msg;
  121. sendpkt.num = 1;
  122. sendpkt.fd = PLINK_INVALID_FD;
  123. if (PLINK_send(plink, 0, &sendpkt) == PLINK_STATUS_ERROR)
  124. errExit("Failed to send data.");
  125. }
  126. if (hdr->type == PLINK_TYPE_2D_RAW)
  127. {
  128. PlinkRawInfo *pic = (PlinkRawInfo *)(recvpkt.list[i]);
  129. printf("[CLIENT] Received RAW picture %d 0x%010llx: fd %d, %dx%d, stride %d\n",
  130. pic->header.id, pic->bus_address, recvpkt.fd,
  131. pic->img_width, pic->img_height, pic->stride);
  132. // Save RAW data to file
  133. if (fp != NULL && params.vir_address != NULL)
  134. fwrite(params.vir_address, pic->stride * pic->img_height, 1, fp);
  135. // return the buffer to source
  136. msg.header.type = PLINK_TYPE_MESSAGE;
  137. msg.header.size = DATA_SIZE(PlinkMsg);
  138. msg.msg = hdr->id;
  139. sendpkt.list[0] = &msg;
  140. sendpkt.num = 1;
  141. sendpkt.fd = PLINK_INVALID_FD;
  142. if (PLINK_send(plink, 0, &sendpkt) == PLINK_STATUS_ERROR)
  143. errExit("Failed to send data.");
  144. }
  145. else if (hdr->type == PLINK_TYPE_MESSAGE)
  146. {
  147. PlinkMsg *msg = (PlinkMsg *)(recvpkt.list[i]);
  148. if (msg->msg == PLINK_EXIT_CODE)
  149. {
  150. exitcode = 1;
  151. printf("Exit\n");
  152. break;
  153. }
  154. }
  155. }
  156. if (VMEM_release(vmem, &params) != VMEM_STATUS_OK)
  157. errExit("Failed to release buffer.");
  158. if (recvpkt.fd != PLINK_INVALID_FD)
  159. close(recvpkt.fd);
  160. frmcnt++;
  161. if (frmcnt >= frames)
  162. {
  163. msg.header.type = PLINK_TYPE_MESSAGE;
  164. msg.header.size = DATA_SIZE(PlinkMsg);
  165. msg.msg = PLINK_EXIT_CODE;
  166. sendpkt.list[0] = &msg;
  167. sendpkt.num = 1;
  168. sendpkt.fd = PLINK_INVALID_FD;
  169. if (PLINK_send(plink, 0, &sendpkt) == PLINK_STATUS_ERROR)
  170. errExit("Failed to send data.");
  171. break;
  172. }
  173. } while (exitcode == 0);
  174. cleanup:
  175. sleep(1); // Sleep one second to make sure server is ready for exit
  176. PLINK_close(plink, 0);
  177. VMEM_destroy(vmem);
  178. if (fp != NULL)
  179. fclose(fp);
  180. exit(EXIT_SUCCESS);
  181. }