GPBUnknownField.m 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507
  1. // Protocol Buffers - Google's data interchange format
  2. // Copyright 2008 Google Inc. All rights reserved.
  3. //
  4. // Use of this source code is governed by a BSD-style
  5. // license that can be found in the LICENSE file or at
  6. // https://developers.google.com/open-source/licenses/bsd
  7. #import "GPBUnknownField.h"
  8. #import "GPBUnknownField_PackagePrivate.h"
  9. #import "GPBArray.h"
  10. #import "GPBCodedOutputStream.h"
  11. #import "GPBCodedOutputStream_PackagePrivate.h"
  12. #import "GPBUnknownFieldSet.h"
  13. #import "GPBUnknownFields.h"
  14. #import "GPBUnknownFields_PackagePrivate.h"
  15. #import "GPBWireFormat.h"
  16. #define ASSERT_FIELD_TYPE(type) \
  17. if (type_ != type) { \
  18. [NSException raise:NSInternalInconsistencyException \
  19. format:@"GPBUnknownField is the wrong type"]; \
  20. }
  21. @implementation GPBUnknownField
  22. @synthesize number = number_;
  23. @synthesize type = type_;
  24. - (instancetype)initWithNumber:(int32_t)number {
  25. if ((self = [super init])) {
  26. number_ = number;
  27. type_ = GPBUnknownFieldTypeLegacy;
  28. }
  29. return self;
  30. }
  31. - (instancetype)initWithNumber:(int32_t)number varint:(uint64_t)varint {
  32. if ((self = [super init])) {
  33. number_ = number;
  34. type_ = GPBUnknownFieldTypeVarint;
  35. storage_.intValue = varint;
  36. }
  37. return self;
  38. }
  39. - (instancetype)initWithNumber:(int32_t)number fixed32:(uint32_t)fixed32 {
  40. if ((self = [super init])) {
  41. number_ = number;
  42. type_ = GPBUnknownFieldTypeFixed32;
  43. storage_.intValue = fixed32;
  44. }
  45. return self;
  46. }
  47. - (instancetype)initWithNumber:(int32_t)number fixed64:(uint64_t)fixed64 {
  48. if ((self = [super init])) {
  49. number_ = number;
  50. type_ = GPBUnknownFieldTypeFixed64;
  51. storage_.intValue = fixed64;
  52. }
  53. return self;
  54. }
  55. - (instancetype)initWithNumber:(int32_t)number lengthDelimited:(nonnull NSData *)data {
  56. if ((self = [super init])) {
  57. number_ = number;
  58. type_ = GPBUnknownFieldTypeLengthDelimited;
  59. storage_.lengthDelimited = [data copy];
  60. }
  61. return self;
  62. }
  63. - (instancetype)initWithNumber:(int32_t)number group:(nonnull GPBUnknownFields *)group {
  64. if ((self = [super init])) {
  65. number_ = number;
  66. type_ = GPBUnknownFieldTypeGroup;
  67. // Taking ownership of the group; so retain, not copy.
  68. storage_.group = [group retain];
  69. }
  70. return self;
  71. }
  72. - (void)dealloc {
  73. switch (type_) {
  74. case GPBUnknownFieldTypeVarint:
  75. case GPBUnknownFieldTypeFixed32:
  76. case GPBUnknownFieldTypeFixed64:
  77. break;
  78. case GPBUnknownFieldTypeLengthDelimited:
  79. [storage_.lengthDelimited release];
  80. break;
  81. case GPBUnknownFieldTypeGroup:
  82. [storage_.group release];
  83. break;
  84. case GPBUnknownFieldTypeLegacy:
  85. [storage_.legacy.mutableVarintList release];
  86. [storage_.legacy.mutableFixed32List release];
  87. [storage_.legacy.mutableFixed64List release];
  88. [storage_.legacy.mutableLengthDelimitedList release];
  89. [storage_.legacy.mutableGroupList release];
  90. break;
  91. }
  92. [super dealloc];
  93. }
  94. // Direct access is use for speed, to avoid even internally declaring things
  95. // read/write, etc. The warning is enabled in the project to ensure code calling
  96. // protos can turn on -Wdirect-ivar-access without issues.
  97. #pragma clang diagnostic push
  98. #pragma clang diagnostic ignored "-Wdirect-ivar-access"
  99. - (uint64_t)varint {
  100. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeVarint);
  101. return storage_.intValue;
  102. }
  103. - (uint32_t)fixed32 {
  104. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeFixed32);
  105. return (uint32_t)storage_.intValue;
  106. }
  107. - (uint64_t)fixed64 {
  108. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeFixed64);
  109. return storage_.intValue;
  110. }
  111. - (NSData *)lengthDelimited {
  112. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLengthDelimited);
  113. return storage_.lengthDelimited;
  114. }
  115. - (GPBUnknownFields *)group {
  116. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeGroup);
  117. return storage_.group;
  118. }
  119. - (GPBUInt64Array *)varintList {
  120. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  121. return storage_.legacy.mutableVarintList;
  122. }
  123. - (GPBUInt32Array *)fixed32List {
  124. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  125. return storage_.legacy.mutableFixed32List;
  126. }
  127. - (GPBUInt64Array *)fixed64List {
  128. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  129. return storage_.legacy.mutableFixed64List;
  130. }
  131. - (NSArray<NSData *> *)lengthDelimitedList {
  132. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  133. return storage_.legacy.mutableLengthDelimitedList;
  134. }
  135. - (NSArray<GPBUnknownFieldSet *> *)groupList {
  136. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  137. return storage_.legacy.mutableGroupList;
  138. }
  139. - (id)copyWithZone:(NSZone *)zone {
  140. switch (type_) {
  141. case GPBUnknownFieldTypeVarint:
  142. case GPBUnknownFieldTypeFixed32:
  143. case GPBUnknownFieldTypeFixed64:
  144. case GPBUnknownFieldTypeLengthDelimited:
  145. // In these modes, the object isn't mutable, so just return self.
  146. return [self retain];
  147. case GPBUnknownFieldTypeGroup: {
  148. // The `GPBUnknownFields` for the group is mutable, so a new instance of this object and
  149. // the group is needed.
  150. GPBUnknownFields *copyGroup = [storage_.group copyWithZone:zone];
  151. GPBUnknownField *copy = [[GPBUnknownField allocWithZone:zone] initWithNumber:number_
  152. group:copyGroup];
  153. [copyGroup release];
  154. return copy;
  155. }
  156. case GPBUnknownFieldTypeLegacy: {
  157. #pragma clang diagnostic push
  158. #pragma clang diagnostic ignored "-Wdeprecated-declarations"
  159. GPBUnknownField *result = [[GPBUnknownField allocWithZone:zone] initWithNumber:number_];
  160. result->storage_.legacy.mutableFixed32List =
  161. [storage_.legacy.mutableFixed32List copyWithZone:zone];
  162. result->storage_.legacy.mutableFixed64List =
  163. [storage_.legacy.mutableFixed64List copyWithZone:zone];
  164. result->storage_.legacy.mutableLengthDelimitedList =
  165. [storage_.legacy.mutableLengthDelimitedList mutableCopyWithZone:zone];
  166. result->storage_.legacy.mutableVarintList =
  167. [storage_.legacy.mutableVarintList copyWithZone:zone];
  168. if (storage_.legacy.mutableGroupList.count) {
  169. result->storage_.legacy.mutableGroupList = [[NSMutableArray allocWithZone:zone]
  170. initWithCapacity:storage_.legacy.mutableGroupList.count];
  171. for (GPBUnknownFieldSet *group in storage_.legacy.mutableGroupList) {
  172. GPBUnknownFieldSet *copied = [group copyWithZone:zone];
  173. [result->storage_.legacy.mutableGroupList addObject:copied];
  174. [copied release];
  175. }
  176. }
  177. #pragma clang diagnostic pop
  178. return result;
  179. }
  180. }
  181. }
  182. - (BOOL)isEqual:(id)object {
  183. if (self == object) return YES;
  184. if (![object isKindOfClass:[GPBUnknownField class]]) return NO;
  185. GPBUnknownField *field = (GPBUnknownField *)object;
  186. if (number_ != field->number_) return NO;
  187. if (type_ != field->type_) return NO;
  188. switch (type_) {
  189. case GPBUnknownFieldTypeVarint:
  190. case GPBUnknownFieldTypeFixed32:
  191. case GPBUnknownFieldTypeFixed64:
  192. return storage_.intValue == field->storage_.intValue;
  193. case GPBUnknownFieldTypeLengthDelimited:
  194. return [storage_.lengthDelimited isEqual:field->storage_.lengthDelimited];
  195. case GPBUnknownFieldTypeGroup:
  196. return [storage_.group isEqual:field->storage_.group];
  197. case GPBUnknownFieldTypeLegacy: {
  198. BOOL equalVarint =
  199. (storage_.legacy.mutableVarintList.count == 0 &&
  200. field->storage_.legacy.mutableVarintList.count == 0) ||
  201. [storage_.legacy.mutableVarintList isEqual:field->storage_.legacy.mutableVarintList];
  202. if (!equalVarint) return NO;
  203. BOOL equalFixed32 =
  204. (storage_.legacy.mutableFixed32List.count == 0 &&
  205. field->storage_.legacy.mutableFixed32List.count == 0) ||
  206. [storage_.legacy.mutableFixed32List isEqual:field->storage_.legacy.mutableFixed32List];
  207. if (!equalFixed32) return NO;
  208. BOOL equalFixed64 =
  209. (storage_.legacy.mutableFixed64List.count == 0 &&
  210. field->storage_.legacy.mutableFixed64List.count == 0) ||
  211. [storage_.legacy.mutableFixed64List isEqual:field->storage_.legacy.mutableFixed64List];
  212. if (!equalFixed64) return NO;
  213. BOOL equalLDList = (storage_.legacy.mutableLengthDelimitedList.count == 0 &&
  214. field->storage_.legacy.mutableLengthDelimitedList.count == 0) ||
  215. [storage_.legacy.mutableLengthDelimitedList
  216. isEqual:field->storage_.legacy.mutableLengthDelimitedList];
  217. if (!equalLDList) return NO;
  218. BOOL equalGroupList =
  219. (storage_.legacy.mutableGroupList.count == 0 &&
  220. field->storage_.legacy.mutableGroupList.count == 0) ||
  221. [storage_.legacy.mutableGroupList isEqual:field->storage_.legacy.mutableGroupList];
  222. if (!equalGroupList) return NO;
  223. return YES;
  224. }
  225. }
  226. }
  227. - (NSUInteger)hash {
  228. const int prime = 31;
  229. NSUInteger result = prime * number_ + type_;
  230. switch (type_) {
  231. case GPBUnknownFieldTypeVarint:
  232. case GPBUnknownFieldTypeFixed32:
  233. case GPBUnknownFieldTypeFixed64:
  234. result = prime * result + (NSUInteger)storage_.intValue;
  235. break;
  236. case GPBUnknownFieldTypeLengthDelimited:
  237. result = prime * result + [storage_.lengthDelimited hash];
  238. break;
  239. case GPBUnknownFieldTypeGroup:
  240. result = prime * result + [storage_.group hash];
  241. case GPBUnknownFieldTypeLegacy:
  242. result = prime * result + [storage_.legacy.mutableVarintList hash];
  243. result = prime * result + [storage_.legacy.mutableFixed32List hash];
  244. result = prime * result + [storage_.legacy.mutableFixed64List hash];
  245. result = prime * result + [storage_.legacy.mutableLengthDelimitedList hash];
  246. result = prime * result + [storage_.legacy.mutableGroupList hash];
  247. break;
  248. }
  249. return result;
  250. }
  251. - (void)writeToOutput:(GPBCodedOutputStream *)output {
  252. #pragma clang diagnostic push
  253. #pragma clang diagnostic ignored "-Wdeprecated-declarations"
  254. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  255. NSUInteger count = storage_.legacy.mutableVarintList.count;
  256. if (count > 0) {
  257. [output writeUInt64Array:number_ values:storage_.legacy.mutableVarintList tag:0];
  258. }
  259. count = storage_.legacy.mutableFixed32List.count;
  260. if (count > 0) {
  261. [output writeFixed32Array:number_ values:storage_.legacy.mutableFixed32List tag:0];
  262. }
  263. count = storage_.legacy.mutableFixed64List.count;
  264. if (count > 0) {
  265. [output writeFixed64Array:number_ values:storage_.legacy.mutableFixed64List tag:0];
  266. }
  267. count = storage_.legacy.mutableLengthDelimitedList.count;
  268. if (count > 0) {
  269. [output writeBytesArray:number_ values:storage_.legacy.mutableLengthDelimitedList];
  270. }
  271. count = storage_.legacy.mutableGroupList.count;
  272. if (count > 0) {
  273. [output writeUnknownGroupArray:number_ values:storage_.legacy.mutableGroupList];
  274. }
  275. #pragma clang diagnostic pop
  276. }
  277. - (size_t)serializedSize {
  278. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  279. #pragma clang diagnostic push
  280. #pragma clang diagnostic ignored "-Wdeprecated-declarations"
  281. __block size_t result = 0;
  282. int32_t number = number_;
  283. [storage_.legacy.mutableVarintList
  284. enumerateValuesWithBlock:^(uint64_t value, __unused NSUInteger idx, __unused BOOL *stop) {
  285. result += GPBComputeUInt64Size(number, value);
  286. }];
  287. [storage_.legacy.mutableFixed32List
  288. enumerateValuesWithBlock:^(uint32_t value, __unused NSUInteger idx, __unused BOOL *stop) {
  289. result += GPBComputeFixed32Size(number, value);
  290. }];
  291. [storage_.legacy.mutableFixed64List
  292. enumerateValuesWithBlock:^(uint64_t value, __unused NSUInteger idx, __unused BOOL *stop) {
  293. result += GPBComputeFixed64Size(number, value);
  294. }];
  295. for (NSData *data in storage_.legacy.mutableLengthDelimitedList) {
  296. result += GPBComputeBytesSize(number, data);
  297. }
  298. for (GPBUnknownFieldSet *set in storage_.legacy.mutableGroupList) {
  299. result += GPBComputeUnknownGroupSize(number, set);
  300. }
  301. #pragma clang diagnostic pop
  302. return result;
  303. }
  304. - (void)writeAsMessageSetExtensionToOutput:(GPBCodedOutputStream *)output {
  305. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  306. for (NSData *data in storage_.legacy.mutableLengthDelimitedList) {
  307. [output writeRawMessageSetExtension:number_ value:data];
  308. }
  309. }
  310. - (size_t)serializedSizeAsMessageSetExtension {
  311. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  312. size_t result = 0;
  313. for (NSData *data in storage_.legacy.mutableLengthDelimitedList) {
  314. result += GPBComputeRawMessageSetExtensionSize(number_, data);
  315. }
  316. return result;
  317. }
  318. - (NSString *)description {
  319. NSMutableString *description =
  320. [NSMutableString stringWithFormat:@"<%@ %p>: Field: %d", [self class], self, number_];
  321. switch (type_) {
  322. case GPBUnknownFieldTypeVarint:
  323. [description appendFormat:@" varint: %llu", storage_.intValue];
  324. break;
  325. case GPBUnknownFieldTypeFixed32:
  326. [description appendFormat:@" fixed32: %u", (uint32_t)storage_.intValue];
  327. break;
  328. case GPBUnknownFieldTypeFixed64:
  329. [description appendFormat:@" fixed64: %llu", storage_.intValue];
  330. break;
  331. case GPBUnknownFieldTypeLengthDelimited:
  332. [description appendFormat:@" fixed64: %@", storage_.lengthDelimited];
  333. break;
  334. case GPBUnknownFieldTypeGroup:
  335. [description appendFormat:@" group: %@", storage_.group];
  336. break;
  337. case GPBUnknownFieldTypeLegacy:
  338. #pragma clang diagnostic push
  339. #pragma clang diagnostic ignored "-Wdeprecated-declarations"
  340. [description appendString:@" {\n"];
  341. [storage_.legacy.mutableVarintList
  342. enumerateValuesWithBlock:^(uint64_t value, __unused NSUInteger idx, __unused BOOL *stop) {
  343. [description appendFormat:@"\t%llu\n", value];
  344. }];
  345. [storage_.legacy.mutableFixed32List
  346. enumerateValuesWithBlock:^(uint32_t value, __unused NSUInteger idx, __unused BOOL *stop) {
  347. [description appendFormat:@"\t%u\n", value];
  348. }];
  349. [storage_.legacy.mutableFixed64List
  350. enumerateValuesWithBlock:^(uint64_t value, __unused NSUInteger idx, __unused BOOL *stop) {
  351. [description appendFormat:@"\t%llu\n", value];
  352. }];
  353. for (NSData *data in storage_.legacy.mutableLengthDelimitedList) {
  354. [description appendFormat:@"\t%@\n", data];
  355. }
  356. for (GPBUnknownFieldSet *set in storage_.legacy.mutableGroupList) {
  357. [description appendFormat:@"\t%@\n", set];
  358. }
  359. [description appendString:@"}"];
  360. #pragma clang diagnostic pop
  361. break;
  362. }
  363. return description;
  364. }
  365. - (void)mergeFromField:(GPBUnknownField *)other {
  366. #pragma clang diagnostic push
  367. #pragma clang diagnostic ignored "-Wdeprecated-declarations"
  368. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  369. GPBUInt64Array *otherVarintList = other.varintList;
  370. if (otherVarintList.count > 0) {
  371. if (storage_.legacy.mutableVarintList == nil) {
  372. storage_.legacy.mutableVarintList = [otherVarintList copy];
  373. } else {
  374. [storage_.legacy.mutableVarintList addValuesFromArray:otherVarintList];
  375. }
  376. }
  377. GPBUInt32Array *otherFixed32List = other.fixed32List;
  378. if (otherFixed32List.count > 0) {
  379. if (storage_.legacy.mutableFixed32List == nil) {
  380. storage_.legacy.mutableFixed32List = [otherFixed32List copy];
  381. } else {
  382. [storage_.legacy.mutableFixed32List addValuesFromArray:otherFixed32List];
  383. }
  384. }
  385. GPBUInt64Array *otherFixed64List = other.fixed64List;
  386. if (otherFixed64List.count > 0) {
  387. if (storage_.legacy.mutableFixed64List == nil) {
  388. storage_.legacy.mutableFixed64List = [otherFixed64List copy];
  389. } else {
  390. [storage_.legacy.mutableFixed64List addValuesFromArray:otherFixed64List];
  391. }
  392. }
  393. NSArray *otherLengthDelimitedList = other.lengthDelimitedList;
  394. if (otherLengthDelimitedList.count > 0) {
  395. if (storage_.legacy.mutableLengthDelimitedList == nil) {
  396. storage_.legacy.mutableLengthDelimitedList = [otherLengthDelimitedList mutableCopy];
  397. } else {
  398. [storage_.legacy.mutableLengthDelimitedList addObjectsFromArray:otherLengthDelimitedList];
  399. }
  400. }
  401. NSArray *otherGroupList = other.groupList;
  402. if (otherGroupList.count > 0) {
  403. if (storage_.legacy.mutableGroupList == nil) {
  404. storage_.legacy.mutableGroupList =
  405. [[NSMutableArray alloc] initWithCapacity:otherGroupList.count];
  406. }
  407. // Make our own mutable copies.
  408. for (GPBUnknownFieldSet *group in otherGroupList) {
  409. GPBUnknownFieldSet *copied = [group copy];
  410. [storage_.legacy.mutableGroupList addObject:copied];
  411. [copied release];
  412. }
  413. }
  414. #pragma clang diagnostic pop
  415. }
  416. - (void)addVarint:(uint64_t)value {
  417. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  418. if (storage_.legacy.mutableVarintList == nil) {
  419. storage_.legacy.mutableVarintList = [[GPBUInt64Array alloc] initWithValues:&value count:1];
  420. } else {
  421. [storage_.legacy.mutableVarintList addValue:value];
  422. }
  423. }
  424. - (void)addFixed32:(uint32_t)value {
  425. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  426. if (storage_.legacy.mutableFixed32List == nil) {
  427. storage_.legacy.mutableFixed32List = [[GPBUInt32Array alloc] initWithValues:&value count:1];
  428. } else {
  429. [storage_.legacy.mutableFixed32List addValue:value];
  430. }
  431. }
  432. - (void)addFixed64:(uint64_t)value {
  433. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  434. if (storage_.legacy.mutableFixed64List == nil) {
  435. storage_.legacy.mutableFixed64List = [[GPBUInt64Array alloc] initWithValues:&value count:1];
  436. } else {
  437. [storage_.legacy.mutableFixed64List addValue:value];
  438. }
  439. }
  440. - (void)addLengthDelimited:(NSData *)value {
  441. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  442. if (storage_.legacy.mutableLengthDelimitedList == nil) {
  443. storage_.legacy.mutableLengthDelimitedList = [[NSMutableArray alloc] initWithObjects:&value
  444. count:1];
  445. } else {
  446. [storage_.legacy.mutableLengthDelimitedList addObject:value];
  447. }
  448. }
  449. - (void)addGroup:(GPBUnknownFieldSet *)value {
  450. ASSERT_FIELD_TYPE(GPBUnknownFieldTypeLegacy);
  451. if (storage_.legacy.mutableGroupList == nil) {
  452. storage_.legacy.mutableGroupList = [[NSMutableArray alloc] initWithObjects:&value count:1];
  453. } else {
  454. [storage_.legacy.mutableGroupList addObject:value];
  455. }
  456. }
  457. #pragma clang diagnostic pop
  458. @end