src/app/WriteClient.cpp - third_party/github/project-chip/connectedhomeip - Git at Google

 /*
  *
  *    Copyright (c) 2021 Project CHIP Authors
  *    All rights reserved.
  *
  *    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.
  */

 /**
  *    @file
  *      This file defines the initiator side of a CHIP Write Interaction.
  *
  */

 #include "lib/core/CHIPError.h"
 #include <app/AppBuildConfig.h>
 #include <app/InteractionModelEngine.h>
 #include <app/TimedRequest.h>
 #include <app/WriteClient.h>

 namespace chip {
 namespace app {

 CHIP_ERROR WriteClient::Init()
 {
     VerifyOrReturnError(mState == State::Uninitialized, CHIP_ERROR_INCORRECT_STATE);
     VerifyOrReturnError(mpExchangeMgr != nullptr, CHIP_ERROR_INCORRECT_STATE);
     VerifyOrReturnError(mpExchangeCtx == nullptr, CHIP_ERROR_INCORRECT_STATE);

     System::PacketBufferHandle packet = System::PacketBufferHandle::New(chip::app::kMaxSecureSduLengthBytes);
     VerifyOrReturnError(!packet.IsNull(), CHIP_ERROR_NO_MEMORY);

     mMessageWriter.Init(std::move(packet));

     ReturnErrorOnFailure(mWriteRequestBuilder.Init(&mMessageWriter));
     mWriteRequestBuilder.TimedRequest(mTimedWriteTimeoutMs.HasValue());
     ReturnErrorOnFailure(mWriteRequestBuilder.GetError());
     mWriteRequestBuilder.CreateWriteRequests();
     ReturnErrorOnFailure(mWriteRequestBuilder.GetError());

     MoveToState(State::Initialized);

     return CHIP_NO_ERROR;
 }

 void WriteClient::Close()
 {
     MoveToState(State::AwaitingDestruction);

     // OnDone below can destroy us before we unwind all the way back into the
     // exchange code and it tries to close itself.  Make sure that it doesn't
     // try to notify us that it's closing, since we will be dead.
     //
     // For more details, see #10344.
     if (mpExchangeCtx != nullptr)
     {
         mpExchangeCtx->SetDelegate(nullptr);
     }

     mpExchangeCtx = nullptr;

     if (mpCallback)
     {
         mpCallback->OnDone(this);
     }
 }

 void WriteClient::Abort()
 {
     //
     // If the exchange context hasn't already been gracefully closed
     // (signaled by setting it to null), then we need to forcibly
     // tear it down.
     //
     if (mpExchangeCtx != nullptr)
     {
         // We might be a delegate for this exchange, and we don't want the
         // OnExchangeClosing notification in that case.  Null out the delegate
         // to avoid that.
         //
         // TODO: This makes all sorts of assumptions about what the delegate is
         // (notice the "might" above!) that might not hold in practice.  We
         // really need a better solution here....
         mpExchangeCtx->SetDelegate(nullptr);
         mpExchangeCtx->Abort();
         mpExchangeCtx = nullptr;
     }
 }

 CHIP_ERROR WriteClient::ProcessWriteResponseMessage(System::PacketBufferHandle && payload)
 {
     CHIP_ERROR err = CHIP_NO_ERROR;
     System::PacketBufferTLVReader reader;
     TLV::TLVReader attributeStatusesReader;
     WriteResponseMessage::Parser writeResponse;
     AttributeStatusIBs::Parser attributeStatusesParser;

     reader.Init(std::move(payload));
     err = reader.Next();
     SuccessOrExit(err);

     err = writeResponse.Init(reader);
     SuccessOrExit(err);

 #if CHIP_CONFIG_IM_ENABLE_SCHEMA_CHECK
     err = writeResponse.CheckSchemaValidity();
     SuccessOrExit(err);
 #endif
     err = writeResponse.GetWriteResponses(&attributeStatusesParser);
     SuccessOrExit(err);

     attributeStatusesParser.GetReader(&attributeStatusesReader);

     while (CHIP_NO_ERROR == (err = attributeStatusesReader.Next()))
     {
         VerifyOrExit(TLV::AnonymousTag() == attributeStatusesReader.GetTag(), err = CHIP_ERROR_INVALID_TLV_TAG);

         AttributeStatusIB::Parser element;

         err = element.Init(attributeStatusesReader);
         SuccessOrExit(err);

         err = ProcessAttributeStatusIB(element);
         SuccessOrExit(err);
     }

     // if we have exhausted this container
     if (CHIP_END_OF_TLV == err)
     {
         err = CHIP_NO_ERROR;
     }

 exit:
     return err;
 }

 CHIP_ERROR WriteClient::PrepareAttribute(const AttributePathParams & attributePathParams)
 {
     if (mState == State::Uninitialized)
     {
         ReturnErrorOnFailure(Init());
     }
     VerifyOrReturnError(attributePathParams.IsValidAttributePath(), CHIP_ERROR_INVALID_PATH_LIST);
     AttributeDataIBs::Builder & writeRequests  = mWriteRequestBuilder.GetWriteRequests();
     AttributeDataIB::Builder & attributeDataIB = writeRequests.CreateAttributeDataIBBuilder();
     ReturnErrorOnFailure(writeRequests.GetError());
     // TODO: Add attribute version support
     attributeDataIB.DataVersion(0);
     ReturnErrorOnFailure(attributeDataIB.GetError());
     AttributePathIB::Builder & path = attributeDataIB.CreatePath();
     ReturnErrorOnFailure(path.Encode(attributePathParams));
     return CHIP_NO_ERROR;
 }

 CHIP_ERROR WriteClient::FinishAttribute()
 {
     AttributeDataIB::Builder & attributeDataIB = mWriteRequestBuilder.GetWriteRequests().GetAttributeDataIBBuilder();
     attributeDataIB.EndOfAttributeDataIB();
     ReturnErrorOnFailure(attributeDataIB.GetError());
     MoveToState(State::AddAttribute);
     return CHIP_NO_ERROR;
 }

 TLV::TLVWriter * WriteClient::GetAttributeDataIBTLVWriter()
 {
     return mWriteRequestBuilder.GetWriteRequests().GetAttributeDataIBBuilder().GetWriter();
 }

 CHIP_ERROR WriteClient::FinalizeMessage(System::PacketBufferHandle & aPacket)
 {
     VerifyOrReturnError(mState == State::AddAttribute, CHIP_ERROR_INCORRECT_STATE);
     AttributeDataIBs::Builder & attributeDataIBsBuilder = mWriteRequestBuilder.GetWriteRequests().EndOfAttributeDataIBs();
     ReturnErrorOnFailure(attributeDataIBsBuilder.GetError());

     mWriteRequestBuilder.IsFabricFiltered(false).EndOfWriteRequestMessage();
     ReturnErrorOnFailure(mWriteRequestBuilder.GetError());
     ReturnErrorOnFailure(mMessageWriter.Finalize(&aPacket));
     return CHIP_NO_ERROR;
 }

 const char * WriteClient::GetStateStr() const
 {
 #if CHIP_DETAIL_LOGGING
     switch (mState)
     {
     case State::Uninitialized:
         return "Uninitialized";

     case State::Initialized:
         return "Initialized";

     case State::AddAttribute:
         return "AddAttribute";

     case State::AwaitingTimedStatus:
         return "AwaitingTimedStatus";

     case State::AwaitingResponse:
         return "AwaitingResponse";

     case State::ResponseReceived:
         return "ResponseReceived";

     case State::AwaitingDestruction:
         return "AwaitingDestruction";
     }
 #endif // CHIP_DETAIL_LOGGING
     return "N/A";
 }

 void WriteClient::MoveToState(const State aTargetState)
 {
     mState = aTargetState;
     ChipLogDetail(DataManagement, "WriteClient moving to [%10.10s]", GetStateStr());
 }

 void WriteClient::ClearState()
 {
     MoveToState(State::Uninitialized);
 }

 CHIP_ERROR WriteClient::SendWriteRequest(const SessionHandle & session, System::Clock::Timeout timeout)
 {
     CHIP_ERROR err = CHIP_NO_ERROR;

     VerifyOrExit(mState == State::AddAttribute, err = CHIP_ERROR_INCORRECT_STATE);

     err = FinalizeMessage(mPendingWriteData);
     SuccessOrExit(err);

     // Create a new exchange context.
     mpExchangeCtx = mpExchangeMgr->NewContext(session, this);
     VerifyOrExit(mpExchangeCtx != nullptr, err = CHIP_ERROR_NO_MEMORY);

     mpExchangeCtx->SetResponseTimeout(timeout);

     if (mTimedWriteTimeoutMs.HasValue())
     {
         err = TimedRequest::Send(mpExchangeCtx, mTimedWriteTimeoutMs.Value());
         SuccessOrExit(err);
         MoveToState(State::AwaitingTimedStatus);
     }
     else
     {
         err = SendWriteRequest();
         SuccessOrExit(err);
     }

 exit:
     if (err != CHIP_NO_ERROR)
     {
         ChipLogError(DataManagement, "Write client failed to SendWriteRequest: %s", ErrorStr(err));
     }
     else
     {
         // TODO: Ideally this would happen async, but to make sure that we
         // handle this object dying (e.g. due to IM enging shutdown) while the
         // async bits are pending we'd need to malloc some state bit that we can
         // twiddle if we die.  For now just do the OnDone callback sync.
         if (session->IsGroupSession())
         {
             // Always shutdown on Group communication
             ChipLogDetail(DataManagement, "Closing on group Communication ");

             // Tell the application to release the object.
             // TODO: Consumers expect to hand off ownership of the WriteClient and wait for OnDone
             // after SendWriteRequest returns success.  Calling OnDone before returning is weird.
             // Need to refactor the code to avoid this.
             Close();
         }
     }

     return err;
 }

 CHIP_ERROR WriteClient::SendWriteRequest()
 {
     using namespace Protocols::InteractionModel;
     using namespace Messaging;

     // kExpectResponse is ignored by ExchangeContext in case of groupcast
     ReturnErrorOnFailure(
         mpExchangeCtx->SendMessage(MsgType::WriteRequest, std::move(mPendingWriteData), SendMessageFlags::kExpectResponse));
     MoveToState(State::AwaitingResponse);
     return CHIP_NO_ERROR;
 }

 CHIP_ERROR WriteClient::OnMessageReceived(Messaging::ExchangeContext * apExchangeContext, const PayloadHeader & aPayloadHeader,
                                           System::PacketBufferHandle && aPayload)
 {
     if (mState == State::AwaitingResponse)
     {
         MoveToState(State::ResponseReceived);
     }

     CHIP_ERROR err = CHIP_NO_ERROR;
     // Assert that the exchange context matches the client's current context.
     // This should never fail because even if SendWriteRequest is called
     // back-to-back, the second call will call Close() on the first exchange,
     // which clears the OnMessageReceived callback.
     VerifyOrExit(apExchangeContext == mpExchangeCtx, err = CHIP_ERROR_INCORRECT_STATE);

     if (mState == State::AwaitingTimedStatus)
     {
         err = HandleTimedStatus(aPayloadHeader, std::move(aPayload));
         // Skip all other processing here (which is for the response to the
         // write request), no matter whether err is success or not.
         goto exit;
     }

     if (aPayloadHeader.HasMessageType(Protocols::InteractionModel::MsgType::WriteResponse))
     {
         err = ProcessWriteResponseMessage(std::move(aPayload));
         SuccessOrExit(err);
     }
     else if (aPayloadHeader.HasMessageType(Protocols::InteractionModel::MsgType::StatusResponse))
     {
         err = StatusResponse::ProcessStatusResponse(std::move(aPayload));
         SuccessOrExit(err);
     }
     else
     {
         err = CHIP_ERROR_INVALID_MESSAGE_TYPE;
     }

 exit:
     if (mpCallback != nullptr)
     {
         if (err != CHIP_NO_ERROR)
         {
             mpCallback->OnError(this, err);
         }
     }

     if (mState != State::AwaitingResponse)
     {
         Close();
     }
     // Else we got a response to a Timed Request and just sent the write.

     return err;
 }

 void WriteClient::OnResponseTimeout(Messaging::ExchangeContext * apExchangeContext)
 {
     ChipLogProgress(DataManagement, "Time out! failed to receive write response from Exchange: " ChipLogFormatExchange,
                     ChipLogValueExchange(apExchangeContext));

     if (mpCallback != nullptr)
     {
         mpCallback->OnError(this, CHIP_ERROR_TIMEOUT);
     }
     Close();
 }

 CHIP_ERROR WriteClient::ProcessAttributeStatusIB(AttributeStatusIB::Parser & aAttributeStatusIB)
 {
     CHIP_ERROR err = CHIP_NO_ERROR;
     AttributePathIB::Parser attributePath;
     StatusIB statusIB;
     StatusIB::Parser StatusIBParser;
     AttributePathParams attributePathParams;

     err = aAttributeStatusIB.GetPath(&attributePath);
     SuccessOrExit(err);
     err = attributePath.GetCluster(&(attributePathParams.mClusterId));
     SuccessOrExit(err);
     err = attributePath.GetEndpoint(&(attributePathParams.mEndpointId));
     SuccessOrExit(err);
     err = attributePath.GetAttribute(&(attributePathParams.mAttributeId));
     SuccessOrExit(err);
     err = attributePath.GetListIndex(&(attributePathParams.mListIndex));
     if (err == CHIP_END_OF_TLV)
     {
         err = CHIP_NO_ERROR;
     }
     // TODO: (#11423) Attribute paths has a pattern of invalid paths, should add a function for checking invalid paths here.
     // NOTE: We don't support wildcard write for now, reject all wildcard paths.
     VerifyOrExit(!attributePathParams.HasAttributeWildcard() && attributePathParams.IsValidAttributePath(),
                  err = CHIP_ERROR_IM_MALFORMED_ATTRIBUTE_PATH);

     err = aAttributeStatusIB.GetErrorStatus(&(StatusIBParser));
     if (CHIP_NO_ERROR == err)
     {
         err = StatusIBParser.DecodeStatusIB(statusIB);
         SuccessOrExit(err);
         if (mpCallback != nullptr)
         {
             ConcreteAttributePath path(attributePathParams.mEndpointId, attributePathParams.mClusterId,
                                        attributePathParams.mAttributeId);
             mpCallback->OnResponse(this, path, statusIB);
         }
     }

 exit:
     return err;
 }

 CHIP_ERROR WriteClient::HandleTimedStatus(const PayloadHeader & aPayloadHeader, System::PacketBufferHandle && aPayload)
 {
     ReturnErrorOnFailure(TimedRequest::HandleResponse(aPayloadHeader, std::move(aPayload)));

     return SendWriteRequest();
 }

 } // namespace app
 } // namespace chip
	/*
	*
	* Copyright (c) 2021 Project CHIP Authors
	* All rights reserved.
	*
	* 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.
	*/

	/**
	* @file
	* This file defines the initiator side of a CHIP Write Interaction.
	*
	*/

	#include "lib/core/CHIPError.h"
	#include <app/AppBuildConfig.h>
	#include <app/InteractionModelEngine.h>
	#include <app/TimedRequest.h>
	#include <app/WriteClient.h>

	namespace chip {
	namespace app {

	CHIP_ERROR WriteClient::Init()
	{
	VerifyOrReturnError(mState == State::Uninitialized, CHIP_ERROR_INCORRECT_STATE);
	VerifyOrReturnError(mpExchangeMgr != nullptr, CHIP_ERROR_INCORRECT_STATE);
	VerifyOrReturnError(mpExchangeCtx == nullptr, CHIP_ERROR_INCORRECT_STATE);

	System::PacketBufferHandle packet = System::PacketBufferHandle::New(chip::app::kMaxSecureSduLengthBytes);
	VerifyOrReturnError(!packet.IsNull(), CHIP_ERROR_NO_MEMORY);

	mMessageWriter.Init(std::move(packet));

	ReturnErrorOnFailure(mWriteRequestBuilder.Init(&mMessageWriter));
	mWriteRequestBuilder.TimedRequest(mTimedWriteTimeoutMs.HasValue());
	ReturnErrorOnFailure(mWriteRequestBuilder.GetError());
	mWriteRequestBuilder.CreateWriteRequests();
	ReturnErrorOnFailure(mWriteRequestBuilder.GetError());

	MoveToState(State::Initialized);

	return CHIP_NO_ERROR;
	}

	void WriteClient::Close()
	{
	MoveToState(State::AwaitingDestruction);

	// OnDone below can destroy us before we unwind all the way back into the
	// exchange code and it tries to close itself. Make sure that it doesn't
	// try to notify us that it's closing, since we will be dead.
	//
	// For more details, see #10344.
	if (mpExchangeCtx != nullptr)
	{
	mpExchangeCtx->SetDelegate(nullptr);
	}

	mpExchangeCtx = nullptr;

	if (mpCallback)
	{
	mpCallback->OnDone(this);
	}
	}

	void WriteClient::Abort()
	{
	//
	// If the exchange context hasn't already been gracefully closed
	// (signaled by setting it to null), then we need to forcibly
	// tear it down.
	//
	if (mpExchangeCtx != nullptr)
	{
	// We might be a delegate for this exchange, and we don't want the
	// OnExchangeClosing notification in that case. Null out the delegate
	// to avoid that.
	//
	// TODO: This makes all sorts of assumptions about what the delegate is
	// (notice the "might" above!) that might not hold in practice. We
	// really need a better solution here....
	mpExchangeCtx->SetDelegate(nullptr);
	mpExchangeCtx->Abort();
	mpExchangeCtx = nullptr;
	}
	}

	CHIP_ERROR WriteClient::ProcessWriteResponseMessage(System::PacketBufferHandle && payload)
	{
	CHIP_ERROR err = CHIP_NO_ERROR;
	System::PacketBufferTLVReader reader;
	TLV::TLVReader attributeStatusesReader;
	WriteResponseMessage::Parser writeResponse;
	AttributeStatusIBs::Parser attributeStatusesParser;

	reader.Init(std::move(payload));
	err = reader.Next();
	SuccessOrExit(err);

	err = writeResponse.Init(reader);
	SuccessOrExit(err);

	#if CHIP_CONFIG_IM_ENABLE_SCHEMA_CHECK
	err = writeResponse.CheckSchemaValidity();
	SuccessOrExit(err);
	#endif
	err = writeResponse.GetWriteResponses(&attributeStatusesParser);
	SuccessOrExit(err);

	attributeStatusesParser.GetReader(&attributeStatusesReader);

	while (CHIP_NO_ERROR == (err = attributeStatusesReader.Next()))
	{
	VerifyOrExit(TLV::AnonymousTag() == attributeStatusesReader.GetTag(), err = CHIP_ERROR_INVALID_TLV_TAG);

	AttributeStatusIB::Parser element;

	err = element.Init(attributeStatusesReader);
	SuccessOrExit(err);

	err = ProcessAttributeStatusIB(element);
	SuccessOrExit(err);
	}

	// if we have exhausted this container
	if (CHIP_END_OF_TLV == err)
	{
	err = CHIP_NO_ERROR;
	}

	exit:
	return err;
	}

	CHIP_ERROR WriteClient::PrepareAttribute(const AttributePathParams & attributePathParams)
	{
	if (mState == State::Uninitialized)
	{
	ReturnErrorOnFailure(Init());
	}
	VerifyOrReturnError(attributePathParams.IsValidAttributePath(), CHIP_ERROR_INVALID_PATH_LIST);
	AttributeDataIBs::Builder & writeRequests = mWriteRequestBuilder.GetWriteRequests();
	AttributeDataIB::Builder & attributeDataIB = writeRequests.CreateAttributeDataIBBuilder();
	ReturnErrorOnFailure(writeRequests.GetError());
	// TODO: Add attribute version support
	attributeDataIB.DataVersion(0);
	ReturnErrorOnFailure(attributeDataIB.GetError());
	AttributePathIB::Builder & path = attributeDataIB.CreatePath();
	ReturnErrorOnFailure(path.Encode(attributePathParams));
	return CHIP_NO_ERROR;
	}

	CHIP_ERROR WriteClient::FinishAttribute()
	{
	AttributeDataIB::Builder & attributeDataIB = mWriteRequestBuilder.GetWriteRequests().GetAttributeDataIBBuilder();
	attributeDataIB.EndOfAttributeDataIB();
	ReturnErrorOnFailure(attributeDataIB.GetError());
	MoveToState(State::AddAttribute);
	return CHIP_NO_ERROR;
	}

	TLV::TLVWriter * WriteClient::GetAttributeDataIBTLVWriter()
	{
	return mWriteRequestBuilder.GetWriteRequests().GetAttributeDataIBBuilder().GetWriter();
	}

	CHIP_ERROR WriteClient::FinalizeMessage(System::PacketBufferHandle & aPacket)
	{
	VerifyOrReturnError(mState == State::AddAttribute, CHIP_ERROR_INCORRECT_STATE);
	AttributeDataIBs::Builder & attributeDataIBsBuilder = mWriteRequestBuilder.GetWriteRequests().EndOfAttributeDataIBs();
	ReturnErrorOnFailure(attributeDataIBsBuilder.GetError());

	mWriteRequestBuilder.IsFabricFiltered(false).EndOfWriteRequestMessage();
	ReturnErrorOnFailure(mWriteRequestBuilder.GetError());
	ReturnErrorOnFailure(mMessageWriter.Finalize(&aPacket));
	return CHIP_NO_ERROR;
	}

	const char * WriteClient::GetStateStr() const
	{
	#if CHIP_DETAIL_LOGGING
	switch (mState)
	{
	case State::Uninitialized:
	return "Uninitialized";

	case State::Initialized:
	return "Initialized";

	case State::AddAttribute:
	return "AddAttribute";

	case State::AwaitingTimedStatus:
	return "AwaitingTimedStatus";

	case State::AwaitingResponse:
	return "AwaitingResponse";

	case State::ResponseReceived:
	return "ResponseReceived";

	case State::AwaitingDestruction:
	return "AwaitingDestruction";
	}
	#endif // CHIP_DETAIL_LOGGING
	return "N/A";
	}

	void WriteClient::MoveToState(const State aTargetState)
	{
	mState = aTargetState;
	ChipLogDetail(DataManagement, "WriteClient moving to [%10.10s]", GetStateStr());
	}

	void WriteClient::ClearState()
	{
	MoveToState(State::Uninitialized);
	}

	CHIP_ERROR WriteClient::SendWriteRequest(const SessionHandle & session, System::Clock::Timeout timeout)
	{
	CHIP_ERROR err = CHIP_NO_ERROR;

	VerifyOrExit(mState == State::AddAttribute, err = CHIP_ERROR_INCORRECT_STATE);

	err = FinalizeMessage(mPendingWriteData);
	SuccessOrExit(err);

	// Create a new exchange context.
	mpExchangeCtx = mpExchangeMgr->NewContext(session, this);
	VerifyOrExit(mpExchangeCtx != nullptr, err = CHIP_ERROR_NO_MEMORY);

	mpExchangeCtx->SetResponseTimeout(timeout);

	if (mTimedWriteTimeoutMs.HasValue())
	{
	err = TimedRequest::Send(mpExchangeCtx, mTimedWriteTimeoutMs.Value());
	SuccessOrExit(err);
	MoveToState(State::AwaitingTimedStatus);
	}
	else
	{
	err = SendWriteRequest();
	SuccessOrExit(err);
	}

	exit:
	if (err != CHIP_NO_ERROR)
	{
	ChipLogError(DataManagement, "Write client failed to SendWriteRequest: %s", ErrorStr(err));
	}
	else
	{
	// TODO: Ideally this would happen async, but to make sure that we
	// handle this object dying (e.g. due to IM enging shutdown) while the
	// async bits are pending we'd need to malloc some state bit that we can
	// twiddle if we die. For now just do the OnDone callback sync.
	if (session->IsGroupSession())
	{
	// Always shutdown on Group communication
	ChipLogDetail(DataManagement, "Closing on group Communication ");

	// Tell the application to release the object.
	// TODO: Consumers expect to hand off ownership of the WriteClient and wait for OnDone
	// after SendWriteRequest returns success. Calling OnDone before returning is weird.
	// Need to refactor the code to avoid this.
	Close();
	}
	}

	return err;
	}

	CHIP_ERROR WriteClient::SendWriteRequest()
	{
	using namespace Protocols::InteractionModel;
	using namespace Messaging;

	// kExpectResponse is ignored by ExchangeContext in case of groupcast
	ReturnErrorOnFailure(
	mpExchangeCtx->SendMessage(MsgType::WriteRequest, std::move(mPendingWriteData), SendMessageFlags::kExpectResponse));
	MoveToState(State::AwaitingResponse);
	return CHIP_NO_ERROR;
	}

	CHIP_ERROR WriteClient::OnMessageReceived(Messaging::ExchangeContext * apExchangeContext, const PayloadHeader & aPayloadHeader,
	System::PacketBufferHandle && aPayload)
	{
	if (mState == State::AwaitingResponse)
	{
	MoveToState(State::ResponseReceived);
	}

	CHIP_ERROR err = CHIP_NO_ERROR;
	// Assert that the exchange context matches the client's current context.
	// This should never fail because even if SendWriteRequest is called
	// back-to-back, the second call will call Close() on the first exchange,
	// which clears the OnMessageReceived callback.
	VerifyOrExit(apExchangeContext == mpExchangeCtx, err = CHIP_ERROR_INCORRECT_STATE);

	if (mState == State::AwaitingTimedStatus)
	{
	err = HandleTimedStatus(aPayloadHeader, std::move(aPayload));
	// Skip all other processing here (which is for the response to the
	// write request), no matter whether err is success or not.
	goto exit;
	}

	if (aPayloadHeader.HasMessageType(Protocols::InteractionModel::MsgType::WriteResponse))
	{
	err = ProcessWriteResponseMessage(std::move(aPayload));
	SuccessOrExit(err);
	}
	else if (aPayloadHeader.HasMessageType(Protocols::InteractionModel::MsgType::StatusResponse))
	{
	err = StatusResponse::ProcessStatusResponse(std::move(aPayload));
	SuccessOrExit(err);
	}
	else
	{
	err = CHIP_ERROR_INVALID_MESSAGE_TYPE;
	}

	exit:
	if (mpCallback != nullptr)
	{
	if (err != CHIP_NO_ERROR)
	{
	mpCallback->OnError(this, err);
	}
	}

	if (mState != State::AwaitingResponse)
	{
	Close();
	}
	// Else we got a response to a Timed Request and just sent the write.

	return err;
	}

	void WriteClient::OnResponseTimeout(Messaging::ExchangeContext * apExchangeContext)
	{
	ChipLogProgress(DataManagement, "Time out! failed to receive write response from Exchange: " ChipLogFormatExchange,
	ChipLogValueExchange(apExchangeContext));

	if (mpCallback != nullptr)
	{
	mpCallback->OnError(this, CHIP_ERROR_TIMEOUT);
	}
	Close();
	}

	CHIP_ERROR WriteClient::ProcessAttributeStatusIB(AttributeStatusIB::Parser & aAttributeStatusIB)
	{
	CHIP_ERROR err = CHIP_NO_ERROR;
	AttributePathIB::Parser attributePath;
	StatusIB statusIB;
	StatusIB::Parser StatusIBParser;
	AttributePathParams attributePathParams;

	err = aAttributeStatusIB.GetPath(&attributePath);
	SuccessOrExit(err);
	err = attributePath.GetCluster(&(attributePathParams.mClusterId));
	SuccessOrExit(err);
	err = attributePath.GetEndpoint(&(attributePathParams.mEndpointId));
	SuccessOrExit(err);
	err = attributePath.GetAttribute(&(attributePathParams.mAttributeId));
	SuccessOrExit(err);
	err = attributePath.GetListIndex(&(attributePathParams.mListIndex));
	if (err == CHIP_END_OF_TLV)
	{
	err = CHIP_NO_ERROR;
	}
	// TODO: (#11423) Attribute paths has a pattern of invalid paths, should add a function for checking invalid paths here.
	// NOTE: We don't support wildcard write for now, reject all wildcard paths.
	VerifyOrExit(!attributePathParams.HasAttributeWildcard() && attributePathParams.IsValidAttributePath(),
	err = CHIP_ERROR_IM_MALFORMED_ATTRIBUTE_PATH);

	err = aAttributeStatusIB.GetErrorStatus(&(StatusIBParser));
	if (CHIP_NO_ERROR == err)
	{
	err = StatusIBParser.DecodeStatusIB(statusIB);
	SuccessOrExit(err);
	if (mpCallback != nullptr)
	{
	ConcreteAttributePath path(attributePathParams.mEndpointId, attributePathParams.mClusterId,
	attributePathParams.mAttributeId);
	mpCallback->OnResponse(this, path, statusIB);
	}
	}

	exit:
	return err;
	}

	CHIP_ERROR WriteClient::HandleTimedStatus(const PayloadHeader & aPayloadHeader, System::PacketBufferHandle && aPayload)
	{
	ReturnErrorOnFailure(TimedRequest::HandleResponse(aPayloadHeader, std::move(aPayload)));

	return SendWriteRequest();
	}

	} // namespace app
	} // namespace chip