Merge changes Idf06a875,I6b10e93f

* changes:
  Make ChangeControl#getRange private and migrate caller
  Migrate ApprovalsUtil#checkApprovals() to PermissionBackend
This commit is contained in:
Patrick Hiesel
2017-09-07 12:12:22 +00:00
committed by Gerrit Code Review
7 changed files with 53 additions and 38 deletions

View File

@@ -30,6 +30,7 @@ import com.google.gerrit.reviewdb.server.ReviewDb;
import com.google.gerrit.server.change.ChangeKindCache;
import com.google.gerrit.server.git.LabelNormalizer;
import com.google.gerrit.server.notedb.NoteDbChangeState.PrimaryStorage;
import com.google.gerrit.server.permissions.PermissionBackendException;
import com.google.gerrit.server.project.ChangeControl;
import com.google.gerrit.server.project.ProjectCache;
import com.google.gerrit.server.project.ProjectState;
@@ -204,7 +205,7 @@ public class ApprovalCopier {
}
}
return labelNormalizer.normalize(ctl, byUser.values()).getNormalized();
} catch (IOException e) {
} catch (IOException | PermissionBackendException e) {
throw new OrmException(e);
}
}

View File

@@ -31,8 +31,6 @@ import com.google.common.primitives.Shorts;
import com.google.gerrit.common.Nullable;
import com.google.gerrit.common.data.LabelType;
import com.google.gerrit.common.data.LabelTypes;
import com.google.gerrit.common.data.Permission;
import com.google.gerrit.common.data.PermissionRange;
import com.google.gerrit.extensions.restapi.AuthException;
import com.google.gerrit.extensions.restapi.BadRequestException;
import com.google.gerrit.extensions.restapi.RestApiException;
@@ -48,6 +46,7 @@ import com.google.gerrit.server.notedb.ChangeUpdate;
import com.google.gerrit.server.notedb.NotesMigration;
import com.google.gerrit.server.notedb.ReviewerStateInternal;
import com.google.gerrit.server.permissions.ChangePermission;
import com.google.gerrit.server.permissions.LabelPermission;
import com.google.gerrit.server.permissions.PermissionBackend;
import com.google.gerrit.server.permissions.PermissionBackendException;
import com.google.gerrit.server.project.ChangeControl;
@@ -308,7 +307,7 @@ public class ApprovalsUtil {
* @param update change update.
* @param labelTypes label types for the containing project.
* @param ps patch set being approved.
* @param changeCtl change control for user adding approvals.
* @param user user adding approvals.
* @param approvals approvals to add.
* @throws RestApiException
* @throws OrmException
@@ -318,10 +317,10 @@ public class ApprovalsUtil {
ChangeUpdate update,
LabelTypes labelTypes,
PatchSet ps,
ChangeControl changeCtl,
CurrentUser user,
Map<String, Short> approvals)
throws RestApiException, OrmException {
Account.Id accountId = changeCtl.getUser().getAccountId();
throws RestApiException, OrmException, PermissionBackendException {
Account.Id accountId = user.getAccountId();
checkArgument(
accountId.equals(ps.getUploader()),
"expected user %s to match patch set uploader %s",
@@ -330,12 +329,12 @@ public class ApprovalsUtil {
if (approvals.isEmpty()) {
return ImmutableList.of();
}
checkApprovals(approvals, changeCtl);
checkApprovals(approvals, permissionBackend.user(user).database(db).change(update.getNotes()));
List<PatchSetApproval> cells = new ArrayList<>(approvals.size());
Date ts = update.getWhen();
for (Map.Entry<String, Short> vote : approvals.entrySet()) {
LabelType lt = labelTypes.byLabel(vote.getKey());
cells.add(newApproval(ps.getId(), changeCtl.getUser(), lt.getLabelId(), vote.getValue(), ts));
cells.add(newApproval(ps.getId(), user, lt.getLabelId(), vote.getValue(), ts));
}
for (PatchSetApproval psa : cells) {
update.putApproval(psa.getLabel(), psa.getValue());
@@ -356,13 +355,15 @@ public class ApprovalsUtil {
}
}
private static void checkApprovals(Map<String, Short> approvals, ChangeControl changeCtl)
throws AuthException {
private static void checkApprovals(
Map<String, Short> approvals, PermissionBackend.ForChange forChange)
throws AuthException, PermissionBackendException {
for (Map.Entry<String, Short> vote : approvals.entrySet()) {
String name = vote.getKey();
Short value = vote.getValue();
PermissionRange range = changeCtl.getRange(Permission.forLabel(name));
if (range == null || !range.contains(value)) {
try {
forChange.check(new LabelPermission.WithValue(name, value));
} catch (AuthException e) {
throw new AuthException(
String.format("applying label \"%s\": %d is restricted", name, value));
}

View File

@@ -378,7 +378,7 @@ public class ChangeInserter implements InsertChangeOp {
@Override
public boolean updateChange(ChangeContext ctx)
throws RestApiException, OrmException, IOException {
throws RestApiException, OrmException, IOException, PermissionBackendException {
change = ctx.getChange(); // Use defensive copy created by ChangeControl.
ReviewDb db = ctx.getDb();
ChangeControl ctl = ctx.getControl();
@@ -444,7 +444,7 @@ public class ChangeInserter implements InsertChangeOp {
filterOnChangeVisibility(db, ctx.getNotes(), reviewersToAdd),
Collections.<Account.Id>emptySet());
approvalsUtil.addApprovalsForNewPatchSet(
db, update, labelTypes, patchSet, ctx.getControl(), approvals);
db, update, labelTypes, patchSet, ctx.getUser(), approvals);
// Check if approvals are changing in with this update. If so, add current user to reviewers.
// Note that this is done separately as addReviewers is filtering out the change owner as
// reviewer which is needed in several other code paths.

View File

@@ -24,13 +24,15 @@ import com.google.common.collect.Lists;
import com.google.gerrit.common.data.LabelType;
import com.google.gerrit.common.data.LabelTypes;
import com.google.gerrit.common.data.LabelValue;
import com.google.gerrit.common.data.Permission;
import com.google.gerrit.common.data.PermissionRange;
import com.google.gerrit.reviewdb.client.Account;
import com.google.gerrit.extensions.restapi.AuthException;
import com.google.gerrit.reviewdb.client.Change;
import com.google.gerrit.reviewdb.client.PatchSetApproval;
import com.google.gerrit.reviewdb.server.ReviewDb;
import com.google.gerrit.server.IdentifiedUser;
import com.google.gerrit.server.notedb.ChangeNotes;
import com.google.gerrit.server.permissions.LabelPermission;
import com.google.gerrit.server.permissions.PermissionBackend;
import com.google.gerrit.server.permissions.PermissionBackendException;
import com.google.gerrit.server.project.ChangeControl;
import com.google.gwtorm.server.OrmException;
import com.google.inject.Inject;
@@ -76,15 +78,18 @@ public class LabelNormalizer {
private final Provider<ReviewDb> db;
private final ChangeControl.GenericFactory changeFactory;
private final IdentifiedUser.GenericFactory userFactory;
private final PermissionBackend permissionBackend;
@Inject
LabelNormalizer(
Provider<ReviewDb> db,
ChangeControl.GenericFactory changeFactory,
IdentifiedUser.GenericFactory userFactory) {
IdentifiedUser.GenericFactory userFactory,
PermissionBackend permissionBackend) {
this.db = db;
this.changeFactory = changeFactory;
this.userFactory = userFactory;
this.permissionBackend = permissionBackend;
}
/**
@@ -96,7 +101,7 @@ public class LabelNormalizer {
* @throws OrmException
*/
public Result normalize(Change change, Collection<PatchSetApproval> approvals)
throws OrmException {
throws OrmException, PermissionBackendException {
IdentifiedUser user = userFactory.create(change.getOwner());
return normalize(changeFactory.controlFor(db.get(), change, user), approvals);
}
@@ -108,7 +113,8 @@ public class LabelNormalizer {
* for the user. Approvals for unknown labels are not included in the output, nor are
* approvals where the user has no permissions for that label.
*/
public Result normalize(ChangeControl ctl, Collection<PatchSetApproval> approvals) {
public Result normalize(ChangeControl ctl, Collection<PatchSetApproval> approvals)
throws PermissionBackendException {
List<PatchSetApproval> unchanged = Lists.newArrayListWithCapacity(approvals.size());
List<PatchSetApproval> updated = Lists.newArrayListWithCapacity(approvals.size());
List<PatchSetApproval> deleted = Lists.newArrayListWithCapacity(approvals.size());
@@ -132,7 +138,7 @@ public class LabelNormalizer {
}
PatchSetApproval copy = copy(psa);
applyTypeFloor(label, copy);
if (!applyRightFloor(ctl, label, copy)) {
if (!applyRightFloor(ctl.getNotes(), label, copy)) {
deleted.add(psa);
} else if (copy.getValue() != psa.getValue()) {
updated.add(copy);
@@ -147,19 +153,24 @@ public class LabelNormalizer {
return new PatchSetApproval(src.getPatchSetId(), src);
}
private PermissionRange getRange(ChangeControl ctl, LabelType lt, Account.Id id) {
String permission = Permission.forLabel(lt.getName());
IdentifiedUser user = userFactory.create(id);
return ctl.forUser(user).getRange(permission);
}
private boolean applyRightFloor(ChangeControl ctl, LabelType lt, PatchSetApproval a) {
PermissionRange range = getRange(ctl, lt, a.getAccountId());
if (range.isEmpty()) {
private boolean applyRightFloor(ChangeNotes notes, LabelType lt, PatchSetApproval a)
throws PermissionBackendException {
PermissionBackend.ForChange forChange =
permissionBackend.user(userFactory.create(a.getAccountId())).database(db).change(notes);
// Check if the user is allowed to vote on the label at all
try {
forChange.check(new LabelPermission(lt.getName()));
} catch (AuthException e) {
return false;
}
a.setValue((short) range.squash(a.getValue()));
return true;
// Squash vote to nearest allowed value
try {
forChange.check(new LabelPermission.WithValue(lt.getName(), a.getValue()));
return true;
} catch (AuthException e) {
a.setValue(forChange.squashThenCheck(lt, a.getValue()));
return true;
}
}
private void applyTypeFloor(LabelType lt, PatchSetApproval a) {

View File

@@ -52,6 +52,7 @@ import com.google.gerrit.server.mail.MailUtil.MailRecipients;
import com.google.gerrit.server.mail.send.ReplacePatchSetSender;
import com.google.gerrit.server.notedb.ChangeNotes;
import com.google.gerrit.server.notedb.ChangeUpdate;
import com.google.gerrit.server.permissions.PermissionBackendException;
import com.google.gerrit.server.project.ProjectCache;
import com.google.gerrit.server.project.ProjectControl;
import com.google.gerrit.server.query.change.ChangeData;
@@ -224,7 +225,7 @@ public class ReplaceOp implements BatchUpdateOp {
@Override
public boolean updateChange(ChangeContext ctx)
throws RestApiException, OrmException, IOException {
throws RestApiException, OrmException, IOException, PermissionBackendException {
notes = ctx.getNotes();
Change change = notes.getChange();
if (change == null || change.getStatus().isClosed()) {
@@ -306,7 +307,7 @@ public class ReplaceOp implements BatchUpdateOp {
update,
projectControl.getProjectState().getLabelTypes(),
newPatchSet,
ctx.getControl(),
ctx.getUser(),
approvals);
approvalCopier.copyInReviewDb(
ctx.getDb(),

View File

@@ -45,6 +45,7 @@ import com.google.gerrit.server.git.MergeUtil;
import com.google.gerrit.server.git.ProjectConfig;
import com.google.gerrit.server.git.SubmoduleException;
import com.google.gerrit.server.notedb.ChangeUpdate;
import com.google.gerrit.server.permissions.PermissionBackendException;
import com.google.gerrit.server.project.ProjectState;
import com.google.gerrit.server.update.BatchUpdateOp;
import com.google.gerrit.server.update.ChangeContext;
@@ -330,7 +331,7 @@ abstract class SubmitStrategyOp implements BatchUpdateOp {
}
private void setApproval(ChangeContext ctx, IdentifiedUser user)
throws OrmException, IOException {
throws OrmException, IOException, PermissionBackendException {
Change.Id id = ctx.getChange().getId();
List<SubmitRecord> records = args.commitStatus.getSubmitRecords(id);
PatchSet.Id oldPsId = toMerge.getPatchsetId();
@@ -352,7 +353,7 @@ abstract class SubmitStrategyOp implements BatchUpdateOp {
}
private LabelNormalizer.Result approve(ChangeContext ctx, ChangeUpdate update)
throws OrmException, IOException {
throws OrmException, IOException, PermissionBackendException {
PatchSet.Id psId = update.getPatchSetId();
Map<PatchSetApproval.Key, PatchSetApproval> byKey = new HashMap<>();
for (PatchSetApproval psa :

View File

@@ -269,7 +269,7 @@ public class ChangeControl {
}
/** The range of permitted values associated with a label permission. */
public PermissionRange getRange(String permission) {
private PermissionRange getRange(String permission) {
return getRefControl().getRange(permission, isOwner());
}