IPCClient.cs 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417
  1. using InABox.Client.IPC;
  2. using InABox.Clients;
  3. using InABox.Core;
  4. namespace InABox.IPC
  5. {
  6. internal static class LocalCache
  7. {
  8. public static string Password { get; set; }
  9. }
  10. public class IPCClient<TEntity> : BaseClient<TEntity> where TEntity : Entity, new()
  11. {
  12. private IPCClientTransport _pipe;
  13. public IPCClient(string pipeName)
  14. {
  15. _pipe = IPCClientFactory.GetClient(pipeName);
  16. Timeout = TimeSpan.FromSeconds(300);
  17. }
  18. public override bool IsConnected() => _pipe?.Disconnected == false;
  19. private static string[]? _types;
  20. public override string[] SupportedTypes()
  21. {
  22. _types ??= CoreUtils.Entities
  23. .Where(x => x.GetInterfaces().Contains(typeof(IPersistent)))
  24. .Select(x => x.EntityName().Replace(".", "_"))
  25. .ToArray();
  26. return _types;
  27. }
  28. public override DatabaseInfo? Info()
  29. {
  30. try
  31. {
  32. var request = new InfoRequest();
  33. PrepareRequest(request, false);
  34. var response = Send(IPCMessage.Info(request), 5000)?.GetResponse<InfoResponse>();
  35. return response?.Info;
  36. }
  37. catch (Exception)
  38. {
  39. return new DatabaseInfo();
  40. }
  41. }
  42. private void PrepareRequest(Request request, bool doCredentials = true)
  43. {
  44. if(request is not ValidateRequest && _pipe.Disconnected)
  45. {
  46. ClientFactory.Validate(ClientFactory.UserID, LocalCache.Password);
  47. }
  48. if (doCredentials)
  49. {
  50. request.Credentials.Platform = ClientFactory.Platform;
  51. request.Credentials.Version = ClientFactory.Version;
  52. request.Credentials.Session = ClientFactory.SessionID;
  53. }
  54. Request.BeforeRequest?.Invoke(request);
  55. }
  56. private IPCMessage Send(IPCMessage request, int? timeout = null)
  57. {
  58. return _pipe.Send(request, timeout ?? Convert.ToInt32(Timeout.TotalMilliseconds));
  59. }
  60. protected override bool DoCheck2FA(string code, Guid? session)
  61. {
  62. var request = new Check2FARequest(code);
  63. PrepareRequest(request);
  64. var response = Send(IPCMessage.Check2FA(request))?.GetResponse<Check2FAResponse>();
  65. if (response != null)
  66. {
  67. return response.Status switch
  68. {
  69. StatusCode.OK => response.Valid,
  70. StatusCode.Unauthenticated => false,
  71. _ => throw new IPCException(response.Messages),
  72. };
  73. }
  74. return false;
  75. }
  76. protected override bool DoPing()
  77. {
  78. try
  79. {
  80. var request = new PingRequest();
  81. PrepareRequest(request);
  82. var message = Send(IPCMessage.Ping(request), 100_000);
  83. var response = message?.GetResponse<PingResponse>();
  84. if (response != null)
  85. {
  86. return response.Status switch
  87. {
  88. StatusCode.Error or StatusCode.BadServer or StatusCode.Incomplete => throw new IPCException(response.Messages),
  89. _ => true
  90. };
  91. }
  92. }
  93. catch (Exception) { }
  94. return false;
  95. }
  96. protected override void DoDelete(TEntity entity, string auditnote)
  97. {
  98. var request = new DeleteRequest<TEntity>(entity, auditnote);
  99. PrepareRequest(request);
  100. var response = Send(IPCMessage.Delete(request)).GetResponse<DeleteResponse<TEntity>>();
  101. switch (response.Status)
  102. {
  103. case StatusCode.OK:
  104. break;
  105. case StatusCode.Unauthenticated:
  106. throw new IPCException("Client not authenticated");
  107. default:
  108. throw new IPCException(response.Messages);
  109. }
  110. }
  111. protected override void DoDelete(IEnumerable<TEntity> entities, string auditnote)
  112. {
  113. var items = entities.AsArray();
  114. var request = new MultiDeleteRequest<TEntity>(items, auditnote);
  115. PrepareRequest(request);
  116. var response = Send(IPCMessage.MultiDelete(request)).GetResponse<MultiDeleteResponse<TEntity>>();
  117. switch (response.Status)
  118. {
  119. case StatusCode.OK:
  120. break;
  121. case StatusCode.Unauthenticated:
  122. throw new IPCException("Client not authenticated");
  123. default:
  124. throw new IPCException(response.Messages);
  125. }
  126. }
  127. protected override TEntity[] DoLoad(Filter<TEntity>? filter = null, SortOrder<TEntity>? sort = null, CoreRange? range = null)
  128. {
  129. var request = new QueryRequest<TEntity>(filter, null, sort, range);
  130. PrepareRequest(request);
  131. var response = Send(IPCMessage.Query(request)).GetResponse<QueryResponse<TEntity>>();
  132. if (response.Items != null)
  133. {
  134. return response.Items.ToArray<TEntity>();
  135. }
  136. else
  137. {
  138. return [];
  139. }
  140. }
  141. protected override CoreTable DoQuery(Filter<TEntity>? filter, Columns<TEntity>? columns, SortOrder<TEntity>? sort = null, CoreRange? range = null)
  142. {
  143. var request = new QueryRequest<TEntity>(filter, columns, sort, range);
  144. PrepareRequest(request);
  145. var response = Send(IPCMessage.Query(request)).GetResponse<QueryResponse<TEntity>>();
  146. return response.Status switch
  147. {
  148. StatusCode.OK => response.Items,
  149. StatusCode.Unauthenticated => throw new IPCException("Client not authenticated", StatusCode.Unauthenticated),
  150. _ => throw new IPCException(response.Messages),
  151. };
  152. }
  153. protected override Dictionary<string, CoreTable> DoQueryMultiple(Dictionary<string, IQueryDef> queries)
  154. {
  155. var request = new MultiQueryRequest();
  156. foreach (var item in queries)
  157. {
  158. request.AddQuery(item.Key, item.Value);
  159. }
  160. PrepareRequest(request);
  161. var response = Send(IPCMessage.QueryMultiple(request)).GetResponse<MultiQueryResponse>();
  162. if (response != null)
  163. {
  164. return response.Status switch
  165. {
  166. StatusCode.OK => response.Tables,
  167. StatusCode.Unauthenticated => throw new IPCException("Client not authenticated"),
  168. _ => throw new IPCException(response.Messages),
  169. };
  170. }
  171. return null;
  172. }
  173. protected override void DoSave(TEntity entity, string auditnote)
  174. {
  175. var request = new SaveRequest<TEntity>(entity, auditnote)
  176. {
  177. ReturnOnlyChanged = true
  178. };
  179. PrepareRequest(request);
  180. var response = Send(IPCMessage.Save(request)).GetResponse<SaveResponse<TEntity>>();
  181. switch (response.Status)
  182. {
  183. case StatusCode.OK:
  184. /*var props = CoreUtils.PropertyList(typeof(TEntity), x => true, true);
  185. entity.SetObserving(false);
  186. foreach (var prop in props.Keys)
  187. {
  188. var value = CoreUtils.GetPropertyValue(response.Item, prop);
  189. CoreUtils.SetPropertyValue(entity, prop, value);
  190. }
  191. entity.CommitChanges();
  192. entity.SetObserving(true);*/
  193. entity.SetObserving(false);
  194. foreach (var (key, value) in response.ChangedValues)
  195. {
  196. if (CoreUtils.TryGetProperty<TEntity>(key, out var property))
  197. {
  198. CoreUtils.SetPropertyValue(entity, key, CoreUtils.ChangeType(value, property.PropertyType));
  199. }
  200. }
  201. entity.CommitChanges();
  202. entity.SetObserving(true);
  203. break;
  204. case StatusCode.Unauthenticated:
  205. throw new IPCException("Client not authenticated");
  206. default:
  207. throw new IPCException(response.Messages);
  208. }
  209. }
  210. protected override void DoSave(IEnumerable<TEntity> entities, string auditnote)
  211. {
  212. var items = entities.ToArray();
  213. var request = new MultiSaveRequest<TEntity>(items, auditnote)
  214. {
  215. ReturnOnlyChanged = true
  216. };
  217. PrepareRequest(request);
  218. var response = Send(IPCMessage.MultiSave(request)).GetResponse<MultiSaveResponse<TEntity>>();
  219. switch (response.Status)
  220. {
  221. case StatusCode.OK:
  222. /*var props = CoreUtils.PropertyList(typeof(TEntity), x => true, true);
  223. for (var i = 0; i < items.Length; i++)
  224. {
  225. items[i].SetObserving(false);
  226. foreach (var prop in props.Keys)
  227. {
  228. var value = CoreUtils.GetPropertyValue(response.Items[i], prop);
  229. CoreUtils.SetPropertyValue(items[i], prop, value);
  230. }
  231. //CoreUtils.DeepClone<TEntity>(response.Items[i], items[i]);
  232. items[i].CommitChanges();
  233. items[i].SetObserving(true);
  234. }*/
  235. for (int i = 0; i < items.Length; ++i)
  236. {
  237. var entity = items[i];
  238. var changedValues = response.ChangedValues[i];
  239. entity.SetObserving(false);
  240. foreach (var (key, value) in changedValues)
  241. {
  242. if (CoreUtils.TryGetProperty<TEntity>(key, out var property))
  243. {
  244. CoreUtils.SetPropertyValue(entity, key, CoreUtils.ChangeType(value, property.PropertyType));
  245. }
  246. }
  247. entity.CommitChanges();
  248. entity.SetObserving(true);
  249. }
  250. break;
  251. case StatusCode.Unauthenticated:
  252. throw new IPCException("Client not authenticated");
  253. default:
  254. throw new IPCException(response.Messages);
  255. }
  256. }
  257. protected override IValidationData DoValidate(Guid session)
  258. {
  259. return Validate(
  260. null, null, false, session);
  261. }
  262. protected override IValidationData DoValidate(string pin, Guid session)
  263. {
  264. return Validate(
  265. null, pin, true, session);
  266. }
  267. protected override IValidationData DoValidate(string userid, string password, Guid session)
  268. {
  269. return Validate(
  270. userid, password, false, session);
  271. }
  272. private IValidationData Validate(string? userid, string? password, bool usePin, Guid session = default)
  273. {
  274. var ticks = DateTime.Now.ToUniversalTime().Ticks.ToString();
  275. var request = new ValidateRequest { UsePIN = usePin };
  276. if (usePin)
  277. {
  278. request.UserID = Encryption.Encrypt(ticks, "wCq9rryEJEuHIifYrxRjxg", true);
  279. request.Password = Encryption.Encrypt(ticks, "7mhvLnqMwkCAzN+zNGlyyg", true);
  280. request.PIN = password;
  281. }
  282. else
  283. {
  284. request.UserID = userid;
  285. request.Password = password;
  286. }
  287. request.Credentials.Platform = ClientFactory.Platform;
  288. request.Credentials.Version = ClientFactory.Version;
  289. PrepareRequest(request, false);
  290. if(session != Guid.Empty)
  291. {
  292. request.Credentials.Session = session;
  293. }
  294. var response = Send(IPCMessage.Validate(request), 10000).GetResponse<ValidateResponse>();
  295. if (response != null)
  296. if (response.Status.Equals(StatusCode.OK))
  297. {
  298. LocalCache.Password = password;
  299. return new ValidationData(
  300. response.ValidationStatus,
  301. response.UserID,
  302. response.UserGuid,
  303. response.SecurityID,
  304. response.Session,
  305. response.Recipient2FA,
  306. response.PasswordExpiration
  307. );
  308. }
  309. else if (response.Status == StatusCode.BadServer)
  310. {
  311. throw new IPCException(response.Messages);
  312. }
  313. return new ValidationData(
  314. ValidationStatus.INVALID,
  315. "",
  316. Guid.Empty,
  317. Guid.Empty,
  318. Guid.Empty,
  319. null,
  320. DateTime.MinValue
  321. );
  322. }
  323. public override string Version()
  324. {
  325. try
  326. {
  327. var request = new VersionRequest();
  328. PrepareRequest(request, false);
  329. var response = Send(IPCMessage.Version(request)).GetResponse<VersionResponse>();
  330. return response.Version;
  331. }
  332. catch (Exception)
  333. {
  334. return "";
  335. }
  336. }
  337. public override string ReleaseNotes()
  338. {
  339. try
  340. {
  341. var request = new ReleaseNotesRequest();
  342. PrepareRequest(request, false);
  343. var response = Send(IPCMessage.ReleaseNotes(request)).GetResponse<ReleaseNotesResponse>();
  344. return response.ReleaseNotes;
  345. }
  346. catch (Exception)
  347. {
  348. return "";
  349. }
  350. }
  351. public override byte[]? Installer()
  352. {
  353. try
  354. {
  355. var request = new InstallerRequest();
  356. PrepareRequest(request, false);
  357. var response = Send(IPCMessage.Installer(request)).GetResponse<InstallerResponse>();
  358. return response.Installer;
  359. }
  360. catch (Exception)
  361. {
  362. return null;
  363. }
  364. }
  365. }
  366. }