Service/Sources/EDOObjectReleaseMessage.m

//
// Copyright 2019 Google LLC.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

#import "Service/Sources/EDOObjectReleaseMessage.h"
#import "Service/Sources/EDOHostService.h"
#import "Service/Sources/EDOMessage.h"
#import "Service/Sources/EDOObject+Private.h"
#import "Service/Sources/EDOServicePort.h"
#import "Service/Sources/EDOServiceRequest.h"

#import "Service/Sources/EDOHostService+Private.h"

static NSString *const kEDOObjectReleaseCoderWeaklyReferencedKey = @"weaklyReferenced";
static NSString *const kEDOObjectReleaseCoderRemoteAddressKey = @"remoteAddress";
static NSString *const kEDOObjectReleaseCoderServicePortKey = @"servicePort";

@interface EDOObjectReleaseRequest ()

@property(nonatomic, readonly) EDOPointerType remoteAddress;
@property(nonatomic, readonly) EDOServicePort *servicePort;

/** Indicates whether the object to be released is a weakly referenced object. */
@property(nonatomic, readonly, getter=isWeaklyReferenced) BOOL weaklyReferenced;

@end

@implementation EDOObjectReleaseRequest

+ (BOOL)supportsSecureCoding {
  return YES;
}

- (instancetype)initWithRemoteAddress:(EDOPointerType)remoteAddress
                          servicePort:(EDOServicePort *)servicePort
                     weaklyReferenced:(BOOL)weaklyReferenced {
  self = [super init];
  if (self) {
    _remoteAddress = remoteAddress;
    _servicePort = servicePort;
    _weaklyReferenced = weaklyReferenced;
  }
  return self;
}

+ (instancetype)requestWithRemoteAddress:(EDOPointerType)remoteAddress
                             servicePort:(EDOServicePort *)servicePort {
  return [[self alloc] initWithRemoteAddress:remoteAddress
                                 servicePort:servicePort
                            weaklyReferenced:NO];
}

+ (instancetype)requestWithWeakRemoteAddress:(EDOPointerType)remoteAddress
                                 servicePort:(EDOServicePort *)servicePort {
  return [[self alloc] initWithRemoteAddress:remoteAddress
                                 servicePort:servicePort
                            weaklyReferenced:YES];
}

- (instancetype)initWithCoder:(NSCoder *)aDecoder {
  self = [super initWithCoder:aDecoder];
  if (self) {
    _remoteAddress = [aDecoder decodeInt64ForKey:kEDOObjectReleaseCoderRemoteAddressKey];
    _weaklyReferenced = [aDecoder decodeBoolForKey:kEDOObjectReleaseCoderWeaklyReferencedKey];
    _servicePort = [aDecoder decodeObjectOfClass:[EDOServicePort class]
                                          forKey:kEDOObjectReleaseCoderServicePortKey];
  }
  return self;
}

- (void)encodeWithCoder:(NSCoder *)aCoder {
  [super encodeWithCoder:aCoder];
  [aCoder encodeInt64:self.remoteAddress forKey:kEDOObjectReleaseCoderRemoteAddressKey];
  [aCoder encodeBool:self.weaklyReferenced forKey:kEDOObjectReleaseCoderWeaklyReferencedKey];
  [aCoder encodeObject:self.servicePort forKey:kEDOObjectReleaseCoderServicePortKey];
}

+ (EDORequestHandler)requestHandler {
  return ^(EDOServiceRequest *request, EDOHostService *service) {
    EDOObjectReleaseRequest *releaseRequest = (EDOObjectReleaseRequest *)request;

    // Safety check: ensure the request is meant for this service.
    if (releaseRequest.servicePort && ![releaseRequest.servicePort match:service.port]) {
      // Ignore release requests from other service instances (e.g. if port was recycled).
      return [[EDOServiceResponse alloc] initWithMessageID:request.messageID];
    }

    EDOPointerType edoRemoteAddress = releaseRequest.remoteAddress;
    if (releaseRequest.weaklyReferenced) {
      [service removeWeakObjectWithAddress:edoRemoteAddress];
    } else {
      [service removeObjectWithAddress:edoRemoteAddress];
    }
    // The return response from the call is not being needed. So we return a generic message.
    return [[EDOServiceResponse alloc] initWithMessageID:request.messageID];
  };
}

@end